qrpa 1.0.81__tar.gz → 1.0.83__tar.gz

{qrpa-1.0.81 → qrpa-1.0.83}/PKG-INFO

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

{qrpa-1.0.81 → qrpa-1.0.83}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "qrpa"
-version = "1.0.81"
+version = "1.0.83"
 description = "qsir's rpa library"
 authors = [{ name = "QSir", email = "1171725650@qq.com" }]
 readme = "README.md"

{qrpa-1.0.81 → qrpa-1.0.83}/qrpa/shein_lib.py

@@ -629,6 +629,19 @@ class SheinLib:
         log(f'正在获取 {self.store_name} 最近一个月出库金额: {last_item["totalCustomerAmount"]}')
         return last_item['totalCustomerAmount']
 
+    def query_attribute_multi(self, attribute_id_list):
+        url = "https://sso.geiwohuo.com/spmp-api-prefix/spmp/attribute/query_attribute_multi"
+        payload = {
+            "attribute_id_list": attribute_id_list,
+        }
+        response_text = fetch(self.web_page, url, payload)
+        error_code = response_text.get('code')
+        if str(error_code) != '0':
+            raise send_exception(json.dumps(response_text, ensure_ascii=False))
+        info = response_text.get('info')
+        lst = info.get('data', {})
+        return lst
+
     def get_product_attr(self, spu, attr_name):
         try:
             product_detail = self.get_product_detail(spu)
@@ -706,6 +719,14 @@ class SheinLib:
         if str(error_code) != '0':
             raise send_exception(json.dumps(response_text, ensure_ascii=False))
         info = response_text.get('info')
+
+        # 获取 area_attribute_id
+        sample_sku_back_size = info.get('sample_sku_back_size', None)
+        if sample_sku_back_size is not None:
+            area_attribute_ids = [item['area_attribute_id'] for item in sample_sku_back_size.get('area_info_list', [])]
+            attribute_multi = self.query_attribute_multi(area_attribute_ids)
+            info["attribute_multi"] = attribute_multi
+
         write_dict_to_file(cache_file, info)
         return info
 

{qrpa-1.0.81 → qrpa-1.0.83}/qrpa/shein_mysql.py

@@ -1,54 +1,62 @@
-import json
-
-from .mysql_module.shein_return_order_model import SheinReturnOrderManager
-from .mysql_module.shein_product_model import SheinProductManager
-from .fun_base import log
-
-class SheinMysql:
-    def __init__(self, config):
-        self.config = config
-
-    def upsert_shein_return_order(self, json_file):
-        log(f'当前使用的数据库: {self.config.db.database_url}')
-        # 创建管理器实例
-        manager = SheinReturnOrderManager(self.config.db.database_url)
-        # 创建数据表
-        manager.create_tables()
-        # 读取JSON文件
-        with open(json_file, 'r', encoding='utf-8') as f:
-            dict = json.load(f)
-            for store_username, data_list in dict.items():
-                manager.upsert_return_order_data(store_username, data_list)
-
-    def upsert_shein_product(self, json_file):
-        log(f'当前使用的数据库: {self.config.db.database_url}')
-        # 创建管理器实例
-        manager = SheinProductManager(self.config.db.database_url)
-        # 创建数据表
-        manager.create_tables()
-        with open(json_file, 'r', encoding='utf-8') as f:
-            file_list = json.load(f)
-            for store_username, store_skc_list_file in file_list.items():
-                with open(store_skc_list_file, 'r', encoding='utf-8') as f:
-                    dict_store_skc_list = json.load(f)
-                    for store_username, data_list in dict_store_skc_list.items():
-                        manager.upsert_product_data(data_list)
-
-    def upsert_shein_product_info(self, json_file):
-        log(f'当前使用的数据库: {self.config.db.database_url}')
-        # 创建管理器实例
-        manager = SheinProductManager(self.config.db.database_url)
-        # 创建数据表
-        manager.create_tables()
-        with open(json_file, 'r', encoding='utf-8') as f:
-            file_list = json.load(f)
-            for store_username, store_spu_list in file_list.items():
-                for spu in store_spu_list:
-                    product_detail_file = f'{self.config.auto_dir}/shein/product_detail/product_detail_{spu}.json'
-                    attribute_file = f'{self.config.auto_dir}/shein/attribute/attribute_template_{spu}.json'
-                    with open(product_detail_file, 'r', encoding='utf-8') as f:
-                        data_list = json.load(f)
-                        manager.upsert_product_detail(spu, 'product_detail', data_list)
-                    with open(attribute_file, 'r', encoding='utf-8') as f:
-                        data_list = json.load(f)
-                        manager.upsert_product_detail(spu, 'attribute_template', data_list)
+import json
+
+from .mysql_module.shein_return_order_model import SheinReturnOrderManager
+from .mysql_module.shein_product_model import SheinProductManager
+from .fun_base import log
+
+import os
+
+class SheinMysql:
+    def __init__(self, config):
+        self.config = config
+
+    def upsert_shein_return_order(self, json_file):
+        log(f'当前使用的数据库: {self.config.db.database_url}')
+        # 创建管理器实例
+        manager = SheinReturnOrderManager(self.config.db.database_url)
+        # 创建数据表
+        manager.create_tables()
+        # 读取JSON文件
+        with open(json_file, 'r', encoding='utf-8') as f:
+            dict = json.load(f)
+            for store_username, data_list in dict.items():
+                manager.upsert_return_order_data(store_username, data_list)
+
+    def upsert_shein_product(self, json_file):
+        log(f'当前使用的数据库: {self.config.db.database_url}')
+        # 创建管理器实例
+        manager = SheinProductManager(self.config.db.database_url)
+        # 创建数据表
+        manager.create_tables()
+        with open(json_file, 'r', encoding='utf-8') as f:
+            file_list = json.load(f)
+            for store_username, store_skc_list_file in file_list.items():
+                with open(store_skc_list_file, 'r', encoding='utf-8') as f:
+                    dict_store_skc_list = json.load(f)
+                    for store_username, data_list in dict_store_skc_list.items():
+                        manager.upsert_product_data(data_list)
+
+    def upsert_shein_product_info(self, json_file):
+        log(f'当前使用的数据库: {self.config.db.database_url}')
+        # 创建管理器实例
+        manager = SheinProductManager(self.config.db.database_url)
+        # 创建数据表
+        manager.create_tables()
+        with open(json_file, 'r', encoding='utf-8') as f:
+            file_list = json.load(f)
+            for store_username, store_spu_list in file_list.items():
+                for spu in store_spu_list:
+                    product_detail_file = f'{self.config.auto_dir}/shein/product_detail/product_detail_{spu}.json'
+                    attribute_file = f'{self.config.auto_dir}/shein/attribute/attribute_template_{spu}.json'
+                    if os.path.exists(product_detail_file):
+                        with open(product_detail_file, 'r', encoding='utf-8') as f:
+                            data_list = json.load(f)
+                            manager.upsert_product_detail(spu, 'product_detail', data_list)
+                    else:
+                        log(f'文件不存在: {product_detail_file}')
+                    if os.path.exists(attribute_file):
+                        with open(attribute_file, 'r', encoding='utf-8') as f:
+                            data_list = json.load(f)
+                            manager.upsert_product_detail(spu, 'attribute_template', data_list)
+                    else:
+                        log(f'文件不存在: {attribute_file}')

{qrpa-1.0.81 → qrpa-1.0.83}/qrpa.egg-info/PKG-INFO

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