class_tree = raw_input('Class (-1 no class): ') class_tree = tuple([int(e) for e in class_tree.split(',')]) except: print 'N, K, K_extend, W must be integers > 0' continue print '(n, k, k_extend, w) = (%s,%s,%s,%s)' % (n, k, k_extend, w) sleep(2) Problem(n, k).calculate_tree(n, k_extend, k, w, class_tree) if option == 32: try: class_to_extend = raw_input('Class to extend: ') class_to_extend = tuple( [int(e) for e in class_to_extend.split(',')]) n_of_the_class = input('Select N: ') extend_from = input('Select original extended k: ') extend_to = input('Select K to extend: ') except: print 'N, K, K_extend, W must be integers > 0' continue Functions().calculate_extended_classes(class_to_extend, n_of_the_class, extend_from, extend_to, True, {}, {}, {}) elif option == 5: print 'Exit' exit_program = True elif option == 400: print 'dropping database...' sleep(2) db = DB() db.drop_database()
k_extend = input('Select K_extend: ') w = input('Select W: ') class_tree = raw_input('Class (-1 no class): ') class_tree = tuple([int(e) for e in class_tree.split(',')]) except: print 'N, K, K_extend, W must be integers > 0' continue print '(n, k, k_extend, w) = (%s,%s,%s,%s)' % (n, k, k_extend, w) sleep(2) Problem(n, k).calculate_tree(n,k_extend,k,w, class_tree) if option == 32: try: class_to_extend = raw_input('Class to extend: ') class_to_extend = tuple([int(e) for e in class_to_extend.split(',')]) n_of_the_class = input('Select N: ') extend_from = input('Select original extended k: ') extend_to = input('Select K to extend: ') except: print 'N, K, K_extend, W must be integers > 0' continue Functions().calculate_extended_classes(class_to_extend, n_of_the_class, extend_from, extend_to, True, {}, {}, {}) elif option == 5: print 'Exit' exit_program = True elif option == 400: print 'dropping database...' sleep(2) db = DB() db.drop_database()