no message

jobs/__init__.py

@@ -6,9 +6,11 @@ from apscheduler.triggers.cron import CronTrigger
 from datetime import datetime, timedelta
 from utils import *
 from models import *
+from pprint import pprint
 from .yzy_job import proc as yzy_proc, yzy_msg_queue_proc
 from .rainfall_conditions_job import proc as rainfall_proc
 from .avcon_job import proc as avcon_proc
+from .duty_job import proc as duty_proc
 
 def register_jobs(scheduler: BaseScheduler):
     scheduler.add_job(yzy_proc, next_run_time=(datetime.now() + timedelta(seconds=3)))
@@ -25,5 +27,9 @@ def register_jobs(scheduler: BaseScheduler):
     # scheduler.add_job(avcon_proc, next_run_time=(datetime.now() + timedelta(seconds=15)))
     scheduler.add_job(avcon_proc, CronTrigger.from_crontab('0 0 * * *'))
 
+    # 值班提醒推送
+    scheduler.add_job(duty_proc, next_run_time=(datetime.now() + timedelta(seconds=3)))
+    scheduler.add_job(duty_proc, CronTrigger.from_crontab('0 0 * * *'))
+
 def tick():
     print(datetime.now())

jobs/duty_job.py

@@ -0,0 +1,84 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+from datetime import datetime, date
+from sqlalchemy.orm import Session
+from sqlalchemy import text, exists, and_, or_, not_
+from utils import *
+from utils.redis_util import *
+from models import *
+from exceptions import *
+from database import get_db_local
+from extensions import logger
+from common import YzyApi
+from config import settings
+import traceback
+from pprint import pprint
+from common.db import db_user, db_yzy, db_msg_center
+from config import settings
+
+# 值班任务
+# 每天00:00将当天值班人员推送一次
+def proc():
+    lock_key = "duty_job_proc"
+    if redis_lock(lock_key):
+        logger.info(datetime.now())
+
+        db = get_db_local()
+        __duty_proc(db)
+
+        db.close()
+        redis_unlock(lock_key)
+
+
+def format_time(time_diff) -> str:
+    hours, minutes, seconds = str(time_diff).split(':')
+    return str(hours).zfill(2) + ":" + str(minutes).zfill(2)
+
+def __duty_proc(db: Session):
+    theDay = date.today()
+    rows = db.query(DutyShift).filter(and_(DutyShift.shift_date == theDay, DutyShift.del_flag == '0')).all()
+    
+    for row in rows:
+        start_time = format_time(row.start_time)
+        end_time = format_time(row.end_time)
+        
+        to_users = [row.leader_id, row.primary_staff_id, row.secondary_staff_id, row.standby_staff_id]
+        for to_user_id in to_users:
+            if(is_not_sent(db, to_user_id, theDay)):
+                send_yzy_msg(db, to_user_id, start_time, end_time, str(row.shift_id))
+
+def is_not_sent(db: Session, to_user_id: str, theDay: DateTime):
+    user_info = db_user.get_user_info(db, to_user_id)
+    yzy_account = user_info.yzy_account
+    
+    where = and_(YzyMsgQueue.mobile == yzy_account, 
+                 YzyMsgQueue.from_scenario == 'duty_shift',
+                 YzyMsgQueue.create_time >= theDay,
+                 YzyMsgQueue.create_time < theDay + timedelta(days=1),
+                )
+    row = db.query(YzyMsgQueue).filter(where).first()
+    return row is None
+
+# 发送粤政易消息
+def send_yzy_msg(db: Session, to_user_id: int, start_time: str, end_time: str, foreign_key: str) -> None:
+    user_info = db_user.get_user_info(db, to_user_id)
+    yzy_account = user_info.yzy_account
+
+    yzy_userid = db_yzy.get_userid_by_account(db, yzy_account)
+    detail_url = "{}{}".format(settings.YZY_WEB_ROOT, "/yjxp/#/worker/duty")
+
+    description = f"您今天的值班时间为{start_time}-{end_time}，请准时到岗。"
+    logger.info("{}, {}", yzy_account, description)
+    data = {
+        "yzy_userid": yzy_userid,
+        "mobile": yzy_account,
+        "content": description,
+        "recorded_by": 0,
+        "detail_url": detail_url,
+        "foreign_key": foreign_key,
+        "from_scenario": "duty_shift",
+        "title": "值班提醒"
+    }
+    YzyApi.add_to_msg_queue(db, data)
+    db_msg_center.add_msg(db, "值班提醒", foreign_key, to_user_id)
+