first commit

main.py

@@ -0,0 +1,78 @@
+"""
+RPA-robot 主入口
+
+统一调度所有自动化任务
+"""
+import argparse
+from asyncio import Task
+import sys
+
+from tasks import open_feishu_chat
+from utils import get_logger, show_warning_countdown
+
+log = get_logger("main")
+
+# 任务注册表
+TASKS = {
+    "feishu-chat": {
+        "desc": "打开飞书并找到指定群聊截图（默认：资产管理组）",
+        "func": open_feishu_chat.feishu_groups_list,
+    },
+}
+
+
+def list_tasks():
+    """列出所有可用任务"""
+    print("\n可用任务：")
+    print("-" * 40)
+    for name, info in TASKS.items():
+        print(f"  {name:20s} {info['desc']}")
+    print()
+
+
+def run_task(task_name: str, *args):
+    """执行指定任务"""
+    task = TASKS.get(task_name)
+    show_warning_countdown(message="即将开始自动截图", seconds=5)
+    if not task:
+        log.error("未知任务: %s", task_name)
+        list_tasks()
+        for task in TASKS:
+            task["func"](*args)
+
+    log.info("===== 开始执行任务: %s =====", task_name)
+    try:
+        if args:
+            task["func"](*args)
+        else:
+            task["func"]()
+        log.info("===== 任务完成: %s =====", task_name)
+    except Exception as e:
+        log.error("任务执行失败 [%s]: %s", task_name, e)
+        sys.exit(1)
+    show_warning_countdown(message="任务完成", seconds=3)
+
+
+def main():
+    parser = argparse.ArgumentParser(description="RPA-robot 自动化任务调度")
+    parser.add_argument(
+        "task",
+        nargs="?",
+        help="要执行的任务名称（不传则列出所有任务）",
+    )
+    parser.add_argument(
+        "args",
+        nargs="*",
+        help="任务参数（如 web 任务的网页名称）",
+    )
+    args = parser.parse_args()
+
+    if not args.task:
+        list_tasks()
+        return
+
+    
+    run_task(args.task, *args.args)
+
+if __name__ == "__main__":
+    main()