57 Commits

Author SHA1 Message Date
813519f642 ujpdated requirements 2025-02-12 18:34:45 +01:00
edabec3e7c Fixed Issue #17 by adding a browser pop-up dialog 2025-02-12 18:30:09 +01:00
194daf89eb Merge pull request #16 from Code-For-Groningen/lol
Update README.md
2024-12-03 17:27:17 +01:00
172e2a0ebf Update README.md 2024-12-03 17:26:57 +01:00
213a4d5471 Problematic line fix 2024-12-02 18:29:13 +01:00
8d0164aa42 Fixed problematic line 2024-12-02 18:27:48 +01:00
5ba67e3b51 Replaced | syntax with Optional (Python types <3) 2024-12-02 18:27:33 +01:00
96488ac69c reformatted to make pylint happy
TODO: get rid of the ignore comments - just fix it
2024-12-02 18:21:54 +01:00
060e5df43e Update pylint.yml 2024-12-02 18:03:53 +01:00
ad3d95a074 Update python-publish.yml 2024-12-02 17:33:17 +01:00
9740e37b64 Create python-publish.yml 2024-12-02 17:27:33 +01:00
569ac0c048 Create pylint.yml 2024-12-02 17:27:09 +01:00
aa7b91de0d New version fix 2024-12-02 17:07:56 +01:00
5c3e884a8b Made program wait until case is done 2024-12-02 17:01:04 +01:00
c14f87aecc Removed accidental overwrite of submit method 2024-12-02 17:00:06 +01:00
6a781ad238 Fixed get_course_by_tag name logic error (#14) 2024-11-21 11:17:10 +01:00
a1104522f1 Merge branch 'main' of github:Code-For-Groningen/temmies 2024-11-21 11:14:53 +01:00
fb8b5cd454 Unignored setup 2024-11-21 11:14:49 +01:00
1367fd667f Smaller temmie 2024-11-20 20:02:29 +01:00
1516ef74be Deliberately lowered resolution of roman scale 2024-11-20 20:01:47 +01:00
c37edb59c6 Updated logo 2024-11-20 19:59:42 +01:00
82a072ee14 Updated api to reflect new changes 2024-11-18 20:17:26 +01:00
3d9ba063ce Fixed accidental format break 2024-11-18 20:17:09 +01:00
82bafa1925 Removed concrete deprecation warning 2024-11-18 20:05:53 +01:00
52d9d86260 Used API instead of scraping to find courses from year 2024-11-18 20:05:12 +01:00
da4705b56a Used the API for year logic 2024-11-18 20:04:02 +01:00
f6e6bc28d2 Prettified submission code 2024-11-18 20:03:31 +01:00
2fa3bfbad8 Moved all downloading logic, submitting and status/tc table parsing into base Group class
Also created 2 methods to get files and test cases without downloading
2024-11-18 20:03:08 +01:00
9f99df54d8 Moved most logic to base Group class 2024-11-18 19:58:17 +01:00
514fcd2438 Moved disable warnings to __init__ 2024-11-18 19:57:48 +01:00
812f384c3d Course group creation logic updated 2024-11-18 19:57:37 +01:00
34bf29d274 Made create_group_from_url a static class method 2024-11-18 16:02:34 +01:00
f3d9a987f6 Removed circular type delcarations
Python is crazy
2024-11-18 15:55:43 +01:00
cb6ec3cc1c Updated deprecation messages 2024-11-18 15:51:59 +01:00
0ff3f28978 Added methods to create an exercise group from a link 2024-11-18 15:45:19 +01:00
de66e9762e Added support for deprecated methods 2024-11-18 15:30:35 +01:00
1a950c0eb2 Update README 2024-11-17 22:19:50 +01:00
a3a9f5dfee Updated index 2024-11-17 22:06:06 +01:00
461b506be5 Added requirements 2024-11-17 21:57:33 +01:00
a0aeca87b1 Updated changelog and doc with new password storage method 2024-11-17 21:57:07 +01:00
17f0bd237a Solved #11 by using system's keyring 2024-11-17 21:53:02 +01:00
2378be4d42 Fixed slight bug with unused method 2024-11-17 21:52:11 +01:00
6a63abf7ae Expose only Themis on package-level 2024-11-17 19:50:19 +01:00
019e2c181e Updated docs ot include changelog and reflect upon recent changes 2024-11-17 19:50:05 +01:00
135420d488 Naming fix
All methods are prefixed with get_
2024-11-17 19:36:28 +01:00
aab574cdb0 Refactor to use base class group for ExerciseGroup and Course to avoid repeated logic 2024-11-17 19:35:16 +01:00
9d92db4644 Updated link to docs 2024-09-22 16:15:13 +02:00
2c2d39c6dd Fixed stupid docs mistake 2024-06-03 22:55:36 +02:00
54e086c93a Slight update in docs. Released version 1.0.21. 2024-04-21 20:37:07 +02:00
667a388da4 Updated files. getting ready for publishing. Some issues 2024-04-21 02:08:58 +02:00
2daee84d4f Fixed mistakes in docs. Elaborated. 2024-04-21 00:29:49 +02:00
cd5e9b2c8d Added submissions. Updated docs. 2024-04-20 21:35:09 +02:00
de6262a4bb removed class map, as it doesn't correspond to current development. 2024-04-10 19:14:35 +02:00
8325a30af8 Added link to badges. 2024-04-10 18:54:38 +02:00
0f92d05bec Added badges :) 2024-04-10 18:53:14 +02:00
2785ec86e9 Updated docs to reflect upon code changes 2024-04-10 18:48:21 +02:00
78aade7c8c Conformed to naming conventions. Pepped the SHIT out of the code. Will reflect in documentation now. 2024-04-10 18:43:03 +02:00
29 changed files with 1075 additions and 528 deletions

25
.github/workflows/pylint.yml vendored Normal file
View File

@ -0,0 +1,25 @@
name: Pylint
on: [push]
jobs:
build:
runs-on: ubuntu-latest
strategy:
matrix:
python-version: ["3.9", "3.10", "3.11", "3.12"]
steps:
- uses: actions/checkout@v4
- name: Set up Python ${{ matrix.python-version }}
uses: actions/setup-python@v3
with:
python-version: ${{ matrix.python-version }}
- name: Install dependencies
run: |
python -m pip install --upgrade pip
pip install pylint
pip install .
- name: Analyzing the code with pylint
run: |
pylint temmies

58
.github/workflows/python-publish.yml vendored Normal file
View File

@ -0,0 +1,58 @@
name: Upload Python Package
on:
release:
types: [published]
permissions:
contents: read
jobs:
release-build:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- uses: actions/setup-python@v5
with:
python-version: "3.x"
- name: Install dependencies
run: |
python -m pip install --upgrade pip
python -m pip install setuptools wheel twine
- name: Build release distributions
run: |
python setup.py bdist_wheel
- name: Upload distributions
uses: actions/upload-artifact@v4
with:
name: release-dists
path: dist/
pypi-publish:
runs-on: ubuntu-latest
needs:
- release-build
permissions:
id-token: write
environment:
name: pypi
steps:
- name: Retrieve release distributions
uses: actions/download-artifact@v4
with:
name: release-dists
path: dist/
- name: Publish release distributions to PyPI
env:
TWINE_PASSWORD: ${{ secrets.PYPI_API_TOKEN }}
run: |
python -m pip install twine
twine upload dist/*

3
.gitignore vendored
View File

@ -1,6 +1,8 @@
# Config - Testing # Config - Testing
config.py config.py
tests/ tests/
pathfinding/
test.py
#Doc env #Doc env
.docs_env .docs_env
@ -329,3 +331,4 @@ cython_debug/
# and can be added to the global gitignore or merged into this file. For a more nuclear # and can be added to the global gitignore or merged into this file. For a more nuclear
# option (not recommended) you can uncomment the following to ignore the entire idea folder. # option (not recommended) you can uncomment the following to ignore the entire idea folder.
#.idea/ #.idea/
project-hierarchy.txt

View File

@ -1,26 +1,26 @@
<p align="center"> <p align="center">
<img src="docs/img/rugemmie.gif" /> <img src="docs/img/temmie.png" width= 200px/>
</p>
<p align="center">
<a href="https://temmies.confest.im"><img alt="Read the Docs" src="https://img.shields.io/readthedocs/temmies"></a>
<img alt="GitHub" src="https://img.shields.io/github/license/Code-For-Groningen/temmies">
</p> </p>
A python library which interacts with themis. Uses bs4. I'll try to end development on a somewhat working state. A python library which interacts with [Themis](https://themis.housing.rug.nl/). Uses bs4. I'll try to end development on a somewhat working state.
## Intended Features ## Intended Features
* [x] Log in * [x] Log in
* [x] Submit * [x] Submit
* [x] Bulk download of test cases and files * [x] Bulk download of test cases and files
* [ ] Submission status * [x] Submission status
* [ ] Classes, methods and attributes described in the map below
## Docs ## Docs
[here](http://temmies.rtfd.io/). - [Click here](http://temmies.confest.im/).
## Class map
![map](images/roadmap.png)
## Possible continuations ## Possible continuations
* [ ] Discord bot * Discord bot
* [ ] CLI program * [CLI program](https://github.com/Code-For-Groningen/temmies-cli)
## Thanks to ## Thanks to
* [Glitchcat](https://glitchcat.github.io/themis-api/), cool docs bro. * [Glitchcat](https://glitchcat.github.io/themis-api/), cool docs bro.
* [Mo](https://github.com/Stylo2k), damn bruh you actually work for Themis * [Mo](https://github.com/Stylo2k), damn bruh you actually work for Themis

View File

@ -1,33 +1,35 @@
# Classes # Classes
--- ---
## `Themis` ## `Themis`
Creates the initial connection to Themis. Creates the initial connection to Themis.
### Usage ### Usage
```python ```python
from temmies.Themis import Themis from temmies.themis import Themis
themis = Themis("s-number", "password") themis = Themis("s-number")
``` ```
On the first run, you will be prompted for your password. Then, on the next run(s), you will be able to log in automatically, as the password is stored in the system keyring. If you want to delete it [click here](https://www.google.com/search?hl=en&q=delete%20a%20password%20from%20keyring).
### Methods ### Methods
#### `login()` #### `login()`
Logs in to Themis. Runs automatically when the class is initialized. Logs in to Themis. Runs automatically when the class is initialized.
#### `getYear(start, end)` #### `get_year(year_path)`
Returns an instance of a [`Year`](#year)(academic year) between `start` and `end`. Returns an instance of a [`Year`](#year) for the academic year specified by `year_path`.
```python ```python
year = themis.getYear(2023, 2024) year = themis.get_year(2023, 2024)
``` ```
#### `allYears()` #### `all_years()`
Returns a list of `Year` instances corresponding to all years visible to the user. Returns a list of `Year` instances corresponding to all years visible to the user.
```python ```python
years = themis.allYears() years = themis.all_years()
``` ```
<sub> I don't see why you would need this, but it's here. </sub>
---- ----
@ -35,22 +37,29 @@ years = themis.allYears()
### Usage ### Usage
```python ```python
year = themis.getYear(2023, 2024) year = themis.get_year(2023, 2024)
``` ```
### Methods ### Methods
#### `getCourse(courseName)` #### `get_course(course_title)`
Returns an instance of a [`Course`](#course) with the name `courseName`. Returns an instance of a [`Course`](#course) with the title `course_title`.
```python ```python
pf = year.getCourse("Programming Fundamentals (for CS)") pf = year.get_course("Programming Fundamentals (for CS)")
``` ```
#### `allCourses()` #### `get_course_by_tag(course_tag)`
Returns an instance of a [`Course`](#course) using the course identifier `course_tag`.
```python
ai_course = year.get_course_by_tag("adinc-ai")
```
#### `all_courses()`
Returns a list of `Course` instances corresponding to all courses visible to the user in a given `Year`. Returns a list of `Course` instances corresponding to all courses visible to the user in a given `Year`.
```python ```python
courses = year.allCourses() courses = year.all_courses()
``` ```
---- ----
@ -58,128 +67,121 @@ courses = year.allCourses()
## `Course` ## `Course`
### Usage ### Usage
```python ```python
pf = year.get_course("Programming Fundamentals (for CS)")
pf = year.getCourse("Programming Fundamentals (for CS)") assignments = pf.get_groups()
print(pf.info) # <- course info attribute
assignments = pf.getGroups()
``` ```
### Methods ### Methods
#### `getGroups(full=False)` #### `get_groups(full=False)`
Returns a list of `ExerciseGroup` instances corresponding to all exercise groups visible to the user in a given `Course`. Default argument is `full=False`, which will only return the (name, link) of each exercise and folder in the group. If `full=True`, it will traverse the whole course. Returns a list of `ExerciseGroup` or `Group` instances corresponding to all items visible to the user in a given `Course`. The default argument is `full=False`, which will only return the top-level (name, link) of each item. If `full=True`, it will traverse the whole course.
You can traverse the course in both cases, although in different ways.
When you have fully traversed the course, you can access everything via indices and the `exercises` and `folders` attributes of the `ExerciseGroup` instances:
```python ```python
ai_group = ai_course.getGroups(full=True) ai_groups = ai_course.get_groups(full=True)
exercise = ai_group[7].exercises[1] # Week 11 -> Suitcase packing exercise = ai_groups[7].exercises[1]
exercise.submit("suitcase.py", silent=False)``` exercise.submit(["solution.py"], silent=False)
``` ```
This is equivalent to the case in which we don't traverse the full course using `getGroup` like so: #### `get_group(name, full=False)`
Returns an instance of an `ExerciseGroup` or `Group` with the name `name`. The default argument is `full=False`, which will only return the (name, link) of the group. If `full=True`, it will traverse the whole group.
```python ```python
ai_group = ai_course.getGroup("Week 11") week1 = pf.get_group("Week 1")
exercise = ai_group.getGroup("Suitcase packing")
exercise.submit("suitcase.py", silent=False)
``` ```
### `getGroup(name, full=False)` #### `create_group(item_data)`
Returns an instance of an `ExerciseGroup` with the name `name`. Default argument is `full=False`, which will only return the (name, link) of each exercise and folder in the group. If `full=True`, it will traverse the whole group. Creates and returns a `Group` or `ExerciseGroup` instance based on `item_data`.
```python ```python
week1 = pf.getGroup("Week 1") group = course.create_group(item_data)
``` ```
----
## `Group`
Represents an item in Themis, which can be either a folder (non-submittable) or an assignment (submittable).
### Methods
#### `get_items()`
Returns all items (groups and assignments) under this group.
```python
items = week1.get_items()
```
#### `get_item_by_title(title)`
Returns a single item by its title (case-insensitive).
```python
item = week1.get_item_by_title("Exercise 2")
```
#### `get_status(text=False)`
Retrieves the status of the group. When `text=True`, returns the status as strings. Otherwise, returns submission objects or strings.
```python
status = group.get_status()
leading_submission = status["leading"]
```
#### `download_files(path=".")`
Downloads all files available for this group to a directory `path`. Defaults to the current directory.
```python
group.download_files()
```
#### `download_tcs(path=".")`
Downloads all test cases for this group to a directory `path`. Defaults to the current directory.
```python
group.download_tcs()
```
#### `submit(files, judge=True, wait=True, silent=True)`
Submits the files to the group. Default arguments are `judge=True`, `wait=True`, and `silent=True`.
```python
group.submit(["solution.py"], silent=False)
```
----
## `ExerciseGroup` ## `ExerciseGroup`
Setting the `full` flag to `True` will traverse the whole course. Represents a submittable exercise. Inherits from `Group`.
You can traverse the course in both cases ### Additional Methods
* Both folders and exercises are represented as `ExerciseGroup` instances. #### `submit(files)`
* Folders will have the `amExercise` attribute set to `False`. Submits files to the exercise. Raises an error if the item is not submittable.
* Folders can have the `downloadFiles` method called on them.
* Exercises can have the `submit`, `downloadFiles` and `downloadTCs` method called on them.
### Example of folder traversal
Let's say we have a folder structure like this:
```
- Course Name
- Week 1
- Exercise 1
- Exercise 2
- Part 1
- Part 2
- Week 2
- Exercise 1
- Exercise 2
```
And we want to get to `Part 2` of `Week 1`'s `Exercise 2`. We would do this:
```python ```python
pf = year.getCourse("Programming Fundamentals (for CS)") exercise.submit(["solution.py"])
assignments = pf.getExerciseGroups()
week1 = assignments[0] # Week 1
exercise2 = week1.folders[1] # Exercise 2
part2 = exercise2.exercises[1] # Part 2
# Or, if you dont want to traverse the whole course:
week1 = pf.getGroup("Week 1")
exercise2 = week1.getGroup("Exercise 2")
part2 = exercise2.getGroup("Part 2")
``` ```
----
## `Submission`
Represents a submission for a specific exercise.
### Methods ### Methods
#### `downloadFiles(path=".")` #### `get_test_cases()`
Downloads all files in the exercise group to a directory `path`. Defaults to the current directory. Returns a dictionary of test cases and their statuses.
```python ```python
assignment.downloadFiles() test_cases = submission.get_test_cases()
``` ```
#### `downloadTCs(path=".")` #### `get_info()`
Downloads all test cases in the exercise group to a directory `path`. Defaults to the current directory. Returns detailed information about the submission.
```python ```python
assignment.downloadTCs() info = submission.get_info()
``` ```
#### getGroup(name, full=False) #### `get_files()`
This is used when you want to traverse the course dynamically(not recurse through the whole thing). Of course, you can use it even if you've traversed the whole course, but that would overcomplicate things. Returns a list of uploaded files in the format `(name, URL)`.
```python ```python
# Week 1 -> Exercise 2 -> Part 2 files = submission.get_files()
week1 = pf.getGroups("Week 1")
exercise2 = week1.getGroup("Exercise 2")
part2 = exercise2.getGroup("Part 2")
# This is equivalent to(but faster than):
week1 = pf.getGroups("Week 1", full=True)
exercise2 = week1[1]
part2 = exercise2[1]
``` ```
#### `submit(files)`
Submits the files to the exercise group. Default arguments are `judge=True`, `wait=True` and `silent=True`. `judge` will judge the submission instantly, and `wait` will wait for the submission to finish. Turning off `silent` will print the submission status dynamically.
```python
suitcase[7].exercises[1].submit("suitcase.py", silent=False)
>>> 1:
>>> 2:
>>> 3:
>>> 4:
>>> 5:
>>> 6:
>>> 7:
>>> 8:
>>> 9:
>>> 10:
```

21
docs/changelog.md Normal file
View File

@ -0,0 +1,21 @@
## **Changelog**
### **Version 1.1.0**
#### **Documentation**
- Fixed method signatures to align with actual functionality.
- Updated `get_status` to properly handle `Submission` instances.
- Ensured all class and method examples are consistent with the codebase.
#### **Codebase**
- Prepended `get_` to all methods in `Submission`
- Created base `Group` from which `Course` and `ExerciseGroup` inherit.
- Using system keyring to store passwords (Issue #11)
### **Version 1.2.0**
#### **Codebase**
- Moved all methods related to downloading files (including test cases) to `Group`.
- Created `get_test_cases` and `get_files` methods in `Group`.
- We are now using the [API](https://themis.housing.rug.nl/api/navigation/2023-2024) (which mysteriously appeared) to get the year/course structure.

Binary file not shown.

Before

Width:  |  Height:  |  Size: 190 KiB

BIN
docs/img/temmie.png Normal file

Binary file not shown.

After

Width:  |  Height:  |  Size: 38 KiB

View File

@ -1,5 +1,5 @@
# Temmies! # Temmies!
<center>![Temmie](img/rugemmie.gif)</center> <center>![Temmie](img/temmie.png)</center>
## What is this? ## What is this?
@ -18,22 +18,22 @@ pip install temmies
## Example Usage ## Example Usage
```python ```python
import temmies from temmies.themis import Themis
# Log in # Log in
themis = temmies.Themis("s-number", "password") themis = Themis("s-number") # You will be prompted for your password
# Get a year # Get a year
year = themis.getYear(2023, 2024) year = themis.get_year("2023-2024")
# Get a course # Get a course
pf = year.getCourse("Programming Fundamentals (for CS)") course = year.get_course("Programming Fundamentals (for CS)")
# Get an assignment # Get an assignment
pf_assignment = pf.getGroup("Assignment 1") assignment = course.get_group("Assignment 1")
# Get a specific exercise # Submit 2 files
exercise = pf_assignment.getGroup("Exercise 1") assignment.submit(["among.c", "us.py"])
``` ```

Binary file not shown.

Before

Width:  |  Height:  |  Size: 141 KiB

View File

@ -3,4 +3,5 @@ nav:
- Temmies: index.md - Temmies: index.md
- API Reference: api.md - API Reference: api.md
- About: about.md - About: about.md
- Change Log: changelog.md
theme: readthedocs theme: readthedocs

21
requirements.txt Normal file
View File

@ -0,0 +1,21 @@
attrs==25.1.0
beautifulsoup4==4.12.3
bs4==0.0.2
certifi==2024.8.30
cffi==1.17.1
charset-normalizer==3.4.0
cryptography==43.0.3
idna==3.10
jaraco.classes==3.4.0
jaraco.context==6.0.1
jaraco.functools==4.1.0
jeepney==0.8.0
keyring==25.5.0
lxml==5.3.0
more-itertools==10.5.0
pycparser==2.22
requests==2.32.3
SecretStorage==3.3.3
selenium==4.28.1
soupsieve==2.6
urllib3==2.2.3

31
setup.py Normal file
View File

@ -0,0 +1,31 @@
from setuptools import find_packages, setup
with open("README.md", "r") as f:
l_description = f.read()
setup(
name="temmies",
version="1.2.121",
packages=find_packages(),
description="A wrapper for the Themis website",
long_description=l_description,
long_description_content_type="text/markdown",
url="https://github.com/Code-For-Groningen/temmies",
author="Boyan K.",
author_email="boyan@confest.im",
license="GPLv3",
classifiers=[
"Development Status :: 4 - Beta",
"Intended Audience :: Developers",
"License :: OSI Approved :: GNU General Public License v3 (GPLv3)",
"Programming Language :: Python :: 3.9",
],
install_requires=[
"urllib3",
"requests",
"lxml",
"beautifulsoup4",
"keyring"
],
python_requires=">=3.9",
)

View File

@ -1,48 +0,0 @@
# Class to handle courses
from bs4 import BeautifulSoup
from requests import Session
from ExerciseGroup import ExerciseGroup
import re
from exceptions.CourseUnavailable import CourseUnavailable
class Course:
# Extend the Base class init
def __init__(self, url:str, name:str, session:Session, parent):
self.url = url
self.name = name
self.__session = session
self.__parent = parent
self.__request = self.__session.get(self.url)
self.__raw = BeautifulSoup(self.__request.text, 'lxml')
self.__courseAvailable(self.__session.get(self.url))
def __str__(self):
return f"Course {self.name} in year {self.__parent.year}"
def __courseAvailable(self, r):
# Check if we got an error
# print(self.url)
if "Something went wrong" in r.text:
raise CourseUnavailable(message="'Something went wrong'. Course most likely not found. ")
def getGroups(self, full:bool=False):
section = self.__raw.find('div', class_="ass-children")
entries = section.find_all('a', href=True)
return [
ExerciseGroup(
f"https://themis.housing.rug.nl{x['href']}",
x,
self.__session,
self,
full
)
for x in entries]
# BAD: Repeated code!!!!
def getGroup(self, name:str, full:bool=False):
group = self.__raw.find("a", text=name)
if not group:
raise IllegalAction(message=f"No such group found: {name}")
return ExerciseGroup(f"https://themis.housing.rug.nl{group['href']}", group, self.__session, self, full)

View File

@ -1,231 +0,0 @@
from bs4 import BeautifulSoup
from exceptions.IllegalAction import IllegalAction
import re
from json import loads
from time import sleep
class ExerciseGroup:
def __init__(self, url: str, soup, session, parent, full:bool):
self.url = url
self.name = soup.text
self.__prev_raw = soup
self.__session = session
self.__request = self.__session.get(self.url)
self.__raw = BeautifulSoup(self.__request.text, "lxml")
self.__full = full
@property
def amExercise(self):
return "ass-submitable" in self.__prev_raw["class"]
def submit(self):
if not self.amExercise:
raise IllegalAction(message="You are submitting to a folder.")
# Logic for submitting
# Test cases
@property
def testCases(self):
section = self.__raw.find_all("div", class_="subsec round shade")
tcs = []
for div in section:
res = div.find("h4", class_="info")
if not res:
continue
if "Test cases" in res.text:
for case in div.find_all("div", class_="cfg-line"):
if link := case.find("a"):
tcs.append(link)
return tcs
return None
def downloadTCs(self, path="."):
# Logic for downloading test cases(if any)
# In a div with class "subsec round shade", where there is an h4 with text "Test cases"
if not self.amExercise:
raise IllegalAction(message="You are downloading test cases from a folder.")
for tc in self.testCases:
url = f"https://themis.housing.rug.nl{tc['href']}"
print(f"Downloading {tc.text}")
# download the files
with open(f"{path}/{tc.text}", "wb") as f:
f.write(self.__session.get(url).content)
return self.testCases
# Files
@property
def files(self):
details = self.__raw.find("div", id=lambda x: x and x.startswith("details"))
cfg_lines = details.find_all("div", class_="cfg-line")
link_list = []
for line in cfg_lines:
key = line.find("span", class_="cfg-key")
if key and "Downloads" in key.text.strip():
# Extract all links in the cfg-val span
links = line.find_all("span", class_="cfg-val")
for link in links:
a = link.find_all("a")
for a in a:
link_list.append(a)
return link_list if link_list else None
def downloadFiles(self, path="."):
for file in self.files:
print(f"Downloading file {file.text}")
url = f"https://themis.housing.rug.nl{file['href']}"
with open(f"{path}/{file.text}", "wb") as f:
f.write(self.__session.get(url).content)
return self.files
@property
def exercises(self) -> list:
if self.amExercise:
return self
section = self.__raw.find("div", class_="ass-children")
try:
submittables = section.find_all("a", class_="ass-submitable")
except AttributeError:
return None
if not self.__full:
return [(x.text,x['href']) for x in submittables]
return [
ExerciseGroup(
f"https://themis.housing.rug.nl{x['href']}", x, self.__session, self, True
)
for x in submittables
]
@property
def folders(self) -> list:
section = self.__raw.find("div", class_="ass-children")
try:
folders = section.find_all("a", class_="ass-group")
except AttributeError:
return None
if not self.__full:
return [(x.text,x['href']) for x in folders]
return [
ExerciseGroup(f"https://themis.housing.rug.nl{x['href']}", x, self.__session, self, True)
for x in folders
]
# Get by name
def getGroup(self, name:str, full:bool=False, link:str=None):
if link:
return ExerciseGroup(link, self.__prev_raw, self.__session, self, full)
group = self.__raw.find("a", text=name)
if not group:
raise IllegalAction(message=f"No such group found: {name}")
return ExerciseGroup(f"https://themis.housing.rug.nl{group['href']}", group, self.__session, self, full)
# Account for judge
def __raceCondition(self, soup, url:str, verbose:bool):
self.__session.get(url.replace("submission", "judge"))
return self.__waitForResult(url, verbose, [])
def __parseTable(self, soup, url:str, verbose:bool, __printed:list):
cases = soup.find_all('tr', class_='sub-casetop')
fail_pass = {}
i = 1
for case in cases:
name = case.find('td', class_='sub-casename').text
status = case.find('td', class_='status-icon')
if "pending" in status.get("class"):
return self.__raceCondition(soup,url,verbose)
# queued status-icon
if "queued" in status.get("class"):
sleep(1) # <- 🗿
return self.__waitForResult(url, verbose, __printed)
if "Passed" in status.text:
fail_pass[int(name)] = True
if int(name) not in __printed and verbose == True:
print(f"{name}: ✅")
elif "Wrong output" in status.text:
fail_pass[int(name)] = False
if int(name) not in __printed and verbose == True:
print(f"{name}: ❌")
elif ("No status" or "error") in status.text:
fail_pass[int(name)] = None
if int(name) not in __printed and verbose == True:
print(f"{name}:🐛")
__printed.append(int(name))
i += 1
return fail_pass
def __waitForResult(self, url:str, verbose:bool, __printed:list):
# This waits for result and returns a bundled info package
r = self.__session.get(url)
soup = BeautifulSoup(r.text, "lxml")
return self.__parseTable(soup, url, verbose, __printed)
# Submit
def submit(self, files: list, judge=True, wait=True, silent=True):
# Find the form with submit and store the action as url
# Store then the data-suffixes as file_types - dictionary
form = self.__raw.find("form")
if not form:
raise IllegalAction(message="You cannot submit to this assignment.")
url = "https://themis.housing.rug.nl" + form["action"]
file_types = loads(form["data-suffixes"])
if isinstance(files, str):
temp = []
temp.append(files)
files = temp
# Package the files up into files[]
# DEBUG: Uncomment for better clarity
# print("Submitting files:")
# [print(f) for f in files]
packaged_files = []
data = {}
found_type = ""
for file in files:
for t in file_types:
if t in file:
found_type = file_types[t]
break
if not found_type:
raise IllegalAction(message="Illegal filetype for this assignment.")
packaged_files.append((f"files[]", (file, open(file, "rb"), "text/x-csrc")))
data = {"judgenow": "true" if judge else "false", "judgeLanguage": found_type}
resp = self.__session.post(url, files=packaged_files, data=data)
# Close each file
i = 0
for f in packaged_files:
f[1][1].close()
if not wait:
return resp.url if "@submissions" in resp.url else None
return self.__waitForResult(resp.url, not silent, [])

View File

@ -1,62 +0,0 @@
from Year import Year
import urllib3
from requests import Session
from bs4 import BeautifulSoup
# Disable warnings
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
class Themis:
def __init__(self,user:str, passwd:str):
self.session = self.login(user,passwd)
self.years = []
self.url = "https://themis.housing.rug.nl/course/"
def login(self, user, passwd):
headers = {
"user-agent": "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chromium/80.0.3987.160 Chrome/80.0.3987.163 Safari/537.36"
}
data = {
"user": user,
"password":passwd,
"null": None
}
with Session() as s:
url = 'https://themis.housing.rug.nl/log/in'
r = s.get(url,headers=headers,verify=False)
soup = BeautifulSoup(r.text, 'lxml')
# get the csrf token and add it to payload
csrfToken = soup.find('input',attrs = {'name':'_csrf'})['value']
data['_csrf'] = csrfToken
data['sudo'] = user.lower()
# Login
r = s.post(url,data=data,headers = headers)
# check if login was successful
log_out = "Welcome, logged in as" in r.text
if not log_out:
raise Exception(f"Login for user {user} failed")
return s
def getYear(self, start:int, end:int):
return Year(self.session, self, start, end)
def allYears(self):
# All of them are in a big ul at the beginning of the page
r = self.session.get(self.url)
soup = BeautifulSoup(r.text, 'lxml')
ul = soup.find('ul', class_='round')
lis = ul.find_all('li', class_='large')
years = []
for li in lis:
# format: 2019-2020
year = li.a.text.split("-")
years.append(Year(self.session, self, int(year[0]), int(year[1])))
return years # Return a list of year objects

View File

@ -1,55 +0,0 @@
# Year class to represent an academic year
from bs4 import BeautifulSoup
from Course import Course
from requests import Session
from exceptions.CourseUnavailable import CourseUnavailable
# Works
class Year:
def __init__(self, session:Session, parent, start_year:int, end_year:int):
self.start = start_year
self.year = end_year
self.url = self.__constructUrl()
self.__session = session
# Method to set the url
def __constructUrl(self):
return f"https://themis.housing.rug.nl/course/{self.start}-{self.year}"
# Method to get the courses of the year
def allCourses(self, errors:bool=False) -> list[Course]:
# lis in a big ul
r = self.__session.get(self.url)
soup = BeautifulSoup(r.text, 'lxml')
lis = soup.find_all('li', class_='large')
courses = []
for li in lis:
try:
suffix = (li.a['href'].replace(f"course/{self.start}-{self.year}", ""))
courses.append(
Course(
self.url + suffix,
li.a.text,
self.__session,
self
)
)
except CourseUnavailable:
if errors:
raise CourseUnavailable(f"Course {li.a.text} in year {self.start}-{self.year} is not available")
else:
print("error with course", li.a.text)
continue
return courses
def getCourse(self, name:str) -> Course:
# Get the course
r = self.__session.get(self.url)
soup = BeautifulSoup(r.text, 'lxml')
# Search by name
course = self.url + soup.find('a', text=name)['href'].replace(f"course/{self.start}-{self.year}", "")
# Get the url and transform it into a course object
return Course(url=course, name=name, session=self.__session, parent=self)

View File

@ -1,4 +0,0 @@
class CourseUnavailable(Exception):
def __init__(self, message:str=""):
self.message = "Course Error: " + message
super().__init__(self.message)

View File

@ -1,4 +0,0 @@
class IllegalAction(Exception):
def __init__(self, message:str=""):
self.message = "Illegal action: " + message
super().__init__(self.message)

8
temmies/__init__.py Normal file
View File

@ -0,0 +1,8 @@
"""
Entry point for the temmies package.
"""
import urllib3
from .themis import Themis
__all__ = ["Themis"]
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)

40
temmies/course.py Normal file
View File

@ -0,0 +1,40 @@
"""
Represents a course.
A course is a group that contains exercises or other groups.
"""
from .group import Group
from .exercise_group import ExerciseGroup
class Course(Group):
"""
Represents a course.
"""
def __init__(self, session, course_path: str, title: str, parent):
super().__init__(session, course_path, title, parent)
self.course_path = course_path # e.g., '/2023-2024/adinc-ai'
def __str__(self):
return f"Course({self.title})"
def create_group(self, item_data):
"""
Create a subgroup (Group or ExerciseGroup) based on item data.
"""
if item_data.get("submitable", False):
return ExerciseGroup(
self.session,
item_data["path"],
item_data["title"],
self,
item_data.get("submitable", False),
)
return Group(
self.session,
item_data["path"],
item_data["title"],
self,
item_data.get("submitable", False),
)

View File

View File

@ -0,0 +1,6 @@
""" This module contains the CourseUnavailable exception. """
class CourseUnavailable(Exception):
"""CourseUnavailable Exception"""
def __init__(self, message: str = ""):
super().__init__(f"Course unavailable: {message}")

View File

@ -0,0 +1,8 @@
"""
Illegal Action Exception
"""
class IllegalAction(Exception):
"""Illegal Action Exception"""
def __init__(self, message: str = ""):
super().__init__(f"Illegal action: {message}")

31
temmies/exercise_group.py Normal file
View File

@ -0,0 +1,31 @@
"""
Represents a submittable exercise.
"""
from bs4 import BeautifulSoup
from .group import Group
class ExerciseGroup(Group):
"""
Represents a submittable exercise.
"""
# pylint: disable=too-many-arguments, too-many-positional-arguments
def __init__(self, session, path: str, title: str, parent, submitable: bool = True):
super().__init__(session, path, title, parent, submitable=submitable)
self.submit_url = f"{self.base_url}/api/submit{self.path}"
self.__find_name()
def __find_name(self):
"""
Find the name of the exercise group.
"""
if self.title == "":
response = self.session.get(self.base_url + self.path)
soup = BeautifulSoup(response.text, "lxml")
title_elements = soup.find_all("a", class_="fill accent large")
if title_elements:
self.title = title_elements[-1].get_text(strip=True)
else:
self.title = self.path.split("/")[-1]
def __str__(self):
return f"ExerciseGroup({self.title})"

357
temmies/group.py Normal file
View File

@ -0,0 +1,357 @@
"""
Abstract-ish Group class for Themis API.
"""
import os
from typing import Optional, Union, Dict
from json import loads
from time import sleep
from bs4 import BeautifulSoup
from .submission import Submission
class Group:
"""
Represents an item in Themis.
Can be either a folder (non-submittable) or an assignment (submittable).
"""
# pylint: disable=too-many-instance-attributes, too-many-arguments, too-many-positional-arguments
def __init__(self, session, path: str, title: str, parent=None, submitable: bool = False):
self.session = session
self.path = path # e.g., '/2023-2024/adinc-ai/labs'
self.title = title
self.parent = parent
self.submitable = submitable
self.base_url = "https://themis.housing.rug.nl"
self.api_url = f"{self.base_url}/api/navigation{self.path}"
self.classes = []
# Adjust URL construction to include '/course' when accessing HTML pages
if not self.path.startswith('/course/'):
group_url = f"{self.base_url}/course{self.path}"
else:
group_url = f"{self.base_url}{self.path}"
# Fetch the page and parse it
response = self.session.get(group_url)
if response.status_code != 200:
raise ConnectionError(f"Failed to retrieve page for '{self.title}'.")
self._raw = BeautifulSoup(response.text, "lxml")
def get_items(self) -> list:
"""
Get all items (groups and assignments) under this group.
"""
section = self._raw.find("div", class_="ass-children")
if not section:
return []
entries = section.find_all("a", href=True)
items = []
for x in entries:
href = x['href']
name = x.text.strip()
classes = x.get('class', [])
submitable = "ass-submitable" in classes
item = Group(
session=self.session,
path=href,
title=name,
parent=self,
submitable=submitable
)
items.append(item)
return items
def get_item_by_title(self, title: str):
"""
Get a single item by its title, case-insensitive.
"""
items = self.get_items()
for item in items:
if (item.title.lower() == title.lower()) or (item.path.split("/")[-1] == title):
return item
raise ValueError(f"Item '{title}' not found under {self.title}.")
def get_status(self, text: bool = False) -> Union[Dict[str, Union[str, 'Submission']], None]:
"""
Get the status of the current group, if available.
"""
status_link = self._raw.find("a", text="Status")
if not status_link:
raise ValueError(
"Status information is not available for this group.")
status_url = f"{self.base_url}{status_link['href']}"
response = self.session.get(status_url)
if response.status_code != 200:
raise ConnectionError(
f"Failed to retrieve status page for '{self.title}'.")
soup = BeautifulSoup(response.text, "lxml")
section = soup.find("div", class_="cfg-container")
if not section:
return None
return self.__parse_status_section(section, text)
def __parse_status_section(
self, section: BeautifulSoup,
text: bool
) -> Dict[str, Union[str, 'Submission']]:
"""
Parse the status section of the group and clean up keys.
"""
key_mapping = {
"leading the submission that counts towards the grade": "leading",
"best the latest submission with the best result": "best",
"latest the most recent submission": "latest",
"first pass the first submission that passed": "first_pass",
"last pass the last submission to pass before the deadline": "last_pass",
}
parsed = {}
cfg_lines = section.find_all("div", class_="cfg-line")
for line in cfg_lines:
key_element = line.find("span", class_="cfg-key")
value_element = line.find("span", class_="cfg-val")
if not key_element or not value_element:
continue
# Normalize key
raw_key = " ".join(key_element.get_text(
separator=" ").strip().replace(":", "").lower().split())
# Use mapped key if available
key = key_mapping.get(raw_key, raw_key)
# Process value
link = value_element.find("a", href=True)
if link and not text:
href = link["href"]
# Construct full URL
if href.startswith("/"):
submission_url = href
elif href.startswith("http"):
submission_url = href.replace(
"https://themis.housing.rug.nl", "")
else:
print(f"Invalid href '{href}' found in status page.")
continue # Skip this entry if href is invalid
# Instantiate Submission with submission_url and session
submission = Submission(submission_url, self.session)
parsed[key] = submission
else:
parsed[key] = value_element.get_text(separator=" ").strip()
return parsed
def get_test_cases(self) -> list[Dict[str, str]]:
"""
Get all test cases for this assignment.
"""
if not self.submitable:
raise ValueError(
f"No test cases for non-submittable item '{self.title}'.")
sections = self._raw.find_all("div", class_="subsec round shade")
tcs = []
for div in sections:
res = div.find("h4", class_="info")
if res and "Test cases" in res.text:
for case in div.find_all("div", class_="cfg-line"):
link = case.find("a")
if link:
tcs.append({
'title': link.text.strip(),
'path': link['href']
})
return tcs
def download_tcs(self, path=".") -> list[str]:
"""
Download all test cases for this assignment.
"""
test_cases = self.get_test_cases()
downloaded = []
for tc in test_cases:
url = f"{self.base_url}{tc['path']}"
print(f"Downloading {tc['title']}")
response = self.session.get(url)
if response.status_code == 200:
tc_filename = os.path.join(path, tc['title'])
with open(tc_filename, 'wb') as f:
f.write(response.content)
downloaded.append(tc_filename)
else:
print(f"Failed to download test case '{tc['title']}'")
return downloaded
def get_files(self) -> list[Dict[str, str]]:
"""
Get all downloadable files for this assignment.
"""
details = self._raw.find(
"div", id=lambda x: x and x.startswith("details"))
if not details:
return []
cfg_lines = details.find_all("div", class_="cfg-line")
files = []
for line in cfg_lines:
key = line.find("span", class_="cfg-key")
if key and "Downloads" in key.text.strip():
vals = line.find_all("span", class_="cfg-val")
for val in vals:
links = val.find_all("a")
for link in links:
files.append({
'title': link.text.strip(),
'path': link['href']
})
return files
def download_files(self, path=".") -> list[str]:
"""
Download all files available for this assignment.
"""
files = self.get_files()
downloaded = []
for file in files:
print(f"Downloading file '{file['title']}'")
url = f"{self.base_url}{file['path']}"
response = self.session.get(url)
if response.status_code == 200:
file_filename = os.path.join(path, file['title'])
with open(file_filename, 'wb') as f:
f.write(response.content)
downloaded.append(file_filename)
else:
print(f"Failed to download file '{file['title']}'")
return downloaded
# pylint: disable=too-many-locals
def submit(
self,
files: list[str],
judge: bool = True,
wait: bool = True,
silent: bool = True
) -> Optional[dict]:
"""
Submit files to this assignment.
Returns a dictionary of test case results or None if wait is False.
"""
if not self.submitable:
raise ValueError(
f"Cannot submit to non-submittable item '{self.title}'.")
form = self._raw.find("form")
if not form:
raise ValueError("Submission form not found.")
url = f"{self.base_url}{form['action']}"
file_types = loads(form.get("data-suffixes", "{}"))
if isinstance(files, str):
files = [files]
packaged_files = []
data = {}
found_type = ""
for file in files:
for suffix, lang in file_types.items():
if file.endswith(suffix):
found_type = lang
break
if not found_type:
print("WARNING: File type not recognized")
with open(file, "rb") as f:
packaged_files.append((found_type, (file, f.read())))
data = {
"judgenow": "true" if judge else "false",
"judgeLanguage": found_type if found_type else "none"
}
if not silent:
print(f"Submitting to {self.title}")
for file in files:
print(f"{file}")
resp = self.session.post(url, files=packaged_files, data=data)
if not wait or not judge:
return resp.url if "@submissions" in resp.url else None
return self.__wait_for_result(resp.url, not silent, [])
def __wait_for_result(self, url: str, verbose: bool, __printed: list) -> dict:
"""
Wait for the submission result and return the test case results.
"""
r = self.session.get(url)
soup = BeautifulSoup(r.text, "lxml")
return self.__parse_table(soup, url, verbose, __printed)
def __parse_table(self, soup: BeautifulSoup, url: str, verbose: bool, __printed: list) -> dict:
"""
Parse the results table from the submission result page.
Wait until all queued status-icons disappear before parsing.
"""
cases = soup.find_all("tr", class_="sub-casetop")
fail_pass = {}
any_queued = False
for case in cases:
name = case.find("td", class_="sub-casename").text.strip()
status = case.find("td", class_="status-icon")
status_classes = status.get("class", [])
if "queued" in status_classes:
any_queued = True
break
if "pending" in status_classes:
sleep(1)
return self.__wait_for_result(url, verbose, __printed)
statuses = {
"Passed": ("", True),
"Wrong output": ("", False),
"No status": ("🐛", None),
"error": ("🐛", None),
}
found = False
for key, (symbol, value) in statuses.items():
if key.lower() in status.text.lower():
found = True
case_number = int(name)
if verbose and case_number not in __printed:
print(f"Case {case_number}: {symbol}")
fail_pass[case_number] = value
break
if not found:
case_number = int(name)
fail_pass[case_number] = None
if verbose and case_number not in __printed:
print(f"{case_number}: Unrecognized status: {status.text.strip()}")
__printed.append(case_number)
# Polling (fix, use ws)
if any_queued:
sleep(1)
return self.__wait_for_result(url, verbose, __printed)
return fail_pass
def __str__(self):
return f"Group({self.title}, submitable={self.submitable})"

104
temmies/submission.py Normal file
View File

@ -0,0 +1,104 @@
# submission.py
"""
File to define the Submission class
"""
from typing import Optional
from bs4 import BeautifulSoup
class Submission:
"""
Submission class
Methods:
get_test_cases: Get a dict of test cases status
get_info: Submission information (in details)
get_files: Get a list of uploaded files (as names)
"""
def __init__(self, url: str, session):
self.url = "https://themis.housing.rug.nl" + url
self.__session = session
self.__request = self.__session.get(self.url)
self.__raw = BeautifulSoup(self.__request.text, "lxml")
self.__info = None
def __clean(self, text: str, value: bool = False) -> str:
"""Clean text"""
clean = text.replace("\t", "").replace("\n", "")
if value:
return clean.strip()
return clean.replace(" ", "_").replace(":", "").lower().strip()
def get_test_cases(self) -> dict[str, str]:
"""Get a dict of test cases status"""
cases = self.__raw.find("div", class_=lambda x: x and "sub-cases" in x.split())
if not cases:
return {}
cases = cases.find("div", class_="cfg-container")
cases = cases.find("table")
results = {}
for entry in cases.find_all("tr", class_="sub-casetop"):
name = entry.find("td", class_="sub-casename").text
status = entry.find(
"td", class_=lambda x: x and "status-icon" in x.split()
).text
results[name.strip()] = self.__clean(status)
return results
def get_info(self) -> Optional[dict[str, str]]:
"""Submission information (in details)"""
if self.__info:
return self.__info
for div in self.__raw.find_all("div", class_="subsec round shade"):
h4 = div.find("h4", class_=lambda x: x and "info" in x.split())
if h4 and "Details" in h4.text:
info = div.find("div", class_="cfg-container")
info_lines = info.find_all("div", class_="cfg-line")
self.__info = {
self.__clean(
key := line.find("span", class_="cfg-key").text
): (
self.__clean(line.find("span", class_="cfg-val").text, value=True)
if "Files" not in key
else [
(self.__clean(a.text), a["href"])
for a in line.find("span", class_="cfg-val").find_all("a")
]
)
for line in info_lines
}
return self.__info
return None
def get_files(self) -> Optional[list[str]]:
"""Get a list of uploaded files in the format [(name, url)]"""
if not self.__info:
self.__info = self.get_info()
return self.__info.get("files", None)
# Deprecated methods
def info(self):
"""
Deprecated method. Use get_info instead.
"""
print("This method is deprecated and will be deleted soon. Use get_info instead.")
return self.get_info()
def test_cases(self):
"""
Deprecated method. Use get_test_cases instead.
"""
print("This method is deprecated and will be deleted in soon. Use get_test_cases instead.")
return self.get_test_cases()
def files(self):
"""
Deprecated method. Use get_files instead.
"""
print("This method is deprecated and will be deleted in soon. Use get_files instead.")
return self.get_files()

160
temmies/themis.py Normal file
View File

@ -0,0 +1,160 @@
"""
Main class for the Themis API using the new JSON endpoints.
"""
from requests import Session
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.common.exceptions import NoSuchElementException
from json import dumps
from .year import Year
import getpass
import keyring
class Themis:
"""
Main class for interacting with Themis.
- login: Login to Themis
- get_year: Get a year object
- all_years: Get all years
"""
def __init__(self, cookies: dict = None, user=None):
"""
Initialize Themis object, logging in with the given user.
Args:
user (str): Username to login with.
Attributes:
user (str): Username.
password (str): Password, retrieved from keyring.
base_url (str): Base URL of the Themis website.
session (requests.Session): Authenticated session.
"""
self.base_url = "https://themis.housing.rug.nl"
self.session = self._setup_agent()
self.user, self.password = None, None
# Old login logic
if user:
self.user = user
self.password = self._get_password()
# Reusing session logic
if not cookies:
self.session = self.login(self.session)
else:
self.session.cookies.update(cookies)
if not self.check_session():
self.session = self.login(self.session)
def _get_password(self) -> str:
"""
Retrieve the password from the keyring, prompting the user if not found.
"""
password = keyring.get_password(f"{self.user}-temmies", self.user)
if not password:
print(f"Password for user '{self.user}' not found in keyring.")
password = getpass.getpass(
prompt=f"Enter password for {self.user}: ")
keyring.set_password(f"{self.user}-temmies", self.user, password)
print("Password saved securely in keyring.")
return password
def _setup_agent(self) -> Session:
session = Session()
user_agent = (
"Mozilla/5.0 (X11; Linux x86_64) "
"AppleWebKit/537.36 (KHTML, like Gecko) "
"Chromium/80.0.3987.160 Chrome/80.0.3987.163 Safari/537.36"
)
session.headers.update({"User-Agent": user_agent})
return session
def check_session(self) -> bool:
"""
Check if the session is still valid.
"""
# look at the /login and find a pre tag
login_url = f"{self.base_url}/login"
response = self.session.get(login_url)
return "pre" in response.text
def login(self, session: Session) -> Session:
"""
Login to Themis by spawning a selenium browser, logging in and storing the session.
"""
login_url = f"{self.base_url}/login"
# Start a full browser to login
driver = webdriver.Chrome()
driver.get(login_url)
while True:
if driver.find_elements(By.TAG_NAME, "pre"):
break
try:
# if any of the fields are already filled, don't fill them
if (passw := driver.find_element(By.NAME, "Ecom_Password")) and not passw.get_attribute("value") and self.password:
passw.send_keys(self.password)
if (user := driver.find_element(By.NAME, "Ecom_User_ID")) and not user.get_attribute("value") and self.user:
user.send_keys(self.user)
except NoSuchElementException:
pass
# destroy the password from memory (security)
self.password = "I-HAVE-BEEN-REMOVED"
# export all stored cookies
cookies = driver.get_cookies()
driver.quit()
# add all cookies to the session
for cookie in cookies:
session.cookies.set(cookie["name"], cookie["value"])
return session
def get_session_cookies(self):
"""
Get the session cookies in json
"""
return dumps(self.session.cookies.get_dict())
def get_year(self, start_year: int = None, end_year: int = None) -> Year:
"""
Gets a Year object using the year path (e.g., 2023, 2024).
"""
year_path = f"{start_year}-{end_year}"
return Year(self.session, year_path)
def all_years(self) -> list:
"""
Gets all visible years as Year objects.
"""
navigation_url = f"{self.base_url}/api/navigation/"
response = self.session.get(navigation_url)
if response.status_code != 200:
raise ConnectionError("Failed to retrieve years from Themis API.")
years_data = response.json()
years = []
for year_info in years_data:
if year_info.get("visible", False):
year_path = year_info["path"].strip("/")
years.append(Year(self.session, year_path))
return years

75
temmies/year.py Normal file
View File

@ -0,0 +1,75 @@
"""
This module defines the Year class for managing academic year courses.
"""
from bs4 import BeautifulSoup
from .course import Course
class Year:
"""
Represents an academic year.
"""
def __init__(self, session, year_path: str):
self.session = session
self.year_path = year_path # e.g., '2023-2024'
self.base_url = "https://themis.housing.rug.nl"
self.api_url = f"{self.base_url}/api/navigation/{self.year_path}"
def all_courses(self) -> list:
"""
Gets all visible courses in this year.
"""
response = self.session.get(self.api_url)
if response.status_code != 200:
raise ConnectionError(f"Failed to retrieve courses for {self.year_path}.")
courses_data = response.json()
courses = []
for course_info in courses_data:
if course_info.get("visible", False):
course_path = course_info["path"]
course_title = course_info["title"]
courses.append(Course(self.session, course_path, course_title, self))
return courses
def get_course(self, course_title: str) -> Course:
"""
Gets a course by its title.
"""
all_courses = self.all_courses()
for course in all_courses:
if course.title == course_title:
return course
raise ValueError(f"Course '{course_title}' not found in year {self.year_path}.")
def get_course_by_tag(self, course_tag: str) -> Course:
"""
Gets a course by its tag (course identifier).
Constructs the course URL using the year and course tag.
"""
course_path = f"/{self.year_path}/{course_tag}"
course_url = f"{self.base_url}/course{course_path}"
response = self.session.get(course_url)
if response.status_code != 200:
raise ConnectionError(
f"Failed to retrieve course '{course_tag}' for year {self.year_path}."
)
soup = BeautifulSoup(response.text, "lxml")
title_elements = soup.find_all("a", class_="fill accent large")
title_element = title_elements[-1] if title_elements else None
if title_element:
course_title = title_element.get_text(strip=True)
else:
raise ValueError(
f"Could not retrieve course title for tag '{course_tag}' in year {self.year_path}."
)
return Course(self.session, course_path, course_title, self)
def __str__(self):
return f"Year({self.year_path})"