79 lines
1.9 KiB
Python
79 lines
1.9 KiB
Python
"""
|
|
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"]()
|
|
|
|
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()
|