aboutsummaryrefslogtreecommitdiff
path: root/libraries
diff options
context:
space:
mode:
Diffstat (limited to 'libraries')
-rw-r--r--libraries/aspen.py49
-rw-r--r--libraries/base.py7
2 files changed, 53 insertions, 3 deletions
diff --git a/libraries/aspen.py b/libraries/aspen.py
index 1d8de39..40e67c1 100644
--- a/libraries/aspen.py
+++ b/libraries/aspen.py
@@ -44,6 +44,51 @@ class Library(BaseLibrary):
# log in now. Why would we not?
self.login()
+ def get_reservations(self, verbose = False):
+ availableReservations = []
+ unavailableReservations = []
+ b = self.baseurl
+ s = self.session
+ # step 1: visit "titles on hold" page so it does not complain that I am taking shortcuts
+ headers = {
+ "Referer": f"{b}/MyAccount/CheckedOut?source=all"
+ }
+ s.get(f"{b}/Holds?source=ils",headers=headers)
+ output = s.get(f"{b}/AJAX?method=getHolds&source=all",headers=headers)
+ output = json.loads(output.content)["holds"].replace("\xa0"," ")
+ soup = BeautifulSoup(output, "html.parser")
+ try:
+ unavailableholds_all = soup.find("label",attrs={"for":"unavailableHoldSort_all"}).parent.next_sibling.next_sibling
+ except AttributeError:
+ # the label will not exist if there are no unavaiableHolds
+ unavailableholds_all = None
+ if unavailableholds_all:
+ items = unavailableholds_all.find_all("div",class_=["result"])
+ for i in items:
+ labels = [j.text for i in i.find_all("div","result-label")]
+ values = [j.text for i in i.find_all("div","result-value")]
+ values_dict = dict(map(lambda i,j:(i,j),labels,values))
+ title_obj = i.find("a",class_="result-title")
+ img_href = i.find("img")["src"]
+ img_b64, img_type = self.get_image(img_href)
+ obj = {
+ "position": values_dict["Position"],
+ "status": values_dict["Status"],
+ "date_placed": values_dict["Date Placed"],
+ "format": values_dict["Format"],
+ "location": values_dict["Pickup Location"],
+ "title": title_obj.text,
+ "img_href": img_href,
+ "img50": img_b64[:50],
+ "img": img_b64,
+ "img_type": img_type,
+ }
+ unavailableReservations.append(obj)
+
+ # WORKHERE: availableHolds, might not be available.
+ # Return a single list of objects
+ return availableReservations + unavailableReservations
+
def get_checkouts(self, verbose = False):
# WORKHERE: no example of possible/completed renewals at this time
checked_out_objects = []
@@ -80,9 +125,7 @@ class Library(BaseLibrary):
print(f"DEBUG: Values_dict: {values_dict}",file=sys.stderr)
# contains Call number, Format, Barcode, Due
img_href = i.find("img", class_="listResultImage")["src"]
- img_response = s.get(img_href)
- img_b64 = base64.b64encode(img_response.content).decode()
- img_type = img_response.headers["Content-Type"]
+ img_b64, img_type = self.get_image(img_href)
# normalize format
item_format = ""
item_format = "book" if "book" in values_dict["Format"].lower() else ""
diff --git a/libraries/base.py b/libraries/base.py
index 74394be..f6f75e4 100644
--- a/libraries/base.py
+++ b/libraries/base.py
@@ -48,3 +48,10 @@ class BaseLibrary:
"""
This is where the login interaction should happen.
"""
+
+ def get_image(self, url):
+ """ Given the url, return the base64-encoded contents and image type as a tuple. """
+ img_response = self.session.get(url)
+ img_b64 = base64.b64encode(img_response.content).decode()
+ img_type = img_response.headers["Content-Type"]
+ return img_b64, img_type
bgstack15