230 lines
8.2 KiB
Python
230 lines
8.2 KiB
Python
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
|
||
from tkinter import messagebox # 单独导入 messagebox
|
||
from ftp import set_proxy
|
||
from ftp import upload_file_to_ftp
|
||
from ftp import upload_directory_to_ftp
|
||
from ftp import create_remote_directory
|
||
import requests
|
||
|
||
# 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.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:
|
||
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)
|
||
|
||
url = f"http://{matches[0]}:{matches[1]}/setup_2.php"
|
||
response = requests.get(url, cookies=session_cookies)
|
||
print(response.raise_for_status())
|
||
print(response.text)
|
||
|
||
time.sleep(5)
|
||
#
|
||
# driver.get("http://125.64.9.222:8022/paper/paper.php")
|
||
#
|
||
# # 等待页面加载
|
||
# wait = WebDriverWait(driver, 20)
|
||
# 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:
|
||
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() |