mirrors/pferd
1
0
Fork 0
mirror of https://github.com/Garmelon/PFERD.git synced 2023-12-21 10:23:01 +01:00
Code Issues Packages Projects Releases Wiki Activity

Clean up

Browse Source 
Use shorter name for responses, like in the requests doc.

Change Organizer's __all__ to be more in line with the other __all__s.
This commit is contained in:
Joscha 2019-04-25 19:02:48 +00:00
parent 82adeb324f
commit f0c42ce8ec
2 changed files with 11 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
PFERD/ilias_authenticators.py
View File

@ -69,8 +69,8 @@ class ShibbolethAuthenticator:
"target": "/shib_login.php",
"home_organization_selection": "Mit KIT-Account anmelden",
}
response = self._session.post(url, data=data)
soup = bs4.BeautifulSoup(response.text, "html.parser")
r = self._session.post(url, data=data)
soup = bs4.BeautifulSoup(r.text, "html.parser")
# Attempt to login using credentials, if necessary
while not self._login_successful(soup):
@ -92,8 +92,8 @@ class ShibbolethAuthenticator:
"j_username": username,
"j_password": password,
}
response = self._session.post(url, data=data)
soup = bs4.BeautifulSoup(response.text, "html.parser")
r = self._session.post(url, data=data)
soup = bs4.BeautifulSoup(r.text, "html.parser")
if not self._login_successful(soup):
print("Incorrect credentials.")
@ -132,8 +132,8 @@ class ShibbolethAuthenticator:
while True:
logger.debug(f"Getting {self.ILIAS_GOTO} {params}")
response = self._session.get(self.ILIAS_GOTO, params=params)
soup = bs4.BeautifulSoup(response.text, "html.parser")
r = self._session.get(self.ILIAS_GOTO, params=params)
soup = bs4.BeautifulSoup(r.text, "html.parser")
if self._is_logged_in(soup):
return soup
@ -145,16 +145,16 @@ class ShibbolethAuthenticator:
return self.get_webpage(f"fold_{ref_id}")
def _download(self, url, params, to_path):
with self._session.get(url, params=params, stream=True) as response:
content_type = response.headers["content-type"]
with self._session.get(url, params=params, stream=True) as r:
content_type = r.headers["content-type"]
if content_type in self.ALLOWED_CONTENT_TYPES:
# Yay, we got the file :)
stream_to_path(response, to_path)
stream_to_path(r, to_path)
return True
elif content_type == "text/html":
# Dangit, we're probably not logged in.
soup = bs4.BeautifulSoup(response.text, "html.parser")
soup = bs4.BeautifulSoup(r.text, "html.parser")
if self._is_logged_in(soup):
raise ContentTypeException(
"Attempting to download a web page, not a file")

4
PFERD/organizer.py
View File

@ -5,9 +5,7 @@ import shutil
from . import utils
__all__ = [
"Organizer",
]
__all__ = ["Organizer"]
logger = logging.getLogger(__name__)
class Organizer: