增加小年糕账号数量通知

common/mysql_db.py

@@ -23,16 +23,19 @@ class MysqlHelper:
         return connection
 
     @classmethod
-    def get_values(cls, sql, params):
+    def get_values(cls, sql, params=None):
         try:
             # 连接数据库
             connect = cls.connect_mysql()
             # 返回一个 Cursor对象
             mysql = connect.cursor()
 
-            # 执行 sql 语句
-            mysql.execute(sql, params)
-
+            if params:
+                # 如果传递了 params 参数
+                mysql.execute(sql, params)
+            else:
+                # 如果没有传递 params 参数
+                mysql.execute(sql)
             # fetchall方法返回的是一个元组，里面每个元素也是元组，代表一行记录
             data = mysql.fetchall()
 

xng_zh_job.py

@@ -0,0 +1,28 @@
+import schedule
+import time
+from common import MysqlHelper, Feishu
+
+
+def select_xng_uid_count():
+    sql = """SELECT COUNT(0) FROM xng_uid WHERE DATE(data_time) = CURDATE();"""
+    data = MysqlHelper.get_values(sql)
+    return data
+
+
+
+def fs_bot():
+    count = select_xng_uid_count()
+    value = int(count[0][0])
+
+    text = (
+        f"**今日小年糕账号入库量**: {value} 条\n"
+    )
+    Feishu.finish_bot(text, "https://open.feishu.cn/open-apis/bot/v2/hook/d4862d7e-c97d-456a-85ac-763301c0d958",
+                      "【 小年糕入库数量每日通知 】")
+
+
+# 每天上午十一点
+schedule.every().day.at("11:00").do(fs_bot)
+while True:
+    schedule.run_pending()
+    time.sleep(1)