基本信息文件创建

base/BaseCore.py

@@ -7,6 +7,7 @@ import time
 
 import logbook
 import logbook.more
+import openpyxl
 import pandas as pd
 import requests
 import zhconv
@@ -586,6 +587,16 @@ class BaseCore:
             return 'cn'
         return result[0]
 
+    #创建excel文件
+    def check_excel_file(self,file_path):
+        if os.path.isfile(file_path):
+            self.getLogger().info("Excel文件已存在")
+            return True
+        else:
+            self.getLogger().info("Excel文件不存在，正在创建...")
+            return False
+
+
     # 追加接入excel
     def writerToExcel(self, detailList, filename):
         # filename='baidu搜索.xlsx'

comData/BaseInfo_qcc/base_info.py

 # -*- coding: utf-8 -*-
+import openpyxl
 import pandas as pd
 
 import time
@@ -407,7 +408,15 @@ if __name__ == '__main__':
             }
             name_list.append(name_compile)
             nowtime = baseCore.getNowTime(1).replace('-', '_')[:10]
-            baseCore.writerToExcel(name_list, f'企业名称对比.xlsx')
+            file_name = f'企业名称对比_{nowtime}.xlsx'
+            file_exist = baseCore.check_excel_file(file_name)
+            if file_exist:
+                pass
+            else:
+                wb = openpyxl.Workbook()
+                wb.save(file_name)
+                log.info("Excel文件已创建")
+            baseCore.writerToExcel(name_list, file_name)
             log.info(f'采集{com_name}成功=======耗时{baseCore.getTimeCost(start_time,time.time())}')
             try:
                 producer = KafkaProducer(bootstrap_servers=['114.115.159.144:9092'], api_version=(2, 0, 2))