feat: enhance WebRequest and BibTextTransformer to support exact signature search and any book mode

This commit is contained in:
2024-11-26 10:21:02 +01:00
parent 070c2cef07
commit a1b2fddc3c
2 changed files with 25 additions and 5 deletions

View File

@@ -38,14 +38,18 @@ class WebRequest:
self.data = None
self.timeout = 5
logger.log_info("Initialized WebRequest")
@property
def use_any_book(self):
"""use any book that matches the search term"""
self.use_any = True
logger.log_info("Using any book")
return self
def set_apparat(self, apparat):
self.apparat = apparat
if int(self.apparat) < 10:
self.apparat = f"0{self.apparat}"
logger.log_info(f"Set apparat to {self.apparat}")
return self
def get_ppn(self, signature):
@@ -102,7 +106,7 @@ class WebRequest:
return return_data
else:
logger.log_error("No <pre> tag found")
return return_data
raise ValueError("No <pre> tag found")
if f"Semesterapparat-{self.apparat}" in item_location:
pre_tag = soup.find_all("pre")
return_data = []
@@ -149,12 +153,18 @@ class BibTextTransformer:
def __init__(self, mode: str = "ARRAY") -> None:
self.mode = mode
self.field = None
self.signature = None
if mode not in self.valid_modes:
logger.log_error(f"Mode {mode} not valid")
raise ValueError(f"Mode {mode} not valid")
self.data = None
# self.bookdata = BookData(**self.data)
def use_signature(self, signature: str):
"""use the exact signature to search for the book"""
self.signature = signature
return self
def get_data(self, data: list):
RIS_IDENT = "TY -"
ARRAY_IDENT = "[kid]"
@@ -201,7 +211,7 @@ class BibTextTransformer:
return None
match self.mode:
case "ARRAY":
return ARRAYData().transform(self.data)
return ARRAYData(self.signature).transform(self.data)
case "COinS":
return COinSData().transform(self.data)
case "BibTeX":