init

app.py

@@ -6,11 +6,10 @@ import shutil
 import os
 import datetime
 from apscheduler.schedulers.background import BackgroundScheduler
-from database.database import DATABASE_FILE as DB_FILE
-from web import demo
 from loguru import logger
 from common.util import date_str
 
+DB_FILE = "app.db"
 TOKEN = os.environ.get('HUB_TOKEN')
 repo = huggingface_hub.Repository(
     local_dir="data",
@@ -25,14 +24,15 @@ shutil.copyfile(DATASET_FILE, DB_FILE)
 
 def backup_db():
     shutil.copyfile(DB_FILE, DATASET_FILE)
-    # db = sqlite3.connect(DB_FILE)
-    # pd.DataFrame(db.execute("SELECT * FROM words").fetchall()).to_csv("./data/words.csv", index=False)
-    # print("save word.csv")
-    # pd.DataFrame(db.execute("SELECT * FROM book").fetchall()).to_csv("./data/book.csv", index=False)
-    # print("save book.csv")
-    # pd.DataFrame(db.execute("SELECT * FROM unit").fetchall()).to_csv("./data/unit.csv", index=False)
-    # print("save unit.csv")
-    # db.close()
+    logger.info(f"save database at {DB_FILE}")
+    db = sqlite3.connect(DB_FILE)
+    pd.DataFrame(db.execute("SELECT * FROM words").fetchall()).to_csv("./data/words.csv", index=False)
+    logger.info("save word.csv")
+    pd.DataFrame(db.execute("SELECT * FROM book").fetchall()).to_csv("./data/book.csv", index=False)
+    logger.info("save book.csv")
+    pd.DataFrame(db.execute("SELECT * FROM unit").fetchall()).to_csv("./data/unit.csv", index=False)
+    logger.info("save unit.csv")
+    db.close()
     repo.push_to_hub(blocking=False, commit_message=f"Updating data at {datetime.datetime.now()}")
 
 
@@ -50,4 +50,5 @@ scheduler.start()
 
 if __name__ == "__main__":
     logger.add(f"output/logs/web_{date_str}.log", rotation="1 day", retention="7 days", level="INFO")
+    from web import demo
     demo.launch()