monjack/AutoLoginLWsWeb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Initial commit

Browse Source
This commit is contained in:
monjack
2025-01-07 17:55:50 +08:00
commit eab1d70061
86 changed files with 2772 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

357
main.py Normal file
View File

@ -0,0 +1,357 @@
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
import requests
from PIL import Image
import pytesseract
import io
import time
import re
import os
import tkinter as tk
import ftplib
#
# def ftp_upload_with_proxy(server, username, password, local_file_path, remote_path, proxy):
# """使用 pycurl 上传单个文件到 FTP 服务器"""
# buffer = BytesIO()
# c = pycurl.Curl()
#
# # 设置 FTP URL
# ftp_url = f'ftp://{server}/{remote_path}'
# c.setopt(c.URL, ftp_url)
#
# # 设置 FTP 认证
# c.setopt(c.USERPWD, f"{username}:{password}")
# c.setopt(c.READDATA, open(local_file_path, 'rb'))
# c.setopt(c.UPLOAD, 1)
# c.setopt(c.PREQUOTE, ['PASV'])
#
# # 设置代理
# c.setopt(c.PROXY, proxy)
#
# try:
# c.perform()
# print(f"文件 {local_file_path} 成功上传到 {remote_path}！")
# except pycurl.error as e:
# print(f"FTP 错误: {e}")
# finally:
# c.close()
# def upload_directory_to_ftp(server, username, password, local_directory, remote_directory, proxy):
# """上传整个目录及其子目录到FTP服务器"""
# # 确保远程目录存在
# try:
# # 此段代码需要检查并创建远程目录
# # pycurl 不直接提供创建目录的功能，需要手动实现
#
# # 上传文件和子目录
# for root, dirs, files in os.walk(local_directory):
# # 上传子文件夹
# for dir_name in dirs:
# remote_dir_path = os.path.join(remote_directory,
# os.path.relpath(os.path.join(root, dir_name), local_directory)).replace(
# "\\", "/")
# print(f"准备上传目录: {remote_dir_path}")
# # 假设已确认远程目录存在
#
# # 上传文件
# for filename in files:
# local_file_path = os.path.join(root, filename) # 本地文件的完整路径
# relative_path = os.path.relpath(local_file_path, local_directory) # 计算相对路径
# remote_file_path = os.path.join(remote_directory, relative_path).replace("\\", "/") # 计算远程文件路径
#
# # 上传文件
# ftp_upload_with_proxy(server, username, password, local_file_path, remote_file_path, proxy)
#
# except Exception as e:
# print(f"上传目录时发生错误: {e}")
def upload_file_to_ftp(server, username, password, file_path, remote_path):
try:
with ftplib.FTP(server) as ftp:
ftp.login(user=username, passwd=password)
with open(file_path, 'rb') as file:
ftp.storbinary(f'STOR {remote_path}', file)
print(f"文件 {file_path} 成功上传到 {remote_path}！")
except ftplib.all_errors as e:
print(f"FTP 错误: {e}")
def upload_directory_to_ftp(server, username, password, local_directory, remote_directory):
"""上传整个目录及其子目录到FTP服务器"""
try:
with ftplib.FTP(server) as ftp:
ftp.login(user=username, passwd=password)
# 确保远程目录存在
try:
ftp.cwd(remote_directory) # 改变到远程目录
except ftplib.error_perm:
ftp.mkd(remote_directory) # 如果远程目录不存在，则创建
ftp.cwd(remote_directory) # 再次进入远程目录
# 遍历本地目录树
for root, dirs, files in os.walk(local_directory):
# 上传子文件夹
for dir_name in dirs:
remote_dir_path = os.path.join(remote_directory, os.path.relpath(os.path.join(root, dir_name), local_directory)).replace("\\", "/")
try:
ftp.mkd(remote_dir_path) # 在FTP服务器上创建子目录
print(f"已创建远程目录: {remote_dir_path}")
except ftplib.error_perm:
pass # 如果目录已经存在则忽略
# 上传文件
for filename in files:
local_file_path = os.path.join(root, filename) # 本地文件的完整路径
relative_path = os.path.relpath(local_file_path, local_directory) # 计算相对路径
remote_file_path = os.path.join(remote_directory, relative_path).replace("\\", "/") # 计算远程文件路径
ftp_upload_with_proxy(server, username, password, local_file_path, remote_file_path, proxy_url) # 上传文件
except ftplib.all_errors as e:
print(f"上传目录时发生错误: {e}")
def launch_proxy():
print("开启代理...")
cmd = "start ./Shadowsocks-4.4.1.0/Shadowsocks.exe"
os.system('chcp 65001')
os.system(cmd)
def close_proxy():
print("杀死代理进程")
cmd = "taskkill /f /im Shadowsocks.exe"
os.system('chcp 65001')
os.system(cmd)
cmd = """
reg add "HKCU\\Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings" /v ProxyEnable /t REG_DWORD /d 0 /f
"""
print(cmd)
os.system(cmd)
def shot(chromedriver_path ,chrome_path):
options = Options()
options.add_experimental_option("detach", True)
options.binary_location = chrome_path
options.add_argument("--no-sandbox")
# 初始化 WebDriver
service = Service(chromedriver_path)
driver = webdriver.Chrome(service=service, options=options)
driver.get("http://125.64.9.222:8022/login_exam.php")
try:
launch_proxy()
time.sleep(5)
exam_id = name_entry.get()
student_id = id_entry.get()
password = password_entry.get()
try:
# 创建文件夹
os.mkdir(folder_path) # 如果目录已存在，则会抛出OSError
print(f"文件夹 '{folder_path}' 创建成功！")
except FileExistsError:
print(f"文件夹 '{folder_path}' 已存在！")
except Exception as e:
print(f"创建文件夹时发生错误: {e}")
# 填写 exam_id, student_id 和 password
driver.find_element(By.ID, "exam_id").send_keys(exam_id) # exam_id
driver.find_element(By.ID, "student_id").send_keys(student_id) # student_id
driver.find_element(By.ID, "password").send_keys(password) # password
# 获取验证码图片
captcha_image = driver.find_element(By.CSS_SELECTOR, "img[src='./lib/image_code_1.php']")
# 截取验证码图片
captcha = captcha_image.screenshot_as_png
# 打开验证码图片
captcha = Image.open(io.BytesIO(captcha))
captcha.save("captcha.png") # 可选，保存图片以供调试
captcha_text = pytesseract.image_to_string(captcha, config="--psm 6")
print(f"识别到的验证码: {captcha_text}")
# 填写验证码
driver.find_element(By.ID, "imageCode").send_keys(captcha_text.strip())
# 提交表单
driver.find_element(By.ID, "ok").click() # 替换为实际的提交按钮 ID
# 获取页面 Cookies
cookies = driver.get_cookies()
session_cookies = {cookie['name']: cookie['value'] for cookie in cookies}
# 用 requests 请求目标 URL
url = "http://125.64.9.222:8022/paper/student_server_info.php"
response = requests.get(url, cookies=session_cookies)
response.raise_for_status() # 确保请求成功
print(response.text) # 打印响应内容
matches = [match.strip() for match in re.findall(r"(?<=:).*?(?=<br>)", response.text)]
print(matches)
with open("info.txt", "w", encoding="utf-8") as f:
for match in matches:
f.write(match + "\n") # 保存结果到文件
# command = "start result.txt"
# subprocess.Popen(command, shell=True) # 打开结果文件
# 等待页面加载或处理结果
# WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.ID, "result"))) # 替换为实际的结果 ID
# print("登录成功！")
exam_config = f"""
<?php
$link=mysqli_connect('localhost','{matches[2]}','{matches[3]}','{matches[5]}');
mysqli_query($link,'set names utf8');
// 设置默认时区
date_default_timezone_set("Asia/Shanghai");
"""
with open(folder_path + "db_config.php", "w", encoding="utf-8") as f:
f.write(exam_config) # 保存 db_config.php 文件
ftp_server = matches[0]
ftp_username = matches[2]
ftp_password = matches[3]
# 调用上传函数
upload_directory_to_ftp(ftp_server, ftp_username, ftp_password, local_file_path, remote_file_path, proxy_url)
# url = f"http://{matches[0]}:{matches[1]}/exam_setup.php"
# response = requests.get(url, cookies=session_cookies)
# print(response.raise_for_status())
# print(response.text)
time.sleep(5)
driver.get(f"http://{matches[0]}:{matches[1]}/setup.html")
# 等待页面加载完成
# WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.ID, "kw")))
# 输入搜索词
try:
button = WebDriverWait(driver, 10).until(
EC.element_to_be_clickable((By.XPATH, "//button[text()='create table']")) # 使用按钮的文本查找元素
)
button.click() # 点击按钮
except Exception as e:
print(f"发生错误: {e}")
finally:
result_content = ""
try:
result_element = WebDriverWait(driver, 100).until(
EC.presence_of_element_located((By.ID, "result"))
)
# 获取内容
result_content = result_element.text
print(f"ID为result的内容: {result_content}")
except Exception as e:
print(f"发生错误: {e}")
print("执行完毕：" + result_content)
driver.get("http://125.64.9.222:8022/paper/paper.php")
# 等待页面加载
wait = WebDriverWait(driver, 30)
for fillingQ_id in fillingQ_ids:
textarea = wait.until(EC.presence_of_element_located((By.ID, fillingQ_id)))
with open (fillingQ_path + f"/{fillingQ_id}.txt", "r", encoding="utf-8") as f:
textarea.send_keys(f.read()) # 填写题目
for save in fillingQ_save_ids:
driver.execute_script(f"{save}")
print(f"已保存 {save} 题目")
time.sleep(1)
tk.messagebox.showinfo("提示","执行成功！请手动进入网站点击调试")
except Exception as e:
print(f"发生错误: {e}")
tk.messagebox.showerror("错误", f"发生错误: {e}")
finally:
close_proxy()
driver.quit()
if __name__ == "__main__":
# 设置 Tesseract OCR 的路径
pytesseract.pytesseract.tesseract_cmd = r'.\Tesseract\tesseract.exe'
folder_path = "./20250103/"
chrome_path = "./chrome-win64/chrome.exe" # 替换为你的路径
chromedriver_path = "./chromedriver.exe" # 替换为你的 ChromeDriver 路径
exam_id = "20250103" # 填写题号
student_id = "" # 填写学号
password = "" # 填写密码
ftp_server = 'ftp.example.com' # 替换为FTP服务器地址
ftp_username = 'your_username' # 替换为FTP用户名
ftp_password = 'your_password' # 替换为FTP密码
proxy_url = "http://127.0.0.1:22222" #代理地址
local_file_path = folder_path # 替换为要上传的本地文件路径
fillingQ_path = "./blankFillingQ"
remote_file_path = '/' # 替换为远程服务器上保存的文件路径
fillingQ_ids = ["textarea_student_answer_01", "textarea_student_answer_01_2_e","textarea_student_answer_01_3_e","textarea_student_answer_02","textarea_student_answer_521_1",
"textarea_student_answer_60_2_e"]
fillingQ_save_ids = ["save('01')", "save('01_2_e')","save('01_3_e')","save('02')","save('521_1')"]
root = tk.Tk()
root.title("一键执行")
# 设置窗口大小
window_width = 300
window_height = 200
root.geometry(f"{window_width}x{window_height}")
# 获取屏幕宽度和高度
screen_width = root.winfo_screenwidth()
screen_height = root.winfo_screenheight()
# 计算主窗口的位置
x = (screen_width // 2) - (window_width // 2)
y = (screen_height // 2) - (window_height // 2)
# 设置窗口位置
root.geometry(f"{window_width}x{window_height}+{x}+{y}")
# 标签
label = tk.Label(root, text="题号")
label.place(x=50, y=20)
name_entry = tk.Entry(root)
name_entry.pack(pady=10)
name_entry.place(x=100, y=20)
name_entry.insert(0, exam_id)
name_entry.config(state="disabled")
label2 = tk.Label(root, text="学号")
label2.place(x=50, y=60)
id_entry = tk.Entry(root)
id_entry.pack(pady=10)
id_entry.place(x=100, y=60)
label3 = tk.Label(root, text="密码")
label3.place(x=50, y=100)
password_entry = tk.Entry(root, show="*")
password_entry.pack(pady=10)
password_entry.place(x=100, y=100)
# 按钮
submit_button = tk.Button(root, text="执行", command=lambda: shot(chromedriver_path, chrome_path))
submit_button.pack(pady=10)
submit_button.pack(side=tk.BOTTOM)
submit_button.config(width=10, height=1)
root.mainloop()