[python] add find a square