qrpa 1.1.79__py3-none-any.whl

qrpa/temu_excel.py

@@ -0,0 +1,139 @@
+from .fun_excel import *
+from .fun_base import log
+from .fun_file import read_dict_from_file, read_dict_from_file_ex, write_dict_to_file, write_dict_to_file_ex, delete_file
+from .time_utils import TimeUtils
+from .wxwork import WxWorkBot
+import os
+
+class TemuExcel:
+
+    def __init__(self, config, bridge):
+        self.config = config
+        self.bridge = bridge
+
+    def write_funds(self):
+        cache_file = f'{self.config.auto_dir}/temu/cache/funds_{TimeUtils.today_date()}.json'
+        dict = read_dict_from_file(cache_file)
+        data = []
+        for key, val in dict.items():
+            data.append(val)
+
+        excel_path = create_file_path(self.config.excel_temu_fund)
+        data.insert(0, ['汇总', '', '', '', ''])
+        data.insert(0, ['店铺名称', '总金额', '可用余额', '-', '导出时间'])
+        log(data)
+        # 删除第 4 列（索引为 3）
+        for row in data:
+            row.pop(3)  # 删除每行中索引为 3 的元素
+
+        write_data(excel_path, 'Sheet1', data)
+
+        app, wb, sheet = open_excel(excel_path, 'Sheet1')
+        beautify_title(sheet)
+        format_to_money(sheet, ['金额', '余额'])
+        format_to_datetime(sheet, ['时间'])
+        add_sum_for_cell(sheet, ['总金额', '可用余额'])
+        add_borders(sheet)
+        close_excel(app, wb)
+
+    def format_purchase_advise_batch(self, sheet):
+        beautify_title(sheet)
+        format_to_datetime(sheet, ['时间'])
+        format_to_number(sheet, ['平均日销', '本地和采购可售天数', '建议采购'], 1)
+        add_borders(sheet)
+        add_formula_for_column(sheet, '平均日销', '=G2/7')
+        add_formula_for_column(sheet, '本地和采购可售天数', '=IF(H2>0,(E2+F2)/H2,0)')
+        add_formula_for_column(sheet, '建议采购', '=IF(J2>I2,H2*9,0)')
+        colorize_by_field(sheet, 'SKC')
+        autofit_column(sheet, ['店铺名称', '商品信息'])
+        column_to_left(sheet, ['商品信息'])
+        InsertImageV2(sheet, ['SKC图片', 'SKU图片'], 'temu', 120)
+        # if sheet.used_range.rows.count > 330:
+        #     log('表格数据行数超过了330行,将删除SKC图片')
+        #     remove_excel_columns(sheet, ['SKC图片'])
+
+    def write_purchase_advise(self, erp='mb'):
+        cache_file = f'{self.config.auto_dir}/temu/cache/warehouse_list_{TimeUtils.today_date()}.json'
+        dict = read_dict_from_file(cache_file)
+
+        store_info = read_dict_from_file(self.config.temu_store_info)
+
+        header = ['店铺名称', 'SKC图片', 'SKU图片', '商品信息', '现有库存数量', '已采购数量', '近7日销量', '平均日销', '本地和采购可售天数', '生产天数', '建议采购', '产品起定量', '备货周期(天)', 'SKC', '导出时间']
+        new_excel_path_list = []
+        
+        for mall_id, subOrderList in dict.items():
+            excel_data = []
+            mall_name = store_info.get(mall_id)[1]
+            
+            for product in subOrderList:
+                spu = str(product['productId'])  # temu平台 spu_id
+                skc = str(product['productSkcId'])  # temu平台 skc_id
+                skcExtCode = product['skcExtCode']  # 商家 SKC货号
+                category = product['category']  # 叶子类目
+                onSalesDurationOffline = product['onSalesDurationOffline']  # 加入站点时长
+                
+                for sku in product['skuQuantityDetailList']:
+                    priceReviewStatus = sku['priceReviewStatus']
+                    if priceReviewStatus == 3:  # 过滤 开款价格状态 已作废的  2是已生效
+                        continue
+                    
+                    mall_info = f'{mall_name}\n{mall_id}'
+                    productSkcPicture = product['productSkcPicture']  # skc图片
+                    skuExtCode = str(sku['skuExtCode'])  # sku货号
+                    sku_img = self.bridge.get_sku_img(skuExtCode, erp)
+                    stock = self.bridge.get_sku_stock(skuExtCode, erp)
+
+                    product_info = f"SPU: {spu}\nSKC: {skc}\nSKC货号: {skcExtCode}\nSKU货号: {skuExtCode}\n属性集: {sku['className']}\n类目: {category}\n加入站点时长: {onSalesDurationOffline}天\n"
+
+                    row_item = []
+                    row_item.append(mall_info)
+                    row_item.append(productSkcPicture)
+                    row_item.append(sku_img)
+                    row_item.append(product_info)
+                    row_item.append(stock)
+                    row_item.append(0)
+                    row_item.append(sku['lastSevenDaysSaleVolume'])
+                    row_item.append(0)
+                    row_item.append(0)
+                    row_item.append(7)
+                    row_item.append(0)
+                    row_item.append(0)
+                    row_item.append(0)
+                    row_item.append(skc)
+                    row_item.append(TimeUtils.current_datetime())
+                    excel_data.append(row_item)
+
+            # 按近7日销量排序
+            excel_data = sort_by_column(excel_data, 6, 1)
+            
+            # 计算需要多少个文件（每个文件最多320行数据，包含表头）
+            max_data_rows = 250 - 1  # 减去表头行
+            total_files = (len(excel_data) + max_data_rows - 1) // max_data_rows  # 通过加(max_data_rows-1)实现向上取整
+            
+            for file_index in range(total_files):
+                start_idx = file_index * max_data_rows
+                end_idx = min((file_index + 1) * max_data_rows, len(excel_data))
+                current_data = excel_data[start_idx:end_idx]
+                
+                # 生成文件名，如果超过一个文件则添加序号
+                if total_files == 1:
+                    new_excel_path = str(self.config.excel_purcase_advice_temu).replace('#store_name#', mall_name).replace(' ', '_')
+                else:
+                    # 在文件名后添加 _2, _3 等序号
+                    base_path = str(self.config.excel_purcase_advice_temu).replace('#store_name#', mall_name).replace(' ', '_')
+                    file_name, file_ext = os.path.splitext(base_path)
+                    new_excel_path = f"{file_name}_{file_index + 1}{file_ext}"
+                
+                new_excel_path_list.append(new_excel_path)
+                sheet_name = 'Sheet1'
+                data = [header] + current_data
+                
+                close_excel_file(new_excel_path)
+                log(f"创建文件: {new_excel_path}, 数据行数: {len(current_data)}")
+                
+                batch_excel_operations(new_excel_path, [
+                    (sheet_name, 'write', data, ['N']),
+                    (sheet_name, 'format', self.format_purchase_advise_batch)
+                ])
+
+        return new_excel_path_list

qrpa/temu_lib.py

@@ -0,0 +1,156 @@
+from .fun_file import read_dict_from_file, write_dict_to_file, read_dict_from_file_ex, write_dict_to_file_ex
+from .fun_base import log, send_exception, md5_string, get_safe_value
+from .time_utils import TimeUtils
+
+import json, requests, time, math
+
+class TemuLib:
+    def __init__(self, config, mobile, password, web_page):
+        self.config = config
+        self.web_page = web_page
+        self.mobile = mobile
+
+        self.dict_mall = {}
+        self.cookie = self.doLoginToTemu(mobile, password)
+
+    # 主账户登录 返回Cookie
+    def doLoginToTemu(self, username, password):
+        cache_cookie = f'{self.config.auto_dir}/temu/cookie/cookie_{username}.json'
+        dict_cookie = read_dict_from_file(cache_cookie, 1)
+        if len(dict_cookie) > 0:
+            self.cookie = dict_cookie.get('cookie')
+            return
+
+        log(f'登录Temu账号: {username}')
+
+        """使用 XPath 登录网站"""
+        # 导航到登录页面
+        self.web_page.goto("https://seller.kuajingmaihuo.com/login")
+        # fixed
+        self.web_page.locator('//div[text()="账号登录"]').click()
+        # 输入用户名
+        self.web_page.locator('//input[@id="usernameId"]').fill("")
+        self.web_page.locator('//input[@id="usernameId"]').fill(username)
+        # 输入密码
+        self.web_page.locator('//input[@id="passwordId"]').fill("")
+        self.web_page.locator('//input[@id="passwordId"]').fill(password)
+        # 勾选隐私政策（checkbox）
+        self.web_page.locator('//input[@type="checkbox"]/following-sibling::div').click()  # 直接check不了 换成点击
+        # 点击登录按钮
+        self.web_page.locator('//button[span[text()="登录"]]').click()
+        # 等待登录完成（根据页面加载情况调整等待策略）
+        self.web_page.wait_for_load_state("load")
+
+        while True:
+            log('等待卖家中心出现')
+            try:
+                if self.web_page.locator('//div[text()="Temu商家中心"]').count() == 1:
+                    log('卖家中心已出现')
+                    break
+                if self.web_page.locator('//div[text()="Seller Central"]').count() == 1:
+                    log('卖家中心已出现')
+                    break
+            except Exception as e:
+                log(f"❌{e}")
+            time.sleep(1.5)
+
+        log("✅ 登录成功")
+
+        self.web_page.wait_for_load_state("load")
+        self.web_page.wait_for_timeout(3000)
+        cookies = self.web_page.context.cookies()
+        cookies_list = [cookie for cookie in cookies if '.kuajingmaihuo.com' in cookie['domain']]
+        self.cookie = '; '.join([f"{cookie['name']}={cookie['value']}" for cookie in cookies_list])
+        log(f'已获取self.cookie:', self.cookie)
+        write_dict_to_file(cache_cookie, {'cookie': self.cookie})
+        return self.cookie
+
+    def post_json(self, str_url, payload, mall_id=None):
+        global response
+        try:
+            headers = {
+                'content-type': 'application/json',
+                'priority'    : 'u=1, i',
+                'referer'     : 'https://seller.kuajingmaihuo.com/settle/site-main',
+                'user-agent'  : 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/133.0.0.0 Safari/537.36',
+                'Cookie'      : self.cookie,
+            }
+            if mall_id:
+                headers.update({'mallid': f"{mall_id}"})
+            response = requests.post(str_url, headers=headers, data=json.dumps(payload))
+            response.raise_for_status()  # 如果响应不正常，会抛出异常
+
+            response_text = response.json()
+            error_code = response_text.get('error_code') or response_text.get('errorCode')
+            if error_code != 1000000:
+                raise send_exception(response_text)
+            return response.json()  # 直接返回 JSON 格式的数据
+        except:
+            raise send_exception()
+
+    def get_shop_list(self):
+        log(f'获取店铺列表')
+        global DictMall
+        url = "https://seller.kuajingmaihuo.com/bg/quiet/api/mms/userInfo"
+        response_text = self.post_json(url, {})
+
+        company_list = response_text['result']['companyList']
+        mall_list = []
+        for company in company_list:
+            mallList = company['malInfoList']
+            # shop_list = [['店铺ID', '店铺名称', '主账号', '店铺类型']]
+            for mall in mallList:
+                mall_id = str(mall['mallId'])
+                mall_name = str(mall['mallName'])
+                shop_info = [mall_id, mall_name, self.mobile, '半托管' if mall['isSemiManagedMall'] else '全托管']
+                write_dict_to_file_ex(self.config.temu_store_info, {mall_id: shop_info}, [mall_id])
+
+                self.dict_mall[str(mall['mallId'])] = mall['mallName']
+
+                if not mall['isSemiManagedMall']:
+                    mall_list.append([str(mall['mallId']), mall['mallName'], '', self.mobile])
+
+        return mall_list
+
+    def get_funds_info(self, mall_id):
+        log(f'获取 {self.dict_mall[mall_id]} 资金信息')
+        url = "https://seller.kuajingmaihuo.com/api/merchant/payment/account/amount/info"
+        response_text = self.post_json(url, {}, mall_id)
+        total_amount = response_text.get('result').get('totalAmount')
+        available_amount = response_text.get('result').get('availableBalance')
+
+        NotifyItem = [self.dict_mall[mall_id], total_amount, available_amount, '', TimeUtils.current_datetime()]
+
+        cache_file = f'{self.config.auto_dir}/temu/cache/funds_{TimeUtils.today_date()}.json'
+        write_dict_to_file_ex(cache_file, {mall_id: NotifyItem}, [mall_id])
+
+        return NotifyItem
+
+    def list_warehouse(self, mall_id, mall_name):
+        log(f'获取店铺 {mall_name} 销售商品列表 第1页')
+        url = "https://seller.kuajingmaihuo.com/marvel-mms/cn/api/kiana/venom/sales/management/listWarehouse"
+        payload = {
+            "pageNo"               : 1,
+            "pageSize"             : 40,
+            "isLack"               : 0,
+            "selectStatusList"     : [12],  # 12 是已加入站点
+            "priceAdjustRecentDays": 30  # 近30日价格调整
+        }
+        response_text = self.post_json(url, payload, mall_id)
+
+        total = response_text['result']['total']
+        subOrderListCount = len(response_text['result']['subOrderList'])
+        totalPage = math.ceil(total / subOrderListCount) if subOrderListCount else 0
+        subOrderList = response_text['result']['subOrderList']
+
+        for page in range(2, totalPage + 1):
+            log(f'获取店铺{mall_name}销售商品列表 第{page}/{totalPage}页')
+            payload['pageNo'] = page
+            response_text = self.post_json(url, payload, mall_id)
+            subOrderList += response_text['result']['subOrderList']
+            time.sleep(0.3)
+
+        cache_file = f'{self.config.auto_dir}/temu/cache/warehouse_list_{TimeUtils.today_date()}.json'
+        write_dict_to_file_ex(cache_file, {mall_id: subOrderList}, [mall_id])
+
+        return subOrderList