libushang 1 月之前
父節點
當前提交
06b5b7123c
共有 2 個文件被更改,包括 14 次插入1 次删除
  1. 10 0
      database.py
  2. 4 1
      jobs/__init__.py

+ 10 - 0
database.py

@@ -4,6 +4,7 @@ from sqlalchemy import create_engine
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import sessionmaker
 from config import settings
+from contextlib import contextmanager
 
 mysql_dwd_config = {
     'drivername': 'mysql+pymysql',
@@ -32,6 +33,15 @@ def get_db():
     finally:
         db.close()
 
+# 适用scheduler
+@contextmanager
+def get_local_db():
+    try:
+        db = SessionLocal()
+        yield db
+    finally:
+        db.close()
+
 def get_db_local():
     return SessionLocal()
 

+ 4 - 1
jobs/__init__.py

@@ -14,9 +14,12 @@ from .duty_job import proc as duty_proc
 from .vehicle_job import proc as vehicle_proc
 from .hkvideo_job import proc as hkvideo_proc
 from common.security import encrypt_password
+from common.TassApi import *
 
 def register_jobs(scheduler: BaseScheduler):
-    print('encrypt_password:', encrypt_password('!Aa123456'))
+    encrptData = TransparentEnc("将原数据按密评要求加密辅助类")
+    print("TransparentDec:", TransparentDec(encrptData))
+
     # scheduler.add_job(yzy_proc, next_run_time=(datetime.now() + timedelta(seconds=3)))
     # scheduler.add_job(yzy_proc, CronTrigger.from_crontab('0 */5 * * *'))
     # scheduler.add_job(yzy_msg_queue_proc, CronTrigger.from_crontab('* * * * *'))