From c402a1d65a38064e1de25aacac3a9bdaa3ae9a99 Mon Sep 17 00:00:00 2001 From: simplypower-bbj Date: Sun, 18 Dec 2022 20:51:17 +0100 Subject: [PATCH] updated code to use less logins to database --- code/show_trans_progress.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/code/show_trans_progress.py b/code/show_trans_progress.py index 26dd690..d3dc96d 100644 --- a/code/show_trans_progress.py +++ b/code/show_trans_progress.py @@ -4,12 +4,12 @@ import time from dotenv import load_dotenv import pandas as pd -def get_progress(con): +def get_progress(dt): query = 'SELECT count(*) from data_table' - res = pd.read_sql_query(query, con) + res = pd.read_sql_query(query, dt.con) total = res['count'][0] query = 'SELECT count(*) from data_table where content_en is not null' - res = pd.read_sql_query(query, con) + res = pd.read_sql_query(query, dt.con) progress = res['count'][0] return progress, total @@ -18,13 +18,13 @@ if __name__ == '__main__': try: with Data_table() as dt: # continously update progress bar - progress, total = get_progress(dt.con) + progress, total = get_progress(dt) with tqdm(total=total, initial=progress) as pbar: while True: pbar.n = progress pbar.refresh() time.sleep(1) - progress, total = get_progress(dt.con) + progress, total = get_progress(dt) except Exception as e: print(f'error occurred: {e}') except KeyboardInterrupt: