diff options
Diffstat (limited to 'libraries')
| -rw-r--r-- | libraries/aspen.py | 49 | ||||
| -rw-r--r-- | libraries/base.py | 7 |
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 |