mdbq 1.8.1__py3-none-any.whl → 1.8.2__py3-none-any.whl

mdbq/other/sku_picture.py

@@ -22,11 +22,10 @@ from mdbq.mysql import mysql
 from mdbq.mysql import s_query
 import ua_sj
 import requests
+
+import io
 from openpyxl import load_workbook
-# from openpyxl.drawing.image import Image
-import openpyxl
-from PIL import Image as PILImage
-from io import BytesIO
+from openpyxl.drawing.image import Image
 from openpyxl.utils import get_column_letter
 
 warnings.filterwarnings('ignore')
@@ -571,77 +570,146 @@ def main2(service_name, database):
 
 class InsertPicture():
     def __init__(self):
-        self.file = '/Users/xigua/Downloads/test.xlsx'
-        self.path = '/Users/xigua/Downloads/sku图片链接'
+        self.filename = 'test.xlsx'
+        self.path = '/Users/xigua/Downloads'
         self.pic_datas = []
         self.header = 0  # sku 的标题栏起始行数
 
     def insert_data(self):
         self.get_filename()
-        # sku_in_files = [item['sku'] for item in self.pic_datas]
-        # print(len(sku_in_files))
-
-        # df = pd.read_excel(self.file, header=self.header)
-        # cols = df.columns.tolist()
-        # print(cols)
 
-        workbook = load_workbook(self.file)
+        workbook = load_workbook(os.path.join(self.path, self.filename))
         sheet = workbook.active
         rows = sheet.max_row  # 总行数
         columns = sheet.max_column  # 总列数
-        # print(columns)
-        # print(rows)
         sheet.insert_cols(0, 1)  # 在第0列开始插入1列空白列
-        sheet['A1'] = '商品图片'
+        # sheet['A1'] = '商品图片'
 
-        for col in range(1, columns+1):
-            for row in range(1, rows+1):
+        is_trange = False
+        for col in range(1, columns+1):  # 遍历每一列
+            # if is_trange == True:
+            #     break
+            for row in range(1, rows+1):  # 遍历每一行
                 # print(f'第{col}列, 第{row}行...')
                 value = sheet.cell(row=row, column=col).value
                 if value:
                     for data in self.pic_datas:
-                        if  str(value) == data['sku']:
-                            print(value, data['文件名称'])
+                        if str(value) == data['sku'] or (len(str(value)) > 16 and str(value) in data['sku']):
+                            # print(value, data['sku'])
+                            print(f'转换: 第{col}列, 第{row}行, sku: {data['sku']} ...')
                             image_path = os.path.join(data['文件路径'], data['文件名称'])
-                            img = PILImage.open(image_path)
-                            output = BytesIO()
-                            img.save(output, format='JPEG')
-                            image_data = output.getvalue()
-
-                            # 调整图片大小
-                            img_resized = img.resize((128, 128))  # 调整为128x128大小
-                            output_resized = BytesIO()
-                            img_resized.save(output_resized, format='JPEG')
-                            image_data_resized = output_resized.getvalue()
+                            with open(image_path, 'rb') as f:
+                                img_data = f.read()
+                            img = Image(io.BytesIO(img_data))
+                            width, height = self.img_resize(img.width, img.height)  # 等比例缩放图片
                             col_letter = 'A'
-                            sheet.add_image(openpyxl.drawing.image.Image(sheet, image_data_resized), f'{col_letter}{row}')
-                            break
-
-        # workbook.save(self.file)
-        # print(filenames)
-        # sheet.insert_cols(0, 1)  # 在第0列开始插入1列空白列
-        # sheet['A1'] = '商品图片'
-
-
-        # col_letter = get_column_letter(col)  # 将数字索引转换为列标签 A、B、C、D...
-        # sheet.column_dimensions[col_letter].width = 10
-        # sheet.row_dimensions[row].height = 80
-        # # sheet.cell(row=row, column=col).value = ""  # 删除原内容
-        # now = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S ")
-        # print(f'{now}正在转换: 第{col}列, 第{row}行..')
+                            # col_letter = get_column_letter(col)  # 将数字索引转换为列标签 A、B、C、D...
+                            sheet.column_dimensions[col_letter].width = 13
+                            sheet.row_dimensions[row].height = 80
+                            img.width = width
+                            img.height = height
+                            sheet.add_image(img, f'{col_letter}{row}')
+                            is_trange = True
+
+        if is_trange == False:  # 如果 sku 没有匹配到任何值，则使用 商家编码
+            for col in range(1, columns + 1):  # 遍历每一列
+                # if is_trange == True:
+                #     break
+                for row in range(1, rows + 1):  # 遍历每一行
+                    # print(f'第{col}列, 第{row}行...')
+                    value = sheet.cell(row=row, column=col).value
+                    if value:
+                        for data in self.pic_datas:
+                            if str(value) == data['商家编码']:
+                                # print(value, data['sku'])
+                                print(f'转换: 第{col}列, 第{row}行, 商家编码: {data['商家编码']} ...')
+                                image_path = os.path.join(data['文件路径'], data['文件名称'])
+                                with open(image_path, 'rb') as f:
+                                    img_data = f.read()
+                                img = Image(io.BytesIO(img_data))
+                                width, height = self.img_resize(img.width, img.height)  # 等比例缩放图片
+                                col_letter = 'A'
+                                # col_letter = get_column_letter(col)  # 将数字索引转换为列标签 A、B、C、D...
+                                sheet.column_dimensions[col_letter].width = 13
+                                sheet.row_dimensions[row].height = 80
+                                img.width = width
+                                img.height = height
+                                sheet.add_image(img, f'{col_letter}{row}')
+                                is_trange = True
+                                break  # 商家编码只需要添加一次，所以必须 break，否则可能添加多个图片到某个单元格
+
+        if is_trange == False:  # 如果 sku 和商家编码都没有匹配到任何值，则使用 商品id
+            for col in range(1, columns + 1):  # 遍历每一列
+                # if is_trange == True:
+                #     break
+                for row in range(1, rows + 1):  # 遍历每一行
+                    # print(f'第{col}列, 第{row}行...')
+                    value = sheet.cell(row=row, column=col).value
+                    if value:
+                        for data in self.pic_datas:
+                            if str(value) == data['商品id']:
+                                # print(value, data['sku'])
+                                print(f'转换: 第{col}列, 第{row}行, 商品id: {data['商品id']} ...')
+                                image_path = os.path.join(data['文件路径'], data['文件名称'])
+                                with open(image_path, 'rb') as f:
+                                    img_data = f.read()
+                                img = Image(io.BytesIO(img_data))
+                                width, height = self.img_resize(img.width, img.height)  # 等比例缩放图片
+                                col_letter = 'A'
+                                # col_letter = get_column_letter(col)  # 将数字索引转换为列标签 A、B、C、D...
+                                sheet.column_dimensions[col_letter].width = 13
+                                sheet.row_dimensions[row].height = 80
+                                img.width = width
+                                img.height = height
+                                sheet.add_image(img, f'{col_letter}{row}')
+                                is_trange = True
+                                break  # 商品id只需要添加一次，所以必须 break，否则可能添加多个图片到某个单元格
+
+        if is_trange == False:
+            print(f'{self.filename}:\n'
+                  f'在该文件中没有找到匹配的 skuid/商品id/商家编码, 注意程序只会转换当前活动的 sheet, \n'
+                  f'1. 如果您确定文件中确实存在 skuid/商品id/商家编码, 可能是因为 sheet 不是活动状态, 请切换后再重新运行本程序。\n'
+                  f'2. 程序只能转换已经收录的商品图, 如果未被收录亦会转换失败, 请联系开发者添加。')
+
+        workbook.save(os.path.join(self.path, f'ok_{self.filename}'))
+
+    def img_resize(self, width, height, num=100):
+        """
+        设置基础大小为 num， 并等比例缩放
+        """
+        if width > height:
+            height = height * num // width
+            width = num
+        else:
+            width = width * num // height
+            height = num
+        return width, height
 
     def get_filename(self):
-        for root, dirs, files in os.walk(self.path, topdown=False):
+        for root, dirs, files in os.walk(os.path.join(self.path, 'sku图片链接'), topdown=False):
             for name in files:
                 if name.endswith('.jpg'):
                     sku_id = re.findall(r'\d+_\d+_(\d+)_|\d+_\d+_(\d+-\d+)_|\d+_\d+_([A-Za-z]+\d+)_', name)
                     sku_id = [item for item in sku_id[0] if item != '']
-                    self.pic_datas.append({'文件路径': root, '文件名称': name, 'sku': sku_id[0]})
+                    sp_id = re.findall(r'^(\d+)_', name)
+                    spbm = re.findall(r'(\d{13})\d+', sku_id[0])
+                    if not spbm:
+                        spbm = ['0']
+                    self.pic_datas.append(
+                        {
+                            '文件路径': root,
+                            '文件名称': name,
+                            'sku': sku_id[0],
+                            '商品id': sp_id[0],
+                            '商家编码': spbm[0],
+                        }
+                    )
 
 
 def main3():
     p = InsertPicture()
-    p.header = 1
+    p.filename = 'test222.xlsx'
+    # p.header = 1
     p.insert_data()
 
 

{mdbq-1.8.1.dist-info → mdbq-1.8.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mdbq
-Version: 1.8.1
+Version: 1.8.2
 Home-page: https://pypi.org/project/mdbsql
 Author: xigua, 
 Author-email: 2587125111@qq.com

{mdbq-1.8.1.dist-info → mdbq-1.8.2.dist-info}/RECORD

@@ -30,13 +30,13 @@ mdbq/mysql/year_month_day.py,sha256=VgewoE2pJxK7ErjfviL_SMTN77ki8GVbTUcao3vFUCE,
 mdbq/other/__init__.py,sha256=jso1oHcy6cJEfa7udS_9uO5X6kZLoPBF8l3wCYmr5dM,18
 mdbq/other/porxy.py,sha256=UHfgEyXugogvXgsG68a7QouUCKaohTKKkI4RN-kYSdQ,4961
 mdbq/other/pov_city.py,sha256=AEOmCOzOwyjHi9LLZWPKi6DUuSC-_M163664I52u9qw,21050
-mdbq/other/sku_picture.py,sha256=sf-3i4sfk8kQUZgXMXlg6aiibw2BJF38tVr-ULUi8TI,28937
+mdbq/other/sku_picture.py,sha256=xNXOzyaO-qz1htYroPmYnsct_CXF4MLJASereVEGN6g,33429
 mdbq/other/ua_sj.py,sha256=JuVYzc_5QZ9s_oQSrTHVKkQv4S_7-CWx4oIKOARn_9U,22178
 mdbq/pbix/__init__.py,sha256=Trtfaynu9RjoTyLLYBN2xdRxTvm_zhCniUkVTAYwcjo,24
 mdbq/pbix/pbix_refresh.py,sha256=JUjKW3bNEyoMVfVfo77UhguvS5AWkixvVhDbw4_MHco,2396
 mdbq/pbix/refresh_all.py,sha256=0uAnBKCd5cx5FLTkawN1GV9yi87rfyMgYal5LABtumQ,7186
 mdbq/spider/__init__.py,sha256=RBMFXGy_jd1HXZhngB2T2XTvJqki8P_Fr-pBcwijnew,18
-mdbq-1.8.1.dist-info/METADATA,sha256=4KLPOuQB3iLxcdwbEQ39prUCY6TNJJTnLZTwEbNNlYA,245
-mdbq-1.8.1.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
-mdbq-1.8.1.dist-info/top_level.txt,sha256=2FQ-uLnCSB-OwFiWntzmwosW3X2Xqsg0ewh1axsaylA,5
-mdbq-1.8.1.dist-info/RECORD,,
+mdbq-1.8.2.dist-info/METADATA,sha256=PsbFqqZUVz03tpO86vKRA68rj6Oy3JtEQ2qOUqnauQI,245
+mdbq-1.8.2.dist-info/WHEEL,sha256=cpQTJ5IWu9CdaPViMhC9YzF8gZuS5-vlfoFihTBC86A,91
+mdbq-1.8.2.dist-info/top_level.txt,sha256=2FQ-uLnCSB-OwFiWntzmwosW3X2Xqsg0ewh1axsaylA,5
+mdbq-1.8.2.dist-info/RECORD,,

{mdbq-1.8.1.dist-info → mdbq-1.8.2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.44.0)
+Generator: setuptools (70.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any