rewrite check on python

переписал на python.
Тут какой-то косяк