def do_auto(self): level_num = input('choose level ') level_num = int(level_num) input_order = [] if level_num is 1: with open('./input1.txt', 'rt') as f: for line in f: line = line.strip('\n') c = line.split(' ') input_order.append(c) elif level_num is 2: with open('./input2.txt', 'rt') as f: for line in f: line = line.strip('\n') c = line.split(' ') input_order.append(c) elif level_num is 3: with open('./input3.txt', 'rt') as f: for line in f: line = line.strip('\n') c = line.split(' ') input_order.append(c) elif level_num is 4: with open('./input4.txt', 'rt') as f: for line in f: line = line.strip('\n') c = line.split(' ') input_order.append(c) elif level_num is 0: with open('./input.txt', 'rt') as f: for line in f: line = line.strip('\n') c = line.split(' ') input_order.append(c) index = 0 output_str = '' for line in input_order: box = Box(line) box.auto_run() output_str += box.output_string() index += 1 if box.time_out: print('%dth puzzle time-out.' % index) else: print('%dth puzzle complete.' % index) counter = 0 flag_counter = 0 for i in output_str.splitlines(): if flag_counter % 2 is 0: counter += len(i) flag_counter += 1 counter = str(counter) output_str += counter output_str += '\n' # sum_time = 0 # for i in output_str.splitlines(): # if 'ms' in i: # m_position = i.index('m') # sum_time += float(i[0:m_position]) # output_str += str(sum_time) # output_str += 'ms\n' output_file(output_str, level_num)