home1018

main.py

@@ -0,0 +1,116 @@
+import time
+import json
+from typing import Dict, List, Any
+from config import ConfigManager
+from login_manager import LoginManager
+
+from logger_config import global_logger as logger
+
+class AutoLoginApp:
+    """
+    自动登录应用主类
+    """
+    
+    def __init__(self):
+        """
+        初始化自动登录应用
+        """
+        self.config_manager = ConfigManager()
+        self.login_manager = LoginManager()
+
+        
+    def run(self):
+        """
+        运行自动登录应用
+        """
+        logger.info("===== 自动登录应用启动 =====")
+        
+        # 加载配置
+        config = self.config_manager.load_config()
+        sites = config.get('sites', [])
+        
+        if not sites:
+            logger.warning("配置中没有设置任何网站，使用默认示例网站")
+            # 添加默认示例网站
+            sites = [
+                {'name': '百度', 'url': 'https://www.baidu.com'},
+                {'name': '新浪', 'url': 'https://www.sina.com.cn'}
+            ]
+        
+        # 处理每个网站
+        for site in sites:
+            site_name = site.get('name', '未知网站')
+            site_url = site.get('url', '')
+            
+            if not site_url:
+                logger.warning(f"网站 '{site_name}' 的URL为空，跳过")
+                continue
+            
+            logger.info(f"处理网站: {site_name} ({site_url})")
+            
+            # 获取cookies（从配置中）
+            cookies = []
+            if 'cookies' in site:
+                config_cookies = site['cookies']
+                # 确保cookies是列表格式
+                if isinstance(config_cookies, dict):
+                    # 直接将字典转换为列表格式
+                    cookies = []
+                    for key, value in config_cookies.items():
+                        cookies.append({
+                            'name': key,
+                            'value': value,
+                            'domain': '',
+                            'path': '/',
+                            'expires': 0,
+                            'secure': False,
+                            'httpOnly': False
+                        })
+                else:
+                    cookies = config_cookies
+                logger.info(f"使用配置文件中的 {len(cookies)} 条cookies记录")
+            
+            # 登录网站
+            login_success = self.login_manager.login(site_url, cookies)
+            
+            if login_success:
+                logger.info(f"登录网站 '{site_name}' 成功")
+                
+                # 保持登录状态一段时间，从配置中获取duration值，默认30秒
+                duration = site.get('duration', 30)
+                self.login_manager.maintain_login(duration=duration)  # 保持{duration}秒
+                
+                # 保持登录状态后获取并更新到配置中的cookies
+                new_cookies = self.login_manager.get_cookies()
+                if new_cookies:
+                    # 更新当前site对象的cookies
+                    site['cookies'] = new_cookies
+                    logger.info(f"已更新网站 '{site_name}' 的cookies到配置中，共 {len(new_cookies)} 条记录")
+                
+            else:
+                logger.error(f"登录网站 '{site_name}' 失败")
+            
+            # 关闭当前浏览器页面，准备下一个网站
+            self.login_manager.close()
+            time.sleep(2)  # 间隔2秒再处理下一个网站
+        
+        # 保存更新后的配置到config.json
+        try:
+            self.config_manager.save_config(config)
+            logger.info("已将更新后的cookies保存到config.json文件中")
+        except Exception as e:
+            logger.error(f"保存配置文件时发生错误: {e}")
+        
+        logger.info("===== 所有网站处理完成 =====")
+
+if __name__ == "__main__":
+    app = AutoLoginApp()
+    try:
+        app.run()
+    except KeyboardInterrupt:
+        logger.info("程序被用户中断")
+    except Exception as e:
+        logger.error(f"程序运行时发生错误: {e}")
+    finally:
+        # 确保关闭浏览器
+        app.login_manager.close()