rewrite check on python

переписал на python.