添加监测站点库存更新情况，添加支持命令行运行

添加监测站点库存更新情况，添加支持命令行运行，支持仅监测自定义分类的商品

Author: sqvktfet

README.md

@@ -1,5 +1,5 @@
 # ddmc.monitor
-叮咚买菜运力监控。监控叮咚站点首页信息和购物车预约时间信息，通过Bark app通知到手机。
+叮咚买菜运力监控，站点库存监控。监控叮咚站点首页信息购物车预约时间信息及站点库存存在日间随机更新的情况，通过Bark app通知到手机。
 
 ## 使用说明
 
@@ -9,7 +9,7 @@
 
 ### 运行
 
-本脚本提供了两种监控方式：
+本脚本提供了三种监控方式，修改 run_type 切换：
 
 1. **站点首页关键字监控**(免登陆，不会封号但可能封ip，默认使用)
     - 抓包获取站点id，填写**station_id**即可
@@ -18,6 +18,9 @@
 2. **购物车预约时间监控**(需要登录，有封号风险，谨慎使用)
     - 需要填写UA等信息作为header
     - 然后`raw_body`填Fiddler里面请求信息的Raw tab最下面的一串url格式的字符串(类似`uid=xxx&longitude=xxx`，购物车点结算-预约时间就有了)
+3. **自己站点菜品库存和更新提醒**
+    - 需要填写 **station_id，longitude, latitude**
+    - 可以自己从 name_of_all_categories 中挑选并添加或修改自己关心的分类的菜品到 name_of_categories_i_care 中。
 
 执行main.py即可
 

config.py

@@ -18,3 +18,12 @@
 api_version = '9.49.2'  # ddmc-api-version
 build_version = '2.82.0'    # ddmc-build-version
 raw_body = ''
+
+name_of_all_categories = [
+    "预制菜", "蔬菜豆制品", "肉禽蛋", "水产海鲜", "水果鲜花", "叮咚特供", "乳品烘焙", "速食冻品", "粮油调味", "酒水饮料", "火锅到家", "熟食卤味",
+    "休闲零食", "日用百货", "方便食品", "营养早餐", "宝妈严选", "轻养星球", "在家烧烤", "云仓快送"
+]
+name_of_categories_i_care = [
+    '预制菜', '蔬菜豆制品', '肉禽蛋', '水产海鲜', '水果鲜花', '叮咚特供', '乳品烘焙', '速食冻品',
+    "火锅到家", "熟食卤味", "休闲零食", "日用百货", "方便食品", "营养早餐"
+]

get_categories.py

@@ -3,7 +3,6 @@
 
 import requests
 import config
-import extract_categories
 
 
 def send_msg_bark(msg):
@@ -88,7 +87,6 @@ def get_menu_with_category_id(category_id):
         r.encoding = 'utf-8'
         res = r.json()
         menu = {'sub_category_name': res['data']['category_name'], 'products': []}
-        # menu['count'] = 0
         if 'cate' in res['data']:
             foods = []
             for sub_category in res['data']['cate']:
@@ -97,7 +95,6 @@ def get_menu_with_category_id(category_id):
                     for product in sub_category['products']:
                         foods.append(product['name'])
                 menu['products'] = foods
-                # menu['count'] = len(foods)
         return menu
 
 

main.py

@@ -1,8 +1,8 @@
 import requests
 import time
+import sys
 import config
-
-bark_msg_url = 'https://api.day.app/' + config.bark_id + '/'
+from get_categories import check_stock, send_msg_bark
 
 
 # 检查主页公告 无需cookie
@@ -25,7 +25,7 @@ def check_home():
                 print('还没有运力！', time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()))
             else:
                 txt = '叮咚买菜有运力啦!!!'
-                requests.get(bark_msg_url + txt, params=params)
+                send_msg_bark(txt)
 
 
 # 检查购物车 需要cookie
@@ -65,7 +65,7 @@ def check_cart():
                 all_full = all_full and reserve_time['fullFlag']
             if not all_full:
                 txt = '叮咚买菜可以预约啦!!!最早可预约时间：' + reserve_time['select_msg']
-                requests.get(bark_msg_url + txt, params=params)
+                send_msg_bark(txt)
             else:
                 print('还没有可预约时间!')
         else:
@@ -76,10 +76,17 @@ def run():
     while True:
         if config.run_type == 1:
             check_home()
-        else:
+        elif config.run_type == 2:
             check_cart()
+        else:
+            check_stock()
         time.sleep(config.duration)
 
 
 if __name__ == '__main__':
+    # 给在服务器后台执行使用
+    if len(sys.argv) > 1:
+        run_type = int(sys.argv[1])
+        if run_type in [1, 2, 3]:
+            config.run_type = run_type
     run()