littlebot
Published on 2025-04-10 / 8 Visits
0

【源码】基于Python和Selenium的天河平台自动化任务管理系统

项目简介

本项目是基于Python和Selenium库开发的自动化脚本系统,用于简化天河平台上计算任务的管理。通过模拟用户在浏览器端发起请求,实现批量开启作业、关闭作业、删除存储以及执行shell命令等功能,提高了天河平台任务操作效率。

项目的主要特性和功能

  1. 批量作业管理:支持自定义名称批量开启作业,开启成功后本地保存记录文件;支持根据指定机器名批量关闭作业。
  2. Shell脚本执行:作为打开机器附属功能,可连接shell运行启动脚本,支持自定义脚本运行后操作。
  3. 存储管理:作为关闭机器附属功能,可删除指定机器家目录下的文件夹以释放磁盘空间。
  4. 自动化模拟操作:利用Selenium库,通过网页元素特征模拟鼠标点击、键盘输入,实现自动化测试和操作。

安装使用步骤

环境准备

  1. 安装Python 3.9。
  2. 安装核心依赖库,执行命令pip install selenium
  3. 安装Chrome浏览器,下载与当前Chrome浏览器版本匹配的ChromeDriver
    • 打开Chrome浏览器【设置】【关于Chrome】查看当前版本。
    • 访问http://chromedriver.storage.googleapis.com/index.html,根据浏览器版本下载对应的ChromeDriver.exe
    • 打开脚本operate.py,将第157行修改为driver = webdriver.Chrome(executable_path=你的ChromeDriver路径)

运行脚本

开启作业

  1. userFunction.py中修改get_open_job_name函数,返回机器名列表。
  2. 在shell中执行python main.py -r open
  3. 输入选择配置对应的序号选择要开启的机器的配置。

关闭作业

  1. userFunction.py中修改get_close_job_name函数,返回机器名列表。
  2. 在shell中执行python main.py -r close

连接shell运行脚本

在shell中执行python main.py -r open -e,可自定义修改get_url()operate_process()函数。

删除家目录文件夹

在shell中执行python main.py -r close -d,并在get_dir_name()中给定要删除的文件夹名列表。

注意事项

  1. 脚本因网页加载慢使用了time.sleep(..),若因网络问题运行中断,刷新后可能死循环,可通过tqdm进度重新运行。
  2. 确保操作的正确性和安全性,脚本操作涉及真实机器和资源。

下载地址

点击下载 【提取码: 4003】【解压密码: www.makuang.net】