prepare for reservations

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