qrpa 1.0.15__py3-none-any.whl → 1.0.17__py3-none-any.whl

qrpa/shein_sqlite.py

@@ -0,0 +1,154 @@
+from .fun_base import log
+
+import sqlite3
+from datetime import datetime, timedelta
+import os
+import sys
+
+
+auto_dir = 'D:/auto'
+db_file = f'{auto_dir}/shein/db/shein_sku_sales.db'
+
+# log(db_file)
+
+
+def main(args):
+    init_db()
+    # exists_sales_1_days_ago('653231597')
+    # log(get_last_week_sales('6960380466'))
+    # log(get_sales('6960380466', '2025-02-25', '2025-03-04'))
+    # log(get_near_week_sales('I46mraado10r'))
+    # log(get_last_week_sales('I46mraado10r','2025-03-08','2025-03-14'))
+    # create_indexes()
+
+def init_db():
+    # 获取文件夹路径
+    folder_path = os.path.dirname(db_file)
+    # 如果文件夹不存在，则创建
+    if not os.path.exists(folder_path):
+        os.makedirs(folder_path)
+        log(f"文件夹已创建: {folder_path}")
+    conn = sqlite3.connect(db_file)
+    cursor = conn.cursor()
+    cursor.execute('''
+        CREATE TABLE IF NOT EXISTS sales (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            skc TEXT,
+            date TEXT,
+            skc_sale INTEGER,
+            skc_order INTEGER,
+            sku TEXT,
+            attr_name TEXT,
+            sku_sale INTEGER,
+            sku_order INTEGER
+        )
+    ''')
+    conn.commit()
+    conn.close()
+    log('新建数据库成功', db_file)
+
+init_db()
+
+def insert_sales(skc, date, skc_sale, skc_order, sku, attr_name, sku_sale, sku_order):
+    conn = sqlite3.connect(db_file)
+    cursor = conn.cursor()
+    cursor.execute('''
+        SELECT 1 FROM sales WHERE sku = ? AND date = ?
+    ''', (sku, date))
+
+    if cursor.fetchone() is None:
+        cursor.execute('''
+            INSERT INTO sales (skc, date, skc_sale, skc_order, sku, attr_name, sku_sale, sku_order)
+            VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+        ''', (skc, date, skc_sale, skc_order, sku, attr_name, sku_sale, sku_order))
+        conn.commit()
+
+    conn.close()
+
+def get_sales(sku, start_date, end_date):
+    conn = sqlite3.connect(db_file)
+    cursor = conn.cursor()
+    cursor.execute('''
+        SELECT SUM(sku_sale), SUM(sku_order) FROM sales WHERE sku = ? AND date BETWEEN ? AND ?
+    ''', (sku, start_date, end_date))
+
+    result = cursor.fetchone()  # 获取查询结果
+    sales_num = result[0] if result[0] is not None else 0
+    orders_num = result[1] if result[1] is not None else 0
+
+    conn.close()
+
+    log('get_sales:', sku, start_date, end_date, sales_num, orders_num)
+    return sales_num, orders_num  # 返回销售量和订单量的元组
+
+def get_last_week_sales(sku):
+    today = datetime.today().date()
+    # last_week_start = today - timedelta(days=today.weekday() + 14)
+    last_week_start = (datetime.now() - timedelta(days=14)).strftime("%Y-%m-%d")
+    last_week_end = (datetime.now() - timedelta(days=8)).strftime("%Y-%m-%d")
+    # log('远7天',last_week_start,last_week_end,sku)
+    return get_sales(sku, str(last_week_start), str(last_week_end))
+
+def get_near_week_sales(sku):
+    today = datetime.today().date()
+    # last_week_start = today - timedelta(days=today.weekday() + 14)
+    last_week_start = (datetime.now() - timedelta(days=7)).strftime("%Y-%m-%d")
+    last_week_end = (datetime.now() - timedelta(days=1)).strftime("%Y-%m-%d")
+    # log('远7天',last_week_start,last_week_end,sku)
+    return get_sales(sku, str(last_week_start), str(last_week_end))
+
+def get_last_month_sales(sku):
+    today = datetime.today().date()
+    # first_day_of_this_month = today.replace(day=1)
+    # last_month_end = first_day_of_this_month - timedelta(days=1)
+    # last_month_start = last_month_end.replace(day=1)
+    last_month_start = (datetime.now() - timedelta(days=60)).strftime("%Y-%m-%d")
+    last_month_end = (datetime.now() - timedelta(days=31)).strftime("%Y-%m-%d")
+    # log('远30天',last_month_start,last_month_end,sku)
+    return get_sales(sku, str(last_month_start), str(last_month_end))
+
+def get_near_month_sales(sku):
+    today = datetime.today().date()
+    # first_day_of_this_month = today.replace(day=1)
+    # last_month_end = first_day_of_this_month - timedelta(days=1)
+    # last_month_start = last_month_end.replace(day=1)
+    last_month_start = (datetime.now() - timedelta(days=30)).strftime("%Y-%m-%d")
+    last_month_end = (datetime.now() - timedelta(days=1)).strftime("%Y-%m-%d")
+    # log('远30天',last_month_start,last_month_end,sku)
+    return get_sales(sku, str(last_month_start), str(last_month_end))
+
+def exists_sales_1_days_ago(skc):
+    conn = sqlite3.connect(db_file)  # 替换成你的数据库文件
+    cursor = conn.cursor()
+    date_threshold = (datetime.today().date() - timedelta(days=1)).isoformat()
+    # 使用 str.format() 格式化 SQL
+    sql = '''
+        SELECT 1 FROM sales WHERE skc = '{skc}' AND date = '{date}' LIMIT 1
+    '''.format(skc=skc, date=date_threshold)
+    log("exists_sales_1_days_ago:", sql.strip())  # 打印 SQL 语句
+    cursor.execute('''
+        SELECT 1 FROM sales WHERE skc = ? AND date <= ? LIMIT 1
+    ''', (skc, date_threshold))
+    result = cursor.fetchone()
+    conn.close()
+    return result is not None
+
+def create_indexes():
+    conn = sqlite3.connect(db_file)
+    cursor = conn.cursor()
+    # 创建 spu + date 联合索引
+    cursor.execute('''
+        CREATE INDEX IF NOT EXISTS idx_sales_skc_date ON sales (skc, date);
+    ''')
+    # 创建 sku + date 唯一索引
+    cursor.execute('''
+        CREATE UNIQUE INDEX IF NOT EXISTS idx_sales_sku_date_unique ON sales (sku, date);
+    ''')
+    conn.commit()  # 提交更改
+    conn.close()  # 关闭连接
+    log("索引创建成功！")
+
+init_db()
+
+if __name__ == '__main__':
+    main(sys.argv)

qrpa/wxwork.py

@@ -17,6 +17,7 @@ import hashlib
 import base64
 import requests
 from requests_toolbelt import MultipartEncoder
+from datetime import datetime
 
 # 通过企微群机器人发送消息
 class WxWorkBot:
@@ -99,6 +100,42 @@ class WxWorkBot:
         }
         self.send_msg(data)
 
+    def send_notify(self, title, sub_title_list, data_list):
+        """
+        发送Markdown消息
+        :param content:
+        :return:
+        """
+
+        current_date = datetime.now().strftime("%Y-%m-%d")
+        header = f"{current_date} {title}\n\n"
+
+        arr_color = ['warning', 'info', 'warning']
+        arr_sub_header = [f"<font color='{arr_color[index]}'>{title}</font>" for index, title in enumerate(sub_title_list)]
+        sub_header = "\t".join(arr_sub_header) + "\n\n"
+
+        # 获取每个元素的行索引和列索引
+        arr_content = [
+            [
+                f'{value}' if col_idx == 0 else f"<font color='{arr_color[col_idx - 1]}'>{value}</font>"
+                for col_idx, value in enumerate(row)
+            ]  # 每行的元素组成一个子列表
+            for row_idx, row in enumerate(data_list)  # 外层循环控制行
+        ]
+        # 将二维数组转换为字符串
+        content = "\n".join(
+            # 对每行的元素列表使用 join()，用 \t 连接
+            "\t".join(row) for row in arr_content
+        )
+
+        data = {
+            "msgtype" : "markdown",
+            "markdown": {
+                "content": header + sub_header + content
+            }
+        }
+        self.send_msg(data)
+
     def send_img(self, img_path):
         """
         发送图片消息

{qrpa-1.0.15.dist-info → qrpa-1.0.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: qrpa
-Version: 1.0.15
+Version: 1.0.17
 Summary: qsir's rpa library
 Author: QSir
 Author-email: QSir <1171725650@qq.com>

{qrpa-1.0.15.dist-info → qrpa-1.0.17.dist-info}/RECORD

@@ -1,18 +1,19 @@
 qrpa/RateLimitedSender.py,sha256=hqvb1qspDFaW4RsLuVufylOrefkMgixANKeBaGEqYb4,1421
-qrpa/__init__.py,sha256=NNE1S-k2R-5Ed_ssVVu-R6_DJ00zBWuPnOvyJAU6DOI,706
+qrpa/__init__.py,sha256=EysO3QHdHWPWuVmtbekAZBSKSlCIAHnLzg2O6e26nNs,801
 qrpa/db_migrator.py,sha256=2VmhzcMsU0MKpl-mNCwKyV8tLTqyEysSpP27-S_rQZ8,21862
 qrpa/fun_base.py,sha256=W_owEa8-yuGG18n9kX3remm9YTzym69ztQjtYCNMTw4,3308
 qrpa/fun_excel.py,sha256=S-A-kTZ2e3dJ8NqdhiOFqTlcr0d6XLMP6imagztRel0,103605
-qrpa/fun_file.py,sha256=BInN-Iuxi8sYiJ031gXI_DzO_n170RsOf7fnpkl_etM,7100
+qrpa/fun_file.py,sha256=yzjDV16WL5vRys7J4uQcNzIFkX4D5MAlSCwxcD-mwQo,11966
 qrpa/fun_web.py,sha256=5QLQorAhRzMOGMRh4eCJ2UH8ZhVHvxkHwobWhmgU5qM,6286
 qrpa/fun_win.py,sha256=-LnTeocdTt72NVH6VgLdpAT9_C5oV9okeudXG6CftMA,8034
-qrpa/shein_excel.py,sha256=6DjK5p1S6y4oElcxmLpvJtQPSf76Vu99Pdsmo2d-Po0,3818
-qrpa/shein_lib.py,sha256=aEN0Z9hOjF0XLIoPAHtbIleVmG9NlPsf1XPseOgFBiM,39099
+qrpa/shein_excel.py,sha256=4dYhyvfphhcaje7rJUJR7GMZDWwm8VMw49BWf8T8f8E,13455
+qrpa/shein_lib.py,sha256=VOKtgUK-2stWLwASjQRfiYHkVG7M2Tzk0VV-XNyXkes,65915
+qrpa/shein_sqlite.py,sha256=ZQwD0Gz81q9WY7tY2HMEYvSF9r3N_G_Aur3bYfST9WY,5707
 qrpa/shein_ziniao.py,sha256=nSqqcEPh4nVQtUxUnIRzeZfTLyXywGPjPZn5pP-w57U,18309
 qrpa/time_utils.py,sha256=ef0hhbN_6b-gcnz5ETIVOoxemIMvcxGVGGIRnHnGaBo,29564
 qrpa/time_utils_example.py,sha256=shHOXKKF3QSzb0SHsNc34M61wEkkLuM30U9X1THKNS8,8053
-qrpa/wxwork.py,sha256=IeJq-1LwKCsCt_AdMSLvYrQJWNoQp_wqVtk6iqwJqf4,9200
-qrpa-1.0.15.dist-info/METADATA,sha256=OnrU6Dk4AtMPz-WSJsjaAIBeb52W300aLT8U0cFC_5g,231
-qrpa-1.0.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-qrpa-1.0.15.dist-info/top_level.txt,sha256=F6T5igi0fhXDucPPUbmmSC0qFCDEsH5eVijfVF48OFU,5
-qrpa-1.0.15.dist-info/RECORD,,
+qrpa/wxwork.py,sha256=8LzRmoHYo0KBfZ1cmkWPj824Zp52NAUYm3RXYzmIIi0,10507
+qrpa-1.0.17.dist-info/METADATA,sha256=KltOu2nc9ZIPMTGxfUoAzr5p14zr_eY-OoyyagV2tXM,231
+qrpa-1.0.17.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+qrpa-1.0.17.dist-info/top_level.txt,sha256=F6T5igi0fhXDucPPUbmmSC0qFCDEsH5eVijfVF48OFU,5
+qrpa-1.0.17.dist-info/RECORD,,