add printing version when start server

gxtrobot · Oct 7, 2019 · d3beef0 · d3beef0
1 parent 3b3356e
commit d3beef0
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 6 deletions.
diff --git a/bustag/app/index.py b/bustag/app/index.py
@@ -203,10 +203,12 @@ def start_app():
 if __name__ == "__main__":
     try:
         freeze_support()
+        from bustag import __version__
+        print(f"Bustag server starting: version: {__version__}\n\n")
         import bustag.model.classifier as clf
         from bustag.util import logger, get_cwd, get_now_time, get_data_path
         from bustag.spider.db import (get_items, get_local_items, RATE_TYPE, RATE_VALUE, ItemRate,
-                              Item, LocalItem, DBError, db as dbconn)
+                                      Item, LocalItem, DBError, db as dbconn)
         from bustag.spider import db
         from bustag.app.schedule import start_scheduler, add_download_job
         from bustag.spider import bus_spider
@@ -216,4 +218,4 @@ def start_app():
         print('system error')
     finally:
         print("Press Enter to continue ...")
-        input()
+        input()
diff --git a/bustag/model/prepare.py b/bustag/model/prepare.py
@@ -12,7 +12,7 @@
 from bustag.model.persist import dump_model, load_model
 from bustag.util import logger, get_data_path, MODEL_PATH
 
-BINARIZER_PATH = MODEL_PATH + ' label_binarizer.pkl'
+BINARIZER_PATH = MODEL_PATH + 'label_binarizer.pkl'
 
 
 def load_data():

diff --git a/bustag/util.py b/bustag/util.py
@@ -104,9 +104,10 @@ def to_localtime(utc_dt):
 
 def check_model_folder():
     model_path = os.path.join(DATA_PATH, MODEL_PATH)
-    if not os.path.exists(model_path):
-        logger.info('created model folder')
-        os.mkdir(model_path)
+    abs_path = os.path.abspath(model_path)
+    if not os.path.exists(abs_path):
+        print(f'created model folder: {abs_path}')
+        os.mkdir(abs_path)
 
 
 def init():