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()