import sys import re def write_answer(answer): with open(output_file, 'a') as file: file.write(answer+'\n') # input_file = 'test.in' # output_file = 'test.out' input_file = sys.argv[1] output_file = sys.argv[2] pattern = re.compile(r'^\d{3}-(\d{3}-\d{2}-\d{2}|\d{2}-\d{2}-\d{3})$') with open(output_file, 'w') as readed_output_file: with open(input_file, 'r') as readed_input_file: for row_input_file in readed_input_file: if pattern.match(row_input_file): write_answer('yes') else: write_answer('no')