refactor(build): 优化构建脚本并调整下载路径设置

- 移除 build 目录和 spec 文件的删除操作
- 修改 gui.py 和 main.py 中的下载路径设置，不再创建 downloads 目录
- 在 payment.py 和 return_goods.py 中添加文件保存完成的日志记录
- 优化 rabbitmq.py 中的连接初始化和关闭逻辑
- 修改 spa.py 中的文件保存日志记录位置

app/helper/rabbitmq.py

@@ -16,7 +16,7 @@ class RabbitMQClient:
         self.username = username
         self.password = password
 
-        self.connection = None
+        self.conn = None
         self.channel = None
 
         self.exchange = None
@@ -27,10 +27,10 @@ class RabbitMQClient:
         """初始化连接和通道"""
         try:
             credentials = pika.PlainCredentials(self.username, self.password)
-            self.connection = pika.BlockingConnection(pika.ConnectionParameters(
+            self.conn = pika.BlockingConnection(pika.ConnectionParameters(
                 host=self.host, port=self.port, credentials=credentials
             ))
-            self.channel = self.connection.channel()
+            self.channel = self.conn.channel()
             print(f"Initialized connection to RabbitMQ at {self.host}.")
         except Exception as e:
             print(f"Error initializing RabbitMQ connection: {e}")
@@ -111,6 +111,6 @@ class RabbitMQClient:
 
     def close(self):
         """关闭连接"""
-        if self.connection and not self.connection.is_closed:
-            self.connection.close()
+        if self.conn and not self.conn.is_closed:
+            self.conn.close()
             print("Connection closed.")

app/vc/payment.py

@@ -391,6 +391,8 @@ class Payment(AutoInterface):
             price_pay_summary = pd.concat(all_price_pay_data, ignore_index=True)
             excel.save_xls(price_pay_summary, new_file_name, "Price导出明细")
 
+        self.logger.info(f"文件 {new_file_name} 保存完成，路径：{os.path.abspath(new_file_name)}")
+
     def push_data_queue(self):
         rabbit.connection()
         rabbit.connect(queue='refund_robot', routing_key='refund_robot', exchange='reports')

app/vc/return_goods.py

@@ -14,7 +14,7 @@ from app.vc.interface import AutoInterface
 
 
 class ReturnGoods(AutoInterface):
-    def __init__(self,logger: Logger, page: Page, country: str,shop_code: str):
+    def __init__(self, logger: Logger, page: Page, country: str, shop_code: str):
         self.logger = logger
         self.page = page
         self.country = country
@@ -76,7 +76,7 @@ class ReturnGoods(AutoInterface):
             })
 
             for _, item_row in item_data_result.iterrows():
-                relation = relations_dict.get(item_row.get('ASIN'))
+                relation = relations_dict.get(item_row.get('ASIN'), {})
                 erp_sku = relation.get('erp_sku', "")
                 data_dict = data.to_dict()
                 data_dict.update({
@@ -98,6 +98,8 @@ class ReturnGoods(AutoInterface):
 
         excel.save_xls(new_list_data, self.result_file_name)
 
+        self.logger.info(f"文件 {self.result_file_name} 保存完成，路径：{os.path.abspath(self.result_file_name)}")
+
     def push_data_queue(self):
         rabbit.connection()
         rabbit.connect(queue='return_robot', routing_key='return_robot', exchange='reports')
@@ -123,4 +125,4 @@ class ReturnGoods(AutoInterface):
             # 推送数据
             rabbit.send_message(push_data)
 
-        rabbit.close()
\ No newline at end of file
+        rabbit.close()

app/vc/spa.py

@@ -196,8 +196,6 @@ class Spa(AutoInterface):
                 #     for sheet_name, data in large_sheet_data.items():
                 #         executor.submit(write_sheet, writer, data, sheet_name)
 
-        self.logger.info(f"文件 {new_file_name} 保存完成，路径：{os.path.abspath(new_file_name)}")
-
     def push_data_queue(self):
         rabbit.connection()
         rabbit.connect(queue='spa_robot', routing_key='spa_robot', exchange='reports')
@@ -281,3 +279,5 @@ class Spa(AutoInterface):
 
         # 保存数据到 Excel 文件
         self.__save_excel(sheet_data, large_sheet_data, self.result_file_name)
+
+        self.logger.info(f"文件 {self.result_file_name} 保存完成，路径：{os.path.abspath(self.result_file_name)}")

build.bat

@@ -2,5 +2,4 @@ pip.exe install -i https://mirrors.cloud.tencent.com/pypi/simple -r requirements
 pyinstaller -F -n amazon_cmd.exe main.py
 pyinstaller -F -n amazon_gui.exe --noconsole gui.py
 pyinstaller -F -n easy_gui.exe --noconsole easy.py
-rd /s /q build
 del *.spec
\ No newline at end of file

gui.py

@@ -213,8 +213,7 @@ class VCManagerGUI(ttk.Window):
         self.page = ChromiumPage()
         self.page.set.load_mode.normal()
         self.page.set.when_download_file_exists('overwrite')
-        download_path = os.path.join(os.getcwd(), "downloads")
-        file.make_dir(download_path)
+        download_path = os.path.join(os.getcwd())
         self.page.set.download_path(download_path)
         self.log(f"下载目录设置为：{download_path}")
 

main.py

@@ -18,8 +18,6 @@ if __name__ == '__main__':
     page.set.when_download_file_exists('overwrite')
     # 下载目录
     download_path = os.getcwd()
-    # 检查下载目录是否存在，如果不存在则创建
-    file.make_dir(download_path)
     # 设置下载路径，确保在打开浏览器前设置
     page.set.download_path(download_path)