mdbq 1.8.0__tar.gz → 1.8.1__tar.gz

{mdbq-1.8.0 → mdbq-1.8.1}/PKG-INFO

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

{mdbq-1.8.0 → mdbq-1.8.1}/mdbq/other/sku_picture.py

@@ -5,6 +5,7 @@ import json
 import os
 import platform
 import random
+from dateutil.relativedelta import relativedelta
 import re
 import time
 import warnings
@@ -18,6 +19,15 @@ from selenium.webdriver.chrome.service import Service
 from mdbq.config import set_support
 from mdbq.config import get_myconf
 from mdbq.mysql import mysql
+from mdbq.mysql import s_query
+import ua_sj
+import requests
+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.utils import get_column_letter
 
 warnings.filterwarnings('ignore')
 
@@ -444,16 +454,86 @@ class SkuPicture:
 
         if results:
             self.df = pd.concat(results)
+            self.df = self.df[['sku图片链接'] != '0']
 
     def read_df(self):
         path = os.path.join(self.path, self.filename)
         df = pd.read_excel(path, header=0)
         df = df[['商品id', '商家编码', '是否新增']]
+        df['是否新增'].fillna(0, inplace=True)
         df = df.astype({'是否新增': int})
         df = df[df['是否新增'] == 1]
         self.urls = df.to_dict('records')
 
 
+class DownloadPicture():
+    """
+    从数据库中下载数据
+    """
+    def __init__(self, service_name):
+        # target_service 从哪个服务器下载数据
+        self.months = 0  # 下载几个月数据, 0 表示当月, 1 是上月 1 号至今
+        # 实例化一个下载类
+        username, password, host, port = get_myconf.select_config_values(target_service=service_name,
+                                                                         database='mysql')
+        self.download = s_query.QueryDatas(username=username, password=password, host=host, port=port)
+        self.df = pd.DataFrame()
+        self.headers = {'User-Agent': ua_sj.get_ua()}
+        self.save_path = '/Users/xigua/Downloads/sku图片链接'
+        self.filename = ''
+        if not os.path.exists(self.save_path):
+            os.mkdir(self.save_path)
+
+    def get_df_from_service(self):
+        start_date, end_date = self.months_data(num=self.months)
+        projection = {
+            '商品id': 1,
+            '商家编码': 1,
+            'sku编码': 1,
+            'sku名称': 1,
+            'sku图片链接': 1
+        }
+        self.df = self.download.data_to_df(
+            db_name='属性设置2',
+            table_name='天猫商品sku信息',
+            start_date=start_date,
+            end_date=end_date,
+            projection=projection,
+        )
+
+    def download_data(self):
+        dict_data = self.df.to_dict('records')
+        num = len(dict_data)
+        i = 0
+        for data in dict_data:
+            url = data['sku图片链接']
+            sku_name = re.sub('/', '_', data['sku名称'])
+            self.filename = f'{data['商品id']}_{data['商家编码']}_{data['sku编码']}_{sku_name}.jpg'
+            if os.path.isfile(os.path.join(self.save_path, self.filename)):
+                i += 1
+                continue
+            if 'https' not in url:
+                i += 1
+                continue
+
+            print(f'正在下载: {i}/{num},   {data['sku编码']}')
+            self.headers.update({'User-Agent': ua_sj.get_ua()})
+            res = requests.get(url, headers=self.headers)  # 下载图片到内存
+            # 保存图片到本地文件夹
+            with open(os.path.join(self.save_path, self.filename), 'wb') as f:
+                f.write(res.content)
+            i += 1
+            time.sleep(0.5)
+
+    @staticmethod
+    def months_data(num=0, end_date=None):
+        """ 读取近 num 个月的数据, 0 表示读取当月的数据 """
+        if not end_date:
+            end_date = datetime.datetime.now()
+        start_date = end_date - relativedelta(months=num)  # n 月以前的今天
+        start_date = f'{start_date.year}-{start_date.month}-01'  # 替换为 n 月以前的第一天
+        return pd.to_datetime(start_date), pd.to_datetime(end_date)
+
 def main(service_name, database):
     if not os.path.exists(Share_Path):
         print(f'当前系统环境不支持')
@@ -483,5 +563,90 @@ def main(service_name, database):
         )  # 3. 回传数据库
 
 
+def main2(service_name, database):
+    d = DownloadPicture(service_name=service_name)
+    d.get_df_from_service()
+    d.download_data()
+
+
+class InsertPicture():
+    def __init__(self):
+        self.file = '/Users/xigua/Downloads/test.xlsx'
+        self.path = '/Users/xigua/Downloads/sku图片链接'
+        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)
+        sheet = workbook.active
+        rows = sheet.max_row  # 总行数
+        columns = sheet.max_column  # 总列数
+        # print(columns)
+        # print(rows)
+        sheet.insert_cols(0, 1)  # 在第0列开始插入1列空白列
+        sheet['A1'] = '商品图片'
+
+        for col in range(1, columns+1):
+            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['文件名称'])
+                            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()
+                            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}行..')
+
+    def get_filename(self):
+        for root, dirs, files in os.walk(self.path, 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]})
+
+
+def main3():
+    p = InsertPicture()
+    p.header = 1
+    p.insert_data()
+
+
+
 if __name__ == '__main__':
-    main(service_name='company', database='mysql')
+    # main(service_name='company', database='mysql')
+    # main2(service_name='company', database='mysql')
+    main3()

{mdbq-1.8.0 → mdbq-1.8.1}/mdbq.egg-info/PKG-INFO

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

{mdbq-1.8.0 → mdbq-1.8.1}/setup.py

@@ -3,7 +3,7 @@
 from setuptools import setup, find_packages
 
 setup(name='mdbq',
-      version='1.8.0',
+      version='1.8.1',
       author='xigua, ',
       author_email="2587125111@qq.com",
       url='https://pypi.org/project/mdbsql',