qrpa 1.1.3__tar.gz → 1.1.5__tar.gz

{qrpa-1.1.3 → qrpa-1.1.5}/PKG-INFO

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

{qrpa-1.1.3 → qrpa-1.1.5}/pyproject.toml

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

{qrpa-1.1.3 → qrpa-1.1.5}/qrpa/shein_excel.py

@@ -15,6 +15,91 @@ class SheinExcel:
         self.bridge = bridge
         pass
 
+    def get_supplier_name(self, store_username):
+        cache_file = f'{self.config.auto_dir}/shein/dict/supplier_data.json'
+        info = read_dict_from_file_ex(cache_file, store_username)
+        return info['supplier_name']
+
+    def write_withdraw_report_2024(self):
+        excel_path = create_file_path(self.config.excel_withdraw_2024)
+        dict_store = read_dict_from_file(self.config.shein_store_alias)
+
+        header = ['店铺名称', '店铺账号', '供应商名称', '交易单号', '提现时间', '提现成功时间', '更新时间', '提现明细单号',
+                  '收款帐户', '收款帐户所在地', '净金额', '保证金', '手续费', '汇率', '收款金额', '提现状态']
+        summary_excel_data = [header]
+        # 先读取提现明细列表写入
+        first_day, last_day = TimeUtils.get_year_range_time(2024)
+        cache_file = f'{self.config.auto_dir}/shein/cache/withdraw_list_{first_day}_{last_day}.json'
+        dict_withdraw = read_dict_from_file(cache_file)
+        account_list = []
+        for store_username, list_withdraw in dict_withdraw.items():
+            store_name = dict_store.get(store_username)
+            supplier_name = self.get_supplier_name(store_username)
+            for withdraw in list_withdraw:
+                row_item = []
+                row_item.append(store_name)
+                row_item.append(store_username)
+                row_item.append(supplier_name)
+                row_item.append(withdraw['withdrawNo'])
+                row_item.append(TimeUtils.convert_timestamp_to_str(withdraw['createTime']))
+                row_item.append(TimeUtils.convert_timestamp_to_str(withdraw.get('transferSuccessTime')))
+                row_item.append(TimeUtils.convert_timestamp_to_str(withdraw['lastUpdateTime']))
+                row_item.append(withdraw['transferNo'])
+                account = withdraw['sourceAccountValue']
+                if account not in account_list:
+                    account_list.append(account)
+                row_item.append(account)
+                row_item.append(withdraw['accountAreaCode'])
+                row_item.append(withdraw['netAmount'])
+                row_item.append(withdraw['depositAmount'])
+                row_item.append(withdraw['commissionAmount'])
+                row_item.append(withdraw['exchangeRate'])
+                row_item.append(withdraw['receivingAmount'])
+                row_item.append(withdraw['withdrawStatusDesc'])
+                summary_excel_data.append(row_item)
+        sheet_name = '提现明细汇总'
+
+        operations = [
+            [sheet_name, 'write', summary_excel_data, ],
+            [sheet_name, 'format', self.format_withdraw_detail]
+        ]
+
+        header = [
+            ['收款账户', '总收款金额'],
+            ['汇总', ''],
+        ]
+        summary_excel_data = header
+        for account in account_list:
+            row_item = []
+            row_item.append(account)
+            row_item.append('')
+            summary_excel_data.append(row_item)
+
+        sheet_name = '汇总2024'
+
+        operations.append([sheet_name, 'write', summary_excel_data])
+        operations.append([sheet_name, 'format', self.format_withdraw_2024])
+        operations.append([sheet_name, 'move', 1])
+        operations.append(['Sheet1', 'delete'])
+
+        batch_excel_operations(excel_path, operations)
+
+    def format_withdraw_detail(self, sheet):
+        beautify_title(sheet)
+        column_to_right(sheet, ['金额'])
+        format_to_money(sheet, ['金额', '保证金', '手续费'])
+        format_to_datetime(sheet, ['时间'])
+        add_borders(sheet)
+
+    def format_withdraw_2024(self, sheet):
+        beautify_title(sheet)
+        column_to_right(sheet, ['金额'])
+        format_to_money(sheet, ['金额'])
+        format_to_datetime(sheet, ['时间'])
+        add_sum_for_cell(sheet, ['总收款金额'])
+        add_formula_for_column(sheet, '总收款金额', "=SUMIF('提现明细汇总'!I:I, A3, '提现明细汇总'!O:O)", 3)
+        add_borders(sheet)
+
     def write_product(self):
         erp = self.config.erp_source
         excel_path = create_file_path(self.config.excel_shein_skc_profit)

{qrpa-1.1.3 → qrpa-1.1.5}/qrpa/shein_lib.py

@@ -281,6 +281,78 @@ class SheinLib:
 
         return info
 
+    # 获取供货商信息
+    def get_supplier_data(self):
+        self.web_page.goto('https://sso.geiwohuo.com/#/mws/seller/new-account-overview')
+        self.web_page.wait_for_load_state('load')
+        cache_file = f'{self.config.auto_dir}/shein/dict/supplier_data.json'
+        info = read_dict_from_file_ex(cache_file, self.store_username, 3600 * 24 * 10)
+        if len(info) > 0:
+            return info
+
+        log(f'正在获取 {self.store_name} 供货商信息')
+        url = "https://sso.geiwohuo.com/mgs-api-prefix/supplierGrowth/querySupplierCommonData"
+        payload = {}
+        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')
+
+        write_dict_to_file_ex(cache_file, {self.store_username: info}, [self.store_username])
+
+        return info
+
+    def get_withdraw_list(self, supplier_id, mode='month'):
+        self.web_page.goto('https://sso.geiwohuo.com/#/mws/seller/new-account-overview')
+        self.web_page.wait_for_load_state("load")
+
+        if mode == 'month':
+            first_day, last_day = TimeUtils.get_last_month_range_time()
+        else:
+            first_day, last_day = TimeUtils.get_year_range_time(2024)
+
+        cache_file = f'{self.config.auto_dir}/shein/cache/withdraw_list_{first_day}_{last_day}.json'
+        withdraw_list = read_dict_from_file_ex(cache_file, self.store_username, 3600 * 12)
+        if len(withdraw_list) > 0:
+            return withdraw_list
+
+        page_num = 1
+        page_size = 200
+
+        url = f"https://sso.geiwohuo.com/mws/mwms/sso/withdraw/transferRecordList"
+        payload = {
+            "reqSystemCode"     : "mws-front",
+            "supplierId"        : supplier_id,
+            "pageNum"           : page_num,
+            "pageSize"          : page_size,
+            "createTimeStart"   : first_day,
+            "createTimeEnd"     : last_day,
+            "withdrawStatusList": [30]
+        }
+        log(payload)
+        response_text = fetch(self.web_page, url, payload)
+        log(response_text)
+        error_code = response_text.get('code')
+        if str(error_code) != '0':
+            raise send_exception(json.dumps(response_text, ensure_ascii=False))
+
+        withdraw_list = response_text['info']['list']
+        total = response_text['info']['count']
+        totalPage = math.ceil(total / page_size)
+
+        for page in range(2, totalPage + 1):
+            log(f'获取提现列表 第{page}/{totalPage}页')
+            page_num = page
+            payload['pageNum'] = page_num
+            response_text = fetch(self.web_page, url, payload)
+            withdraw_list = response_text['info']['list']
+            time.sleep(0.1)
+
+        write_dict_to_file_ex(cache_file, {self.store_username: withdraw_list}, [self.store_username])
+
+        return withdraw_list
+
     # 获取质检报告pdf地址
     def get_qc_report_url(self, deliverCode, purchaseCode):
         log(f'获取质检报告:{deliverCode} {purchaseCode}')

{qrpa-1.1.3 → qrpa-1.1.5}/qrpa/time_utils.py

@@ -358,6 +358,17 @@ class TimeUtils:
 
         return start_timestamp, end_timestamp
 
+    @staticmethod
+    def get_year_range_time(year=2024):
+        """获取某一整年的第一天和最后一天的毫秒级时间戳"""
+        first_day = datetime(year, 1, 1, 0, 0, 0)
+        last_day = datetime(year, 12, 31, 0, 0, 0, 0)
+
+        start_timestamp = int(first_day.timestamp() * 1000)
+        end_timestamp = int(last_day.timestamp() * 1000)
+
+        return start_timestamp, end_timestamp
+
     @staticmethod
     def is_in_month(time_str: str, month: int, fmt: str = "%Y-%m-%d") -> bool:
         """

{qrpa-1.1.3 → qrpa-1.1.5}/qrpa.egg-info/PKG-INFO

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