|
|
@@ -24,6 +24,7 @@ class ExcelFileManager:
|
|
|
self.db = DbManager()
|
|
|
self.wb:Workbook = self._prepare_workbook()
|
|
|
logger.info(f"{self.wb.sheetnames}")
|
|
|
+ logger.info(f"{self.wb.worksheets}")
|
|
|
|
|
|
def _prepare_workbook(self):
|
|
|
"""准备工作簿"""
|
|
|
@@ -37,8 +38,10 @@ class ExcelFileManager:
|
|
|
self.wb.save(self.output_path)
|
|
|
self.wb.close()
|
|
|
|
|
|
- def write_competie_sheet(self, sheet_name: str = "竞品关键词调研", sheet_index: int = 0):
|
|
|
- if not sheet_name in self.wb.worksheets:
|
|
|
+ def write_competie_sheet(self, sheet_name: str = "竞品关键词调研", sheet_index: int = 0, overwrite: bool = False):
|
|
|
+ if overwrite and sheet_name in self.wb.sheetnames:
|
|
|
+ self.wb.remove(self.wb[sheet_name])
|
|
|
+ if sheet_name not in self.wb.sheetnames:
|
|
|
extract_data = self.load_s3_extract_data()
|
|
|
competitive_sheet_writer = CompetitiveAnalysisWriter(self.wb, sheet_name=sheet_name, sheet_index=sheet_index)
|
|
|
competitive_sheet_writer.add_data(extract_data)
|