HydroOJ 0.0.1__py3-none-any.whl

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/__init__.py

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

hydrooj-0.0.1.dist-info/METADATA

@@ -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.dist-info/RECORD

@@ -0,0 +1,6 @@
+hydrooj/HydroOJ.py,sha256=Kau1b818Zr-AEqnJqwvZtKmKsdUoOPKRoifhd91bNPI,9704
+hydrooj/__init__.py,sha256=iQbvB4S_hcTY8SL2Ps3igjKbcpnqc6dEt4oEkx-4EHM,108
+hydrooj-0.0.1.dist-info/METADATA,sha256=RBcpLNTnxF1047Hkfp00xTMP7HL3pYZkgqoCKB3JL68,3740
+hydrooj-0.0.1.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+hydrooj-0.0.1.dist-info/top_level.txt,sha256=h29DxaZ6m7a1vzKYl0vdWogrgpq8ovsWN_gqmeBfgm8,8
+hydrooj-0.0.1.dist-info/RECORD,,

hydrooj-0.0.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

hydrooj-0.0.1.dist-info/top_level.txt

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