import re import sys def isNip(s): pattern = re.compile(r'^\d{3}-\d{3}-\d{2}-\d{2}$|^\d{3}-\d{2}-\d{2}-\d{3}$') return bool(re.match(pattern, s)) inFile = sys.argv[1] outFile = sys.argv[2] # inFile = 'test.in' # outFile = 'test.out' with open(inFile, 'r', encoding='utf-8') as inputFile, open(outFile, 'w', encoding='utf-8') as outputFile: for line in inputFile: line = line.strip() result = "yes" if isNip(line) else "no" outputFile.write(result+'\n')