petlja-api 0.0.1__py3-none-any.whl

petlja_api/__init__.py

@@ -0,0 +1,4 @@
+from .auth import login
+from .competition import create_competition, get_competition_id, get_added_problem_ids, add_problem, upload_scoring, add_language
+from .problem import get_problem_id, get_problem_name, create_problem, upload_testcases, upload_statement
+from .submit import submit_solution

petlja_api/auth.py

@@ -0,0 +1,48 @@
+from getpass import getpass
+
+import requests
+from bs4 import BeautifulSoup
+
+from .urls import ARENA_URL
+
+
+def _get_credentials():
+    username = input("Petlja username: ")
+    password = getpass("Petlja password: ")
+    return username, password
+
+
+def get_csrf_token(page):
+    soup = BeautifulSoup(page, "html.parser")
+    csrf_token = soup.find("input", {"name": "__RequestVerificationToken"}).get("value")
+    return csrf_token
+
+
+def login(username=None, password=None):
+    if username is None or password is None:
+        username, password = _get_credentials()
+
+    session = requests.Session()
+    mainpage = session.get(ARENA_URL)
+    csrf_token = get_csrf_token(mainpage.text)
+    login_res = session.post(
+        f"{ARENA_URL}/sr-Latn-RS/Account/Login",
+        data={
+            "LoginViewModel.UserNameOrEmail": username,
+            "LoginViewModel.Password": password,
+            "__RequestVerificationToken": csrf_token,
+            "LoginViewModel.RememberMe": "false",
+        },
+        allow_redirects=False,
+    )
+
+    if login_res.status_code == 302:
+        jwt = session.cookies["PetljaCookie"]
+        session.headers.update({"Authorization": f"Bearer {jwt}"})
+        return session
+    elif login_res.status_code == 200:
+        raise PermissionError("Wrong username or password")
+    else:
+        raise RuntimeError(
+            f"Unknown error: login failed with status code {login_res.status_code}"
+        )

petlja_api/competition.py

@@ -0,0 +1,131 @@
+import json
+import re
+from datetime import datetime
+
+from bs4 import BeautifulSoup
+
+from .auth import get_csrf_token
+from .problem import get_problem_name
+from .urls import ARENA_URL, CPANEL_URL, PETLJA_URL
+from .submit import LANGUAGE_IDS
+
+
+def get_competition_id(session, alias):
+    page = session.get(f"{ARENA_URL}/competition/{alias}")
+    if (
+        page.status_code == 404
+        or 'error' in page.url
+    ):
+        raise ValueError(f"Competition with alias {alias} does not exist")
+
+    soup = BeautifulSoup(page.text, "html.parser")
+    competition_id = soup.find("button", attrs={"id": "ciRun"})["data-competition-id"]
+    return competition_id
+
+
+def get_added_problem_ids(session, competition_id):
+    page = session.get(f"{CPANEL_URL}/CompetitionTasks/{competition_id}")
+    soup = BeautifulSoup(page.text, "html.parser")
+    # Get object viewModel from inline script in html which contains data about added problems
+    # FIXME regex hack, should be replaced with a proper parser
+    regex = re.compile(r"var viewModel=({.*?});\n")
+    match = regex.search(soup.prettify()).group(1)
+    # Can be parsed as json since it is a javascript object
+    data = json.loads(match)
+    problem_ids = [str(problem["problemId"]) for problem in data["problems"]]
+    return problem_ids
+
+
+def create_competition(
+    session, name, alias=None, description=None, start_date=None, end_date=None
+):
+    if alias is None:
+        alias = ""
+    if description is None:
+        description = ""
+    if start_date is None:
+        start_date = datetime.now()
+    if end_date is None:
+        end_date = ""
+
+    regex = re.compile(r"^[a-z0-9-]+$")
+    if not regex.match(alias):
+        raise NameError(
+            f"Invalid alias {alias}: must contain only lowercase alphanumeric characters and dashes"
+        )
+
+    url = f"{CPANEL_URL}/CreateCompetition"
+    page = session.get(url)
+    csrf_token = get_csrf_token(page.text)
+    resp = session.post(
+        url,
+        data={
+            "Name": name,
+            "Alias": alias,
+            "Description": description,
+            "StartDate": start_date,
+            "EndDate": end_date,
+            "HasNotEndDate": [True, False],  # Not sure what this field does
+            "__RequestVerificationToken": csrf_token,
+        },
+        allow_redirects=False,
+    )
+
+    if resp.status_code == 302:
+        header_loc = resp.headers["Location"]  # /cpanel/CompetitionSettings/:comp_id
+        comp_id = header_loc.split("/")[-1]
+        return comp_id
+    elif resp.status_code == 200:
+        raise ValueError("Competition alias already exists")
+    else:
+        raise RuntimeError(f"Unknown error: {resp.status_code}")
+
+
+def add_problem(session, competition_id, problem_id):
+    already_added = get_added_problem_ids(session, competition_id)
+    if problem_id in already_added:
+        return
+
+    url = f"{PETLJA_URL}/api/dashboard/competitions/problems/add"
+    problem_name = get_problem_name(session, problem_id)
+    resp = session.post(
+        url,
+        json={
+            "competitionId": competition_id,
+            "problemId": problem_id,
+            "name": problem_name,
+            # "sortOrder": 0, # Seems to be optional
+        },
+    )
+
+    # TODO: Check for errors
+
+
+def upload_scoring(session, competition_id, problem_id, scoring_path):
+    url = f"{PETLJA_URL}/api/dashboard/competitions/problems/addGraderHints"
+    with open(scoring_path) as scoring_file:
+        scoring = scoring_file.read()
+    resp = session.post(
+        url,
+        json={
+            "competitionId": competition_id,
+            "problemId": problem_id,
+            "hintsYML": scoring,
+            "testCaseCount": 100,  # FIXME Count number of testcases from scoring file
+        },
+    )
+
+    errors = resp.json()["errors"]
+    if errors:
+        raise ValueError(f"Error uploading scoring, petlja response: {errors[0]}")
+
+def add_language(session, competition_id, extension):
+    url = f'{PETLJA_URL}/api/dashboard/competitions/programmingLanguages/add'
+    res = session.post(
+        url,
+        json={
+            "competitionId": competition_id,
+            "languageId": LANGUAGE_IDS[extension]
+        }
+    )
+    # TODO Check for errors

petlja_api/problem.py

@@ -0,0 +1,98 @@
+from bs4 import BeautifulSoup
+
+from .auth import get_csrf_token
+from .urls import CPANEL_URL, PROBLEMS_URL
+
+
+def get_problem_id(session, alias):
+    page = session.get(f"{PROBLEMS_URL}/{alias}")
+    if page.status_code == 404:
+        raise ValueError(f"Problem with alias {alias} does not exist")
+
+    soup = BeautifulSoup(page.text, "html.parser")
+    problem_id = soup.find("button", attrs={"class": "btn-solution-submit"})[
+        "data-problem-id"
+    ]
+    return problem_id
+
+
+def get_problem_name(session, problem_id):
+    page = session.get(f"{CPANEL_URL}/EditProblem/{problem_id}")
+    if page.status_code == 404:
+        raise ValueError(f"Problem with id {problem_id} does not exist")
+
+    soup = BeautifulSoup(page.text, "html.parser")
+    problem_name = soup.find("input", attrs={"id": "Problem_Name"})["value"]
+    return problem_name
+
+
+def create_problem(session, name, alias):
+    if not alias or not alias.isalnum() or not alias.islower():
+        raise NameError(
+            f"Invalid problem alias {alias}: must be alphanumeric and lowercase"
+        )
+
+    create_problem_page = session.get(f"{CPANEL_URL}/CreateTask")
+    csrf_token = get_csrf_token(create_problem_page.text)
+    resp = session.post(
+        f"{CPANEL_URL}/CreateTask",
+        data={
+            "Name": name,
+            "UniqueId": alias,
+            "Type": "0",
+            "__RequestVerificationToken": csrf_token,
+        },
+        allow_redirects=False
+    )
+    if resp.status_code == 302:
+        return get_problem_id(session, alias)
+    elif resp.status_code == 200:
+        raise ValueError("Problem alias already exists")
+    else:
+        raise RuntimeError(
+            "Unknown error while creating problem (status code {resp.status_code})"
+        )
+
+
+def check_testcase_upload(page):
+    soup = BeautifulSoup(page.text, "html.parser")
+    error = soup.find("div", attrs={"class": "validation-summary-errors"})
+    if error:
+        errmsg = f"Testcase upload failed: {error.text.strip()}"
+        if error.text.strip() == "UserIdNotFound":
+            errmsg += " (Is the zip file in the correct format?)"
+        raise ValueError(errmsg)
+
+
+def upload_testcases(session, problem_id, testcases_path):
+    page = session.get(f"{CPANEL_URL}/EditProblem/{problem_id}?tab=testcases")
+    csrf_token = get_csrf_token(page.text)
+    with open(testcases_path, "rb") as zipfile:
+        resp = session.post(
+            f"{CPANEL_URL}/EditProblem/{problem_id}",
+            files={"TestCases": zipfile},
+            data={
+                "PostAction": "EditTestCases",
+                "__RequestVerificationToken": csrf_token,
+            },
+        )
+    # Have to scrape the response page to check for errors
+    # because the response is 302 even if there is an error
+    check_testcase_upload(resp)
+
+
+def upload_statement(session, problem_id, statement_path):
+    page = session.get(f"{CPANEL_URL}/EditProblem/{problem_id}?tab=statement")
+    csrf_token = get_csrf_token(page.text)
+    with open(statement_path, encoding="utf-8") as statement:
+        resp = session.post(
+            f"{CPANEL_URL}/EditProblem/{problem_id}",
+            data={
+                "Problem.ProblemStatementMD": statement.read(),
+                "PostAction": "EditStatement",
+                "Problem.MDSupported": "true",
+                "__RequestVerificationToken": csrf_token,
+            },
+            allow_redirects=False,
+        )
+    # TODO: check for errors

petlja_api/submit.py

@@ -0,0 +1,54 @@
+from .urls import ARENA_URL
+import time
+
+# Weird Petlja API language IDs
+LANGUAGE_IDS = {
+    ".c": 10,
+    ".cs": 1,
+    ".cpp": 2,
+    ".java": 3,
+    ".m": 7,
+    ".pas": 4,
+    ".py": 9,
+}
+
+TIMEOUT = 30
+
+
+def submit_solution(session, competition_id, problem_id, source_path):
+    with open(source_path) as source_file:
+        source = source_file.read()
+
+    extension = source_path.suffix
+    submit_res = session.post(
+        f"{ARENA_URL}/api/competition/submit-competition-problem",
+        json={
+            "competitionId": competition_id,
+            "problemId": problem_id,
+            "source": source,
+            "languageId": LANGUAGE_IDS[extension],
+        },
+    )
+    success = submit_res.json()["succeeded"]
+    if not success:
+        error = submit_res.json()["errors"][0]["description"]
+        raise Exception(error)
+    submission_id = submit_res.json()["value"]
+    # Polling the server every x seconds
+    # Better solution may exist
+    tries = 0
+    while tries < TIMEOUT:
+        submission_data = session.post(
+            f"{ARENA_URL}/api/competition/submissions",
+            json={
+                "competitionId": competition_id,
+                "idStamp": submission_id,
+                "loadNew": True,
+            },
+        )
+        score = submission_data.json()["value"]["item1"][0]["score"]
+        if score != '-':
+            return score
+        time.sleep(1)
+
+    raise TimeoutError

petlja_api/urls.py

@@ -0,0 +1,4 @@
+ARENA_URL = "https://arena.petlja.org"
+PETLJA_URL = "https://petlja.org"
+CPANEL_URL = f"{PETLJA_URL}/sr-Latn-RS/cpanel"
+PROBLEMS_URL = f"{PETLJA_URL}/sr-Latn-RS/problems"

petlja_api-0.0.1.dist-info/METADATA

@@ -0,0 +1,40 @@
+Metadata-Version: 2.3
+Name: petlja_api
+Version: 0.0.1
+Summary: A python library for interacting with the petlja.org API
+Project-URL: Homepage, https://github.com/PavleSekesan/petlja_api
+Author-email: Pavle Sekešan <pavlesekesan@gmail.com>
+License-File: LICENSE
+Requires-Python: >=3.7
+Requires-Dist: beautifulsoup4==4.12.2
+Requires-Dist: requests==2.31.0
+Description-Content-Type: text/markdown
+
+# petlja_api
+
+A python library for interacting with the [petlja.org](https://petlja.org/) API.
+
+## Installation
+```
+pip install petlja_api
+```
+
+## Basic usage
+
+```py
+import petlja_api as petlja
+
+session = petlja.login()
+
+# Create problem
+prob_id = petlja.create_problem(session, name="My Problem", alias="my-prob")
+petlja.upload_testcases(session, prob_id, "my-prob/testcases.zip")
+petlja.upload_statement(session, prob_id, "my-prob/statement.md")
+
+# Create competition
+comp_id = petlja.create_competition(session, name="My Competition", alias="my-comp")
+petlja.add_problem(session, comp_id, prob_id)
+
+# Upload solution
+petlja.score = submit(session, prob_id, "my-prob/sol.cpp", comp_id)
+```

petlja_api-0.0.1.dist-info/RECORD

@@ -0,0 +1,10 @@
+petlja_api/__init__.py,sha256=FoR5XbbCMOReJx9SpUv6Meut21WesaeaDSou9Dt0oNk,295
+petlja_api/auth.py,sha256=m6hvN_hq73bpaawokdugI4TChWxWNh9E3Een1m8o0sE,1442
+petlja_api/competition.py,sha256=Obzh90yvaP2AKcj0UDr_WwQ1nwiG60rUEyn3gjdnnEY,4209
+petlja_api/problem.py,sha256=RhIgkCCWG33JKD9mCnYGGsBHEXCUNQKJvvyg2xcU3OQ,3457
+petlja_api/submit.py,sha256=yMY_VdS1pktL47yzUPvp66RA01ezEZOFUatF0Kt_Sqs,1436
+petlja_api/urls.py,sha256=ol2xDoGg8pZS3I_kXrAMfsHX_NI54XtDlCI9qfI3u84,170
+petlja_api-0.0.1.dist-info/METADATA,sha256=V4zxVbggFapS5n7wguNaqOoat-poeg66k630462tb68,1088
+petlja_api-0.0.1.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+petlja_api-0.0.1.dist-info/licenses/LICENSE,sha256=vjjmZ_TFg_A8t-6PHeFUaEo1pSrRtnO59PBzc527jS0,1071
+petlja_api-0.0.1.dist-info/RECORD,,

petlja_api-0.0.1.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.25.0
+Root-Is-Purelib: true
+Tag: py3-none-any

petlja_api-0.0.1.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2023 Pavle Sekešan
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.