boyan_k/temmies
boyan_k/temmies
1
0
Fork 0
mirror of https://github.com/Code-For-Groningen/temmies.git synced 2025-03-15 07:10:15 +01:00
Code Issues Packages Projects Releases Wiki Activity

Solved #11 by using system's keyring

Browse Source
This commit is contained in:
Boyan 2024-11-17 21:53:02 +01:00
parent 2378be4d42
commit 17f0bd237a
1 changed files with 25 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
temmies/themis.py
View File

@ -4,6 +4,8 @@ Main class for the Themis API
""" """
import urllib3 import urllib3
import keyring
import getpass
from requests import Session from requests import Session
from bs4 import BeautifulSoup from bs4 import BeautifulSoup
from .year import Year from .year import Year
@ -21,10 +23,22 @@ class Themis:
all_years: Get all years all_years: Get all years
""" """
def __init__(self, user: str, passwd: str): def __init__(self, user: str):
self.session = self.login(user, passwd) self.user = user
self.years = [] self.password = self.__get_password()
self.url = "https://themis.housing.rug.nl/course/" self.session = self.login(user, self.password)
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 login(self, user: str, passwd: str) -> Session: def login(self, user: str, passwd: str) -> Session:
""" """
@ -58,9 +72,13 @@ class Themis:
# check if login was successful # check if login was successful
log_out = "Welcome, logged in as" in r.text log_out = "Welcome, logged in as" in r.text
if not log_out: if "Invalid credentials" in r.text:
raise IllegalAction(message=f"Login for user {user} failed") # Prompt for password again
print("Invalid credentials. Please try again.")
passwd = getpass.getpass(prompt="Enter password: ")
keyring.set_password(f'{self.user}-temmies', self.user, passwd)
return self.login(user, passwd)
return s return s
def get_year(self, start: int, end: int) -> Year: def get_year(self, start: int, end: int) -> Year: