HydroOJ 0.0.1__tar.gz

hydrooj-0.0.1/HydroOJ.egg-info/PKG-INFO

@@ -0,0 +1,127 @@
+Metadata-Version: 2.4
+Name: HydroOJ
+Version: 0.0.1
+Summary: A Python SDK for interacting with HydroOJ via Selenium
+Author-email: YANGRENRUIYRR <yangrenruiyrr@yeah.net>
+Keywords: HydroOJ,selenium,online judge
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: selenium
+
+# HydroOJ
+
+基于Selenium开发的HydroOJ自动化操作库，支持账号登录、题目/比赛管理、附件上传等自动化操作。
+
+## 安装方法
+
+### PyPI在线安装
+
+```bash
+pip install HydroOJ
+```
+
+## 依赖列表
+
+```
+selenium
+```
+
+支持浏览器：Edge、Chrome、Firefox、Ie、Safari、WebKitGTK
+
+## 快速使用示例
+
+```python
+from HydroOJ import HydroOJ, LoginFailedError
+
+# 初始化客户端
+client = HydroOJ(
+    url="https://hydro.ac/",
+    webdriver_id="Chrome",
+    headless=True
+)
+
+# 登录账号
+try:
+    client.Login("用户名", "密码")
+    print("登录成功")
+except LoginFailedError:
+    print("登录失败，账号或密码错误")
+
+# 获取第一页题目列表
+print(client.ViewProblemList(page=1))
+
+# 查看题目内容
+client.ViewProblem("1001")
+
+# 释放浏览器资源
+client.Quit()
+```
+
+## API 说明
+
+### HydroOJ 初始化参数
+
+```python
+HydroOJ(
+    url="https://hydro.ac/",
+    webdriver_id="Edge",
+    webdriver_path=None,
+    headless=True,
+    disable_blink=True
+)
+```
+
+- url：HydroOJ站点地址
+- webdriver_id：浏览器标识，支持 Edge / Chrome / Firefox / Ie / Safari / WebKitGTK
+- webdriver_path：浏览器驱动本地路径，留空自动寻找
+- headless：是否启用无头模式，不弹出浏览器窗口
+- disable_blink：关闭自动化检测特征，避免站点拦截
+
+### 全部可用方法
+
+1. Login(username, password)
+   登录站点，登录失败抛出 LoginFailedError
+2. Logined()
+   检测当前登录状态，返回布尔值
+3. ViewProblemList(page=1)
+   获取指定页码的题目列表文本，无数据抛出 ProblemListNotFoundError
+4. ViewProblem(problem_id)
+   打开并打印指定ID题目详情，题目不存在抛出 ProblemNotFoundError
+5. CreateProblem(pid, title, content, tag=[], difficulty=None, hidden=False)
+   创建新题目，创建失败抛出 CreateProblemFailedError
+6. ViewContestList(page=1)
+   获取指定页码比赛列表，无数据抛出 ContestListNotFoundError
+7. ViewContest(contest_id)
+   打开并打印指定ID比赛详情，比赛不存在抛出 ContestNotFoundError
+8. UploadFileToProblem(problem_id, file_name, file_content)
+   为指定题目上传文本附件，上传失败抛出 UploadFileFailedError
+9. GotoPage(url)
+   浏览器跳转至任意链接
+10. RunJavaScript(script)
+    在当前页面执行JS脚本，返回执行结果
+11. Quit()
+    关闭浏览器，释放进程资源
+
+### 内置异常类
+
+- LoginFailedError：登录校验失败
+- ProblemNotFoundError：目标题目不存在
+- CreateProblemFailedError：新建题目操作异常
+- ProblemListNotFoundError：题目列表加载失败
+- ContestListNotFoundError：比赛列表加载失败
+- ContestNotFoundError：目标比赛不存在
+- UploadFileFailedError：题目附件上传失败
+
+## 使用注意事项
+
+1. 高频批量操作建议添加 time.sleep() 降低访问频率，触发网站风控
+2. 程序结束必须调用 Quit()，否则浏览器后台进程残留占用内存
+3. 仅适配原版HydroOJ前端页面，二次修改部署的OJ可能存在元素定位失效问题
+
+## License
+
+MIT

hydrooj-0.0.1/HydroOJ.egg-info/SOURCES.txt

@@ -0,0 +1,14 @@
+README.md
+pyproject.toml
+HydroOJ.egg-info/PKG-INFO
+HydroOJ.egg-info/SOURCES.txt
+HydroOJ.egg-info/dependency_links.txt
+HydroOJ.egg-info/requires.txt
+HydroOJ.egg-info/top_level.txt
+hydrooj/HydroOJ.py
+hydrooj/__init__.py
+hydrooj.egg-info/PKG-INFO
+hydrooj.egg-info/SOURCES.txt
+hydrooj.egg-info/dependency_links.txt
+hydrooj.egg-info/requires.txt
+hydrooj.egg-info/top_level.txt

hydrooj-0.0.1/HydroOJ.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

hydrooj-0.0.1/HydroOJ.egg-info/requires.txt

@@ -0,0 +1 @@
+selenium

hydrooj-0.0.1/HydroOJ.egg-info/top_level.txt

@@ -0,0 +1 @@
+hydrooj

hydrooj-0.0.1/PKG-INFO

@@ -0,0 +1,127 @@
+Metadata-Version: 2.4
+Name: HydroOJ
+Version: 0.0.1
+Summary: A Python SDK for interacting with HydroOJ via Selenium
+Author-email: YANGRENRUIYRR <yangrenruiyrr@yeah.net>
+Keywords: HydroOJ,selenium,online judge
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: selenium
+
+# HydroOJ
+
+基于Selenium开发的HydroOJ自动化操作库，支持账号登录、题目/比赛管理、附件上传等自动化操作。
+
+## 安装方法
+
+### PyPI在线安装
+
+```bash
+pip install HydroOJ
+```
+
+## 依赖列表
+
+```
+selenium
+```
+
+支持浏览器：Edge、Chrome、Firefox、Ie、Safari、WebKitGTK
+
+## 快速使用示例
+
+```python
+from HydroOJ import HydroOJ, LoginFailedError
+
+# 初始化客户端
+client = HydroOJ(
+    url="https://hydro.ac/",
+    webdriver_id="Chrome",
+    headless=True
+)
+
+# 登录账号
+try:
+    client.Login("用户名", "密码")
+    print("登录成功")
+except LoginFailedError:
+    print("登录失败，账号或密码错误")
+
+# 获取第一页题目列表
+print(client.ViewProblemList(page=1))
+
+# 查看题目内容
+client.ViewProblem("1001")
+
+# 释放浏览器资源
+client.Quit()
+```
+
+## API 说明
+
+### HydroOJ 初始化参数
+
+```python
+HydroOJ(
+    url="https://hydro.ac/",
+    webdriver_id="Edge",
+    webdriver_path=None,
+    headless=True,
+    disable_blink=True
+)
+```
+
+- url：HydroOJ站点地址
+- webdriver_id：浏览器标识，支持 Edge / Chrome / Firefox / Ie / Safari / WebKitGTK
+- webdriver_path：浏览器驱动本地路径，留空自动寻找
+- headless：是否启用无头模式，不弹出浏览器窗口
+- disable_blink：关闭自动化检测特征，避免站点拦截
+
+### 全部可用方法
+
+1. Login(username, password)
+   登录站点，登录失败抛出 LoginFailedError
+2. Logined()
+   检测当前登录状态，返回布尔值
+3. ViewProblemList(page=1)
+   获取指定页码的题目列表文本，无数据抛出 ProblemListNotFoundError
+4. ViewProblem(problem_id)
+   打开并打印指定ID题目详情，题目不存在抛出 ProblemNotFoundError
+5. CreateProblem(pid, title, content, tag=[], difficulty=None, hidden=False)
+   创建新题目，创建失败抛出 CreateProblemFailedError
+6. ViewContestList(page=1)
+   获取指定页码比赛列表，无数据抛出 ContestListNotFoundError
+7. ViewContest(contest_id)
+   打开并打印指定ID比赛详情，比赛不存在抛出 ContestNotFoundError
+8. UploadFileToProblem(problem_id, file_name, file_content)
+   为指定题目上传文本附件，上传失败抛出 UploadFileFailedError
+9. GotoPage(url)
+   浏览器跳转至任意链接
+10. RunJavaScript(script)
+    在当前页面执行JS脚本，返回执行结果
+11. Quit()
+    关闭浏览器，释放进程资源
+
+### 内置异常类
+
+- LoginFailedError：登录校验失败
+- ProblemNotFoundError：目标题目不存在
+- CreateProblemFailedError：新建题目操作异常
+- ProblemListNotFoundError：题目列表加载失败
+- ContestListNotFoundError：比赛列表加载失败
+- ContestNotFoundError：目标比赛不存在
+- UploadFileFailedError：题目附件上传失败
+
+## 使用注意事项
+
+1. 高频批量操作建议添加 time.sleep() 降低访问频率，触发网站风控
+2. 程序结束必须调用 Quit()，否则浏览器后台进程残留占用内存
+3. 仅适配原版HydroOJ前端页面，二次修改部署的OJ可能存在元素定位失效问题
+
+## License
+
+MIT

hydrooj-0.0.1/README.md

@@ -0,0 +1,113 @@
+# HydroOJ
+
+基于Selenium开发的HydroOJ自动化操作库，支持账号登录、题目/比赛管理、附件上传等自动化操作。
+
+## 安装方法
+
+### PyPI在线安装
+
+```bash
+pip install HydroOJ
+```
+
+## 依赖列表
+
+```
+selenium
+```
+
+支持浏览器：Edge、Chrome、Firefox、Ie、Safari、WebKitGTK
+
+## 快速使用示例
+
+```python
+from HydroOJ import HydroOJ, LoginFailedError
+
+# 初始化客户端
+client = HydroOJ(
+    url="https://hydro.ac/",
+    webdriver_id="Chrome",
+    headless=True
+)
+
+# 登录账号
+try:
+    client.Login("用户名", "密码")
+    print("登录成功")
+except LoginFailedError:
+    print("登录失败，账号或密码错误")
+
+# 获取第一页题目列表
+print(client.ViewProblemList(page=1))
+
+# 查看题目内容
+client.ViewProblem("1001")
+
+# 释放浏览器资源
+client.Quit()
+```
+
+## API 说明
+
+### HydroOJ 初始化参数
+
+```python
+HydroOJ(
+    url="https://hydro.ac/",
+    webdriver_id="Edge",
+    webdriver_path=None,
+    headless=True,
+    disable_blink=True
+)
+```
+
+- url：HydroOJ站点地址
+- webdriver_id：浏览器标识，支持 Edge / Chrome / Firefox / Ie / Safari / WebKitGTK
+- webdriver_path：浏览器驱动本地路径，留空自动寻找
+- headless：是否启用无头模式，不弹出浏览器窗口
+- disable_blink：关闭自动化检测特征，避免站点拦截
+
+### 全部可用方法
+
+1. Login(username, password)
+   登录站点，登录失败抛出 LoginFailedError
+2. Logined()
+   检测当前登录状态，返回布尔值
+3. ViewProblemList(page=1)
+   获取指定页码的题目列表文本，无数据抛出 ProblemListNotFoundError
+4. ViewProblem(problem_id)
+   打开并打印指定ID题目详情，题目不存在抛出 ProblemNotFoundError
+5. CreateProblem(pid, title, content, tag=[], difficulty=None, hidden=False)
+   创建新题目，创建失败抛出 CreateProblemFailedError
+6. ViewContestList(page=1)
+   获取指定页码比赛列表，无数据抛出 ContestListNotFoundError
+7. ViewContest(contest_id)
+   打开并打印指定ID比赛详情，比赛不存在抛出 ContestNotFoundError
+8. UploadFileToProblem(problem_id, file_name, file_content)
+   为指定题目上传文本附件，上传失败抛出 UploadFileFailedError
+9. GotoPage(url)
+   浏览器跳转至任意链接
+10. RunJavaScript(script)
+    在当前页面执行JS脚本，返回执行结果
+11. Quit()
+    关闭浏览器，释放进程资源
+
+### 内置异常类
+
+- LoginFailedError：登录校验失败
+- ProblemNotFoundError：目标题目不存在
+- CreateProblemFailedError：新建题目操作异常
+- ProblemListNotFoundError：题目列表加载失败
+- ContestListNotFoundError：比赛列表加载失败
+- ContestNotFoundError：目标比赛不存在
+- UploadFileFailedError：题目附件上传失败
+
+## 使用注意事项
+
+1. 高频批量操作建议添加 time.sleep() 降低访问频率，触发网站风控
+2. 程序结束必须调用 Quit()，否则浏览器后台进程残留占用内存
+3. 仅适配原版HydroOJ前端页面，二次修改部署的OJ可能存在元素定位失效问题
+
+## License
+
+MIT

hydrooj-0.0.1/hydrooj/HydroOJ.py

@@ -0,0 +1,252 @@
+import json
+from selenium import webdriver
+from selenium.webdriver.common.by import By
+
+
+class LoginFailedError(Exception):
+    pass
+
+
+class ProblemNotFoundError(Exception):
+    pass
+
+
+class CreateProblemFailedError(Exception):
+    pass
+
+
+class ProblemListNotFoundError(Exception):
+    pass
+
+
+class ContestListNotFoundError(Exception):
+    pass
+
+
+class ContestNotFoundError(Exception):
+    pass
+
+
+class UploadFileFailedError(Exception):
+    pass
+
+
+class HydroOJ:
+
+    def __init__(
+        self,
+        url="https://hydro.ac/",
+        webdriver_id="Edge",
+        webdriver_path=None,
+        headless=True,
+        disable_blink=True,
+    ):
+        self.url = url
+        if self.url[len(url) - 1] != "/":
+            self.url += "/"
+        if webdriver_id == "Edge":
+            self.options = webdriver.EdgeOptions()
+            if headless:
+                self.options.add_argument("headless")
+            if disable_blink:
+                self.options.add_argument(
+                    "--disable-blink-features=AutomationControlled"
+                )
+            if webdriver_path:
+                self.service = webdriver.EdgeService(webdriver_path)
+            else:
+                self.service = webdriver.EdgeService()
+            self.driver = webdriver.Edge(service=self.service, options=self.options)
+        elif webdriver_id == "Chrome":
+            self.options = webdriver.ChromeOptions()
+            if headless:
+                self.options.add_argument("headless")
+            if disable_blink:
+                self.options.add_argument(
+                    "--disable-blink-features=AutomationControlled"
+                )
+            if webdriver_path:
+                self.service = webdriver.ChromeService(webdriver_path)
+            else:
+                self.service = webdriver.ChromeService()
+            self.driver = webdriver.Chrome(service=self.service, options=self.options)
+        elif webdriver_id == "Firefox":
+            self.options = webdriver.FirefoxOptions()
+            if headless:
+                self.options.add_argument("headless")
+            if disable_blink:
+                self.options.add_argument(
+                    "--disable-blink-features=AutomationControlled"
+                )
+            if webdriver_path:
+                self.service = webdriver.FirefoxService(webdriver_path)
+            else:
+                self.service = webdriver.FirefoxService()
+            self.driver = webdriver.Firefox(service=self.service, options=self.options)
+        elif webdriver_id == "Ie":
+            self.options = webdriver.IeOptions()
+            if headless:
+                self.options.add_argument("headless")
+            if disable_blink:
+                self.options.add_argument(
+                    "--disable-blink-features=AutomationControlled"
+                )
+            if webdriver_path:
+                self.service = webdriver.IeService(webdriver_path)
+            else:
+                self.service = webdriver.IeService()
+            self.driver = webdriver.Ie(service=self.service, options=self.options)
+        elif webdriver_id == "Safari":
+            self.options = webdriver.SafariOptions()
+            if headless:
+                self.options.add_argument("headless")
+            if disable_blink:
+                self.options.add_argument(
+                    "--disable-blink-features=AutomationControlled"
+                )
+            if webdriver_path:
+                self.service = webdriver.SafariService(webdriver_path)
+            else:
+                self.service = webdriver.SafariService()
+            self.driver = webdriver.Safari(service=self.service, options=self.options)
+        elif webdriver_id == "WebKitGTK":
+            self.options = webdriver.WebKitGTKOptions()
+            if headless:
+                self.options.add_argument("headless")
+            if disable_blink:
+                self.options.add_argument(
+                    "--disable-blink-features=AutomationControlled"
+                )
+            if webdriver_path:
+                self.service = webdriver.WebKitGTKService(webdriver_path)
+            else:
+                self.service = webdriver.WebKitGTKService()
+            self.driver = webdriver.WebKitGTK(
+                service=self.service, options=self.options
+            )
+        else:
+            raise ValueError("Unsupported webdriver_id.")
+
+    def Logined(self):
+        self.driver.get(self.url)
+        try:
+            self.driver.find_element(By.NAME, "nav_login")
+            return False
+        except:
+            return True
+
+    def Login(self, username, password):
+        self.driver.get(self.url + "login")
+        self.driver.find_element(By.NAME, "uname").send_keys(username)
+        self.driver.find_element(By.NAME, "password").send_keys(password)
+        self.driver.find_element(By.NAME, "login_submit").click()
+        if not self.Logined():
+            raise LoginFailedError("Login failed.")
+
+    def ViewProblemList(self, page=1):
+        self.driver.get(self.url + "p?page=" + str(page))
+        try:
+            problem_list = self.driver.find_elements(
+                By.XPATH, '//div[@data-fragment-id="problem_list"]//a[@target="_blank"]'
+            )
+            if not problem_list:
+                raise ProblemListNotFoundError("Problem list not found.")
+            ret = ""
+            for i in problem_list:
+                ret += i.text + "\n"
+            return ret
+        except:
+            raise ProblemListNotFoundError("Problem list not found.")
+
+    def ViewProblem(self, problem_id):
+        self.driver.get(self.url + "p/" + problem_id)
+        try:
+            content = self.driver.execute_script("return UiContext.pdoc.content;")
+            language = self.driver.execute_script("return UserContext.viewLang;")
+            content = json.loads(content)
+            print(content[language])
+        except:
+            raise ProblemNotFoundError("Problem not found.")
+
+    def CreateProblem(self, pid, title, content, tag=[], difficulty=None, hidden=False):
+        self.driver.get(self.url + "problem/create")
+        try:
+            self.driver.find_element(By.NAME, "pid").send_keys(pid)
+            self.driver.find_element(By.NAME, "title").send_keys(title)
+            if hidden:
+                self.driver.find_element(
+                    By.CSS_SELECTOR, ".checkbox:nth-child(2)"
+                ).click()
+            for i in tag:
+                self.driver.find_element(By.NAME, "tag").send_keys(i)
+                if i != tag[len(tag) - 1]:
+                    self.driver.find_element(By.NAME, "tag").send_keys(",")
+            if difficulty and difficulty >= 0 and difficulty <= 10:
+                self.driver.find_element(By.NAME, "difficulty").send_keys(difficulty)
+            self.driver.find_element(By.NAME, "content")
+            self.driver.execute_script(
+                "document.getElementsByName('content')[0].textContent=arguments[0]",
+                content,
+            )
+            self.driver.find_element(By.CSS_SELECTOR, ".rounded").click()
+        except:
+            raise CreateProblemFailedError("Failed to create problem.")
+
+    def ViewContestList(self, page=1):
+        self.driver.get(self.url + "contest?page=" + str(page))
+        try:
+            contest_list = self.driver.find_elements(
+                By.XPATH, '//*[contains(@class,"contest__title")]'
+            )
+            if not contest_list:
+                raise ContestListNotFoundError("Contest list not found.")
+            ret = ""
+            for i in contest_list:
+                i_a = i.find_element(By.XPATH, ".//a")
+                ret += (
+                    i.text
+                    + ":"
+                    + i_a.get_attribute("href")
+                    .replace("/contest/", "")
+                    .replace(self.url[:-1], "")
+                    + "\n"
+                )
+            return ret
+        except:
+            raise ContestListNotFoundError("Contest list not found.")
+
+    def ViewContest(self, contest_id):
+        self.driver.get(self.url + "contest/" + contest_id)
+        try:
+            content = self.driver.execute_script("return UiContext.tdoc.content;")
+            language = self.driver.execute_script("return UserContext.viewLang;")
+            content = json.loads(content)
+            print(content[language])
+        except:
+            raise ContestNotFoundError("Contest not found.")
+
+    def UploadFileToProblem(self, problem_id, file_name, file_content):
+        self.driver.get(self.url + "p/" + problem_id + "/files")
+        try:
+            self.driver.find_element(By.NAME, "create_file").click()
+            assert self.driver.switch_to.alert.text == "文件名"
+            alert = self.driver.switch_to.alert
+            alert.send_keys(file_name)
+            alert.accept()
+            self.driver.find_element(By.NAME, "fileContent")
+            self.driver.execute_script(
+                "document.getElementsByName('fileContent')[0].textContent=arguments[0]",
+                file_content,
+            )
+            self.driver.find_element(By.CSS_SELECTOR, ".float-right > .primary").click()
+        except:
+            raise UploadFileFailedError("Failed to upload file.")
+
+    def GotoPage(self, url):
+        self.driver.get(url)
+
+    def RunJavaScript(self, script):
+        return self.driver.execute_script(script)
+
+    def Quit(self):
+        self.driver.quit()

hydrooj-0.0.1/hydrooj/__init__.py

@@ -0,0 +1,5 @@
+from .HydroOJ import HydroOJ
+
+__version__ = "0.0.1"
+__author__ = "YANGRENRUIYRR"
+__all__ = ["HydroOJ"]

hydrooj-0.0.1/pyproject.toml

@@ -0,0 +1,26 @@
+[build-system]
+requires = ["setuptools", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "HydroOJ"
+version = "0.0.1"
+authors = [
+    { name = "YANGRENRUIYRR", email = "yangrenruiyrr@yeah.net" }
+]
+description = "A Python SDK for interacting with HydroOJ via Selenium"
+keywords = ["HydroOJ", "selenium", "online judge"]
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+    "Topic :: Software Development :: Libraries :: Python Modules"
+]
+requires-python = ">=3.8"
+dependencies = [
+    "selenium"
+]
+
+[project.readme]
+file = "README.md"
+content-type = "text/markdown"

hydrooj-0.0.1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+