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)
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(
