def init_db(cookies): if cookies is not None: c.check_type(cookies, 'cookies', str) # login in with cookies file if cookies: author = ZhihuClient(cookies=cookies).me() # login in terminal else: client = ZhihuClient() try: cookies = client.login_in_terminal() except KeyboardInterrupt: print() cookies = '' if not cookies: L.error(s.log_login_failed) L.info(s.exit) exit(0) author = client.me() try: conn = db.create_db(author) db.create_table(conn) db.dump_init_data_to_db(conn, author) db.close_db(conn) print(s.success) except FileExistsError as e: L.error(s.file_exist.format(e.filename)) print(s.failed)
if os.path.exists(TEST_DIR): print("Cleaning it...", end='') shutil.rmtree(TEST_DIR) print("Done") else: print("Test dir not exist.") os.chdir(BASE_DIR) if os.path.isfile(Cookies_File): print("Cookies file found.") client = ZhihuClient(Cookies_File) else: print("Cookies file not exist, please login...") client = ZhihuClient() cookies_str = client.login_in_terminal() with open(Cookies_File, 'w') as f: f.write(cookies_str) print("Making test dir...", end="") os.mkdir(TEST_DIR) print("Done", end="\n\n") os.chdir(TEST_DIR) print("===== test start =====") import timeit try: time = timeit.timeit('test()',
if os.path.exists(TEST_DIR): print("Cleaning it...", end='') shutil.rmtree(TEST_DIR) print("Done") else: print("Test dir not exist.") os.chdir(BASE_DIR) if os.path.isfile(Cookies_File): print("Cookies file found.") client = ZhihuClient(Cookies_File) else: print("Cookies file not exist, please login...") client = ZhihuClient() cookies_str = client.login_in_terminal() with open(Cookies_File, 'w') as f: f.write(cookies_str) print("Making test dir...", end="") os.mkdir(TEST_DIR) print("Done", end="\n\n") os.chdir(TEST_DIR) print("===== test start =====") import timeit try: time = timeit.timeit(