def main(argv=sys.argv[1:]):
"""main entry point mini-project-2"""
parser = get_parser()
args = parser.parse_args(argv)
sys.argv = []
# configure logging
handlers_ = []
log_format = Formatter(fmt="[%(asctime)s] [%(levelname)s] - %(message)s")
if args.log_dir:
os.makedirs(args.log_dir, exist_ok=True)
file_handler = TimedRotatingFileHandler(
os.path.join(args.log_dir, "mini_project_2.log"),
when="d",
interval=1,
backupCount=7,
encoding="UTF-8",
)
file_handler.setFormatter(log_format)
file_handler.setLevel(args.log_level)
handlers_.append(file_handler)
if args.verbose:
stream_handler = logging.StreamHandler(stream=sys.stderr)
stream_handler.setFormatter(log_format)
stream_handler.setLevel(args.log_level)
handlers_.append(stream_handler)
basicConfig(handlers=handlers_, level=args.log_level)
if args.phase == 1:
generate_data_files(args.input_file)
if args.phase == 2:
phase2()
if args.phase == 3:
phase3()
if args.phase < 1 or args.phase > 3:
return 1
return 0
def test_generate_terms_files():
generate_data_files(small_input_file)
assert open(small_terms_file).read() == open(terms_file).read()
def test_generate_data_files(): # todo: kinda slow; speed it up somehow?
generate_data_files(med_input_file)
assert open(med_ads_file).read() == open(ads_file).read()
assert open(med_prices_file).read() == open(prices_file).read()
assert open(med_terms_file).read() == open(terms_file).read()
assert open(med_pdates_file).read() == open(pdates_file).read()
def test_generate_prices_files():
generate_data_files(small_input_file)
assert open(small_prices_file).read() == open(prices_file).read()