项目简介
本项目是基于Python和Selenium库开发的自动化脚本系统,用于简化天河平台上计算任务的管理。通过模拟用户在浏览器端发起请求,实现批量开启作业、关闭作业、删除存储以及执行shell命令等功能,提高了天河平台任务操作效率。
项目的主要特性和功能
- 批量作业管理:支持自定义名称批量开启作业,开启成功后本地保存记录文件;支持根据指定机器名批量关闭作业。
- Shell脚本执行:作为打开机器附属功能,可连接shell运行启动脚本,支持自定义脚本运行后操作。
- 存储管理:作为关闭机器附属功能,可删除指定机器家目录下的文件夹以释放磁盘空间。
- 自动化模拟操作:利用Selenium库,通过网页元素特征模拟鼠标点击、键盘输入,实现自动化测试和操作。
安装使用步骤
环境准备
- 安装Python 3.9。
- 安装核心依赖库,执行命令
pip install selenium
。 - 安装Chrome浏览器,下载与当前Chrome浏览器版本匹配的
ChromeDriver
:- 打开Chrome浏览器【设置】【关于Chrome】查看当前版本。
- 访问
http://chromedriver.storage.googleapis.com/index.html
,根据浏览器版本下载对应的ChromeDriver.exe
。 - 打开脚本
operate.py
,将第157行修改为driver = webdriver.Chrome(executable_path=你的ChromeDriver路径)
。
运行脚本
开启作业
- 在
userFunction.py
中修改get_open_job_name
函数,返回机器名列表。 - 在shell中执行
python main.py -r open
。 - 输入选择配置对应的序号选择要开启的机器的配置。
关闭作业
- 在
userFunction.py
中修改get_close_job_name
函数,返回机器名列表。 - 在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()
中给定要删除的文件夹名列表。
注意事项
- 脚本因网页加载慢使用了
time.sleep(..)
,若因网络问题运行中断,刷新后可能死循环,可通过tqdm
进度重新运行。 - 确保操作的正确性和安全性,脚本操作涉及真实机器和资源。
下载地址
点击下载 【提取码: 4003】【解压密码: www.makuang.net】