crawl add sql file

crossborder/anhui/crawl_gov_anhui_full.py

@@ -17,6 +17,8 @@ from crossborder.anhui import gov_commodity_anhui_city, download_dir
 from crossborder.anhui import gov_commodity_anhui_country
 from crossborder.anhui import gov_commodity_anhui_import_export
 from crossborder.utils import base_country_code, base_mysql
+from crossborder.utils.base_country_code import get_last_month
+from crossborder.utils.base_mysql import get_commodity_trade_by_prov_year_month
 from crossborder.utils.dingtalk import send_dingtalk_message
 from crossborder.utils.log import  get_logger
 
@@ -339,6 +341,13 @@ def main():
             if res == 'finish':
                 duration = time.time() - start_time
                 send_dingtalk_message(f'【安徽省海关】增量数据采集完成，耗时 {duration:.2f} 秒')
+
+                res = get_commodity_trade_by_prov_year_month('安徽省', get_last_month())
+                if res is not None:
+                    send_dingtalk_message(f"【安徽省海关】 commodity_trade 查询到 {len(res)} 条记录，文件已生成")
+                else:
+                    send_dingtalk_message("【安徽省海关】 未查询到任何记录或发生错误")
+
     except Exception as e:
         send_dingtalk_message(f'【安徽省海关】发生错误：{e}')
 

crossborder/hebei/crawl_gov_hebei_full.py

@@ -18,6 +18,8 @@ from crossborder.hebei import gov_commodity_hebei_city
 from crossborder.hebei import gov_commodity_hebei_country
 from crossborder.hebei import gov_commodity_hebei_import_export
 from crossborder.utils import base_country_code, base_mysql
+from crossborder.utils.base_country_code import get_last_month
+from crossborder.utils.base_mysql import get_commodity_trade_by_prov_year_month
 from crossborder.utils.dingtalk import send_dingtalk_message
 from crossborder.utils.log import  get_logger
 
@@ -317,6 +319,13 @@ def main():
             if res == 'finish':
                 duration = time.time() - start_time
                 send_dingtalk_message(f'【河北省海关】增量数据采集完成，耗时 {duration:.2f} 秒')
+
+                res = get_commodity_trade_by_prov_year_month('河北省', get_last_month())
+                if res is not None:
+                    send_dingtalk_message(f"【河北省海关】 commodity_trade 查询到 {len(res)} 条记录，文件已生成")
+                else:
+                    send_dingtalk_message("【河北省海关】 未查询到任何记录或发生错误")
+
     except Exception as e:
         send_dingtalk_message(f"【河北省海关】发生错误：{e}")
 

crossborder/jiangsu/crawl_gov_jiangsu_full.py

@@ -23,6 +23,8 @@ from crossborder.jiangsu import gov_commodity_jiangsu_city
 from crossborder.jiangsu import gov_commodity_jiangsu_import_export
 
 from crossborder.utils import base_country_code, base_mysql
+from crossborder.utils.base_country_code import get_last_month
+from crossborder.utils.base_mysql import get_commodity_trade_by_prov_year_month
 from crossborder.utils.dingtalk import send_dingtalk_message
 from crossborder.utils.log import  get_logger
 
@@ -354,6 +356,12 @@ def main():
             if res == 'finish':
                 duration = time.time() - start_time
                 send_dingtalk_message(f'【江苏省海关】增量数据采集完成，耗时 {duration:.2f} 秒')
+
+                res = get_commodity_trade_by_prov_year_month('江苏省', get_last_month())
+                if res is not None:
+                    send_dingtalk_message(f"【江苏省海关】 commodity_trade 查询到 {len(res)} 条记录，文件已生成")
+                else:
+                    send_dingtalk_message("【江苏省海关】 未查询到任何记录或发生错误")
     except Exception as e:
         send_dingtalk_message(f"【江苏省海关】发生错误：{e}")
 

crossborder/utils/base_country_code.py

@@ -1,10 +1,8 @@
-import os
 import re
-from pathlib import Path
-
+from datetime import datetime
 import pandas as pd
 
-from crossborder.utils.log import  get_logger
+from crossborder.utils.log import get_logger
 
 log = get_logger(__name__)
 
@@ -330,4 +328,19 @@ def extract_year_month_from_path(path):
             raise ValueError(f"无效月份格式：{month_part}")
         return int(year_part), int(month_part)
     except IndexError:
-        raise ValueError("路径结构不符合要求，示例：.../shandong/2025/04")
+        raise ValueError("路径结构不符合要求，示例：.../shandong/2025/04")
+
+def get_last_month():
+    # 获取当前时间
+    today = datetime.today()
+
+    # 计算上个月
+    if today.month == 1:
+        # 如果是1月，则上个月是去年12月
+        last_month = today.replace(year=today.year - 1, month=12)
+    else:
+        # 否则，直接减去一个月
+        last_month = today.replace(month=today.month - 1)
+
+    # 返回格式为 yyyy-MM 的字符串
+    return last_month.strftime('%Y-%m')

crossborder/utils/base_mysql.py

@@ -336,18 +336,74 @@ def _update_shandong_new_yoy_origin(region_name):
         log.info(f"{region_name} 新数据更新数: {result.rowcount}")
         return result.rowcount
 
+def generate_sql_file(results, prov_name, year_month):
+    filename = f"export_{prov_name}_{year_month}.sql"
+
+    with open(filename, 'w', encoding='utf-8') as f:
+        for row in results:
+            # 过滤掉不需要的字段（如 id）
+            filtered_row = {k: v for k, v in row.items() if k != 'id'}
+
+            # 构建INSERT语句
+            columns = ', '.join([f"`{k}`" for k in filtered_row.keys()])
+            values = ', '.join([
+                f"'{v}'" if isinstance(v, (str,)) else str(v) if v is not None else 'NULL'
+                for v in filtered_row.values()
+            ])
+
+            sql = f"INSERT INTO `t_yujin_crossborder_prov_commodity_trade` ({columns}) VALUES ({values});\n"
+            f.write(sql)
+
+    log.info(f"成功生成SQL文件：{filename}")
+
+def get_commodity_trade_by_prov_year_month(prov_name, year_month):
+    connection = None
+    try:
+        # 连接数据库
+        db_config = DB_CONFIG.copy()
+        db_config['password'] = get_decrypted_password()
+        connection = pymysql.connect(**db_config)
+
+        # 使用 DictCursor 来获取字典形式的结果
+        with connection.cursor(pymysql.cursors.DictCursor) as cursor:
+            # 执行查询
+            sql = "SELECT * FROM t_yujin_crossborder_prov_commodity_trade WHERE prov_name = %s AND crossborder_year_month = %s"
+            cursor.execute(sql, (prov_name, year_month))
+
+            # 获取结果
+            results = cursor.fetchall()
+
+            # 生成SQL文件
+            if results:
+                generate_sql_file(results, prov_name, year_month)
+
+            return results
+
+    except Exception as e:
+        log.error(f"get_commodity_trade_by_prov_year_month error: {str(e)}")
+        return None
+    finally:
+        if connection:
+            connection.close()
+
 if __name__ == '__main__':
-    commodity_code, commodity_name_fix = get_commodity_id('农产品')
-    print(commodity_code, commodity_name_fix)
+    res = get_commodity_trade_by_prov_year_month('江苏省','2025-04')
+    if res is not None:
+        print(f"查询到 {len(res)} 条记录")
+    else:
+        print("未查询到任何记录或发生错误")
+
+    # commodity_code, commodity_name_fix = get_commodity_id('农产品')
+    # print(commodity_code, commodity_name_fix)
     # check_year, check_month = 2024, 4
     # count = get_code_exist(f'{check_year}-{check_month:02d}', "340000")
     # print(count)
 
     # 新表更新地级市同比
-    for province in provinces:
-        update_shandong_yoy(province)
+    # for province in provinces:
+    #     update_shandong_yoy(province)
 
     # 旧表更新省份同比
     # for province in provinces:
     #     update_shandong_yoy_origin(province)
-    log.info("同比sql处理完成")
+    # log.info("同比sql处理完成")

crossborder/zhejiang/crawl_gov_zhejiang_full.py

@@ -16,6 +16,8 @@ from selenium.webdriver.common.by import By
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.webdriver.support.ui import WebDriverWait
 
+from crossborder.utils.base_country_code import get_last_month
+from crossborder.utils.base_mysql import get_commodity_trade_by_prov_year_month
 from crossborder.utils.dingtalk import send_dingtalk_message
 from crossborder.zhejiang import download_dir
 from crossborder.zhejiang import gov_commodity_zhejiang_city
@@ -401,6 +403,12 @@ def main():
             if res == 'finish':
                 duration = time.time() - start_time
                 send_dingtalk_message(f'【浙江省海关】增量数据采集完成，耗时 {duration:.2f} 秒')
+
+                res = get_commodity_trade_by_prov_year_month('浙江省', get_last_month())
+                if res is not None:
+                    send_dingtalk_message(f"【浙江省海关】 commodity_trade 查询到 {len(res)} 条记录，文件已生成")
+                else:
+                    send_dingtalk_message("【浙江省海关】 未查询到任何记录或发生错误")
     except Exception as e:
         send_dingtalk_message(f"【浙江省海关】发生错误：{e}")