from query_builder import QueryBuilder
except ImportError:
from modules.query_builder import QueryBuilder
print("Testing QueryBuilder...")
print(" -def push_bool_operation()")
################################################################################
print(" Testing push_bool_operation's error-finding...")
tests = {"DROP TABLES":False, "a":False, "DROP CARDS":False, "SELECT":False,
"1|6":False,"0|1":True,"(0|1)":True,"0":True,"{":False, "0&1":True,
"0|1 DROP CARDS":False,"0&(1|2)":True}
num_tests = len(tests)
failures = []
for test in tests.keys():
result = QueryBuilder.push_bool_operation(test) == tests[test]
if not result:
failures.append(test)
print(str(len(failures)) + '/' + str(len(tests)) + ' tests FAILED:')
for failure in failures:
print('FAILURE: ', failure)
print('EXPECTED:', tests[failure])
print('ACTUAL: ', not(tests[failure]))
################################################################################
#Resetting
QueryBuilder.bool_operation = None
print(" Testing push_bool_operation's converting correctly")
tests = {"0|1":"{0} OR {1}","0&1":"{0} AND {1}","0&(1|2)":"{0} AND ({1} OR {2})"}
def parse_args():
#arg_name:db_col_name
#usage: -power ">5"
#usage: -toughness 5
int_keys = {"power":"POWER","toughness":"TOUGHNESS"
,"p":"POWER","t":"TOUGHNESS","cmc":"CMC"}
#usage: -name REGEXPATTERN
re_keys = {"name":"NAME","text":"CARD_TEXT"}
#usage:
#-type -i type_a type_b -x type_c
#include types a, types b, exclude type c
list_keys = {"type":"ALL_TYPES","legal":"LEGALITIES"
,"colorid":"COLOR_IDENTITY","printings":"PRINTINGS","format":"LEGALITIES"}
misc_keys = ["-print","-help",",","-bool","-debug","-sort","-random"]
valid_print_options = ['count_bare','custom']
valid_cols = ['CMC', 'COLOR_IDENTITY', 'COLORS', 'LEGALITIES', 'LOYALTY',
'MANA_COST', 'NAME', 'POWER', 'PRINTINGS', 'RULINGS', 'CARD_TEXT',
'TOUGHNESS', 'TYPE', 'ALL_TYPES']
i = 1
arr_len = len(sys.argv)
while i < arr_len:
if sys.argv[i][0] != "-" and sys.argv[i] != ",":
#SET THIS CRITERIA NAME
qb.add_regex("NAME",sys.argv[i])
i += 1
continue
if sys.argv[i][1:] in re_keys.keys():
col = re_keys[sys.argv[i][1:]]
i += 1
qb.add_regex(col,sys.argv[i])
i += 1
continue
if sys.argv[i][1:] in int_keys.keys():
col = int_keys[sys.argv[i][1:]]
i += 1
try:
int(sys.argv[i])
val = sys.argv[i]
qb.add_equality_operation(col,val,"==")
i += 1
continue
except ValueError:
if sys.argv[i][1] != "=" or sys.argv[i][1] != ">" or sys.argv[i][1] != "<":
qb.add_equality_operation(col,str(int(sys.argv[i][1:])),sys.argv[i][0])
i += 1
continue
qb.add_equality_operation(col,str(int(sys.argv[i][1:])),sys.argv[i][0:2])
i += 1
continue
#TODO: look into the sqlite3's adapter features for Python <-> sqlite data
if sys.argv[i][1:] in list_keys.keys():
arg = list_keys[sys.argv[i][1:]]
i += 1
inclusions = []
exclusions = []
if sys.argv[i] != '-x':
inclusions.append(sys.argv[i])
else:
i += 1
exclusions.append(sys.argv[i])
qb.add_list(arg,inclusions,exclusions)
i += 1
continue
#IF WE HIT HERE, WE'VE HIT MISCELANNY OPTIONS
if sys.argv[i] in misc_keys:
#PROCESS...
if sys.argv[i] == "-print":
i += 1
if sys.argv[i] not in valid_print_options:
print("BAD ARGUMENT FOR -print:")
print(sys.argv[i])
sys.exit(1)
qb.print_setting = sys.argv[i]
if sys.argv[i] == 'custom':
i += 1
qb.custom_print_str = open(sys.argv[i],'r').read()
if sys.argv[i] == '-random':
try:
qb.randoms_to_get = int(sys.argv[i + 1])
i += 1
except ValueError:
qb.randoms_to_get = 1
except IndexError:
qb.randoms_to_get = 1
if sys.argv[i] == '-debug':
i += 1
qb.debug_options[sys.argv[i]] = True
if sys.argv[i] == '-bool':
i += 1
qb.push_bool_operation(sys.argv[i])
if sys.argv[i] == "-sort":
i += 1
if sys.argv[i][0] != 'a' and sys.argv[i][0] != 'd':
print("-sort arg must begin with a or d")
print(sys.argv[i])
sys.exit(3)
s = sys.argv[i][1:]
if s not in valid_cols:
print("Invalid column name:")
print(s)
sys.exit(3)
if sys.argv[i][0] == 'a':
qb.sort_cols.append(s + ' ASC')
else:
qb.sort_cols.append(s + ' DESC')
elif sys.argv[i] == ',':
qb.combine_this_criteria()
i += 1
continue
print("Argument not recognized:")
print(sys.argv[i])
sys.exit(1)