87 lines
2.5 KiB
Python
87 lines
2.5 KiB
Python
import requests
|
|
import json
|
|
import os
|
|
|
|
|
|
class ABSConnector:
|
|
def __init__(self, abs_url, token=None):
|
|
self.abs_url = abs_url
|
|
self.requests = requests.Session()
|
|
self.requests.headers = {"Authorization": f"Bearer {token}"}
|
|
|
|
def get_library_ids(self):
|
|
endpoint = f"{self.abs_url}/api/libraries"
|
|
response = self.requests.get(endpoint)
|
|
response.raise_for_status()
|
|
data = response.json()
|
|
return data["libraries"]
|
|
|
|
def _get_library_page(self, library_id, page=0, page_size=100):
|
|
endpoint = f"{self.abs_url}/api/libraries/{library_id}/series"
|
|
response = self.requests.get(
|
|
endpoint,
|
|
params={
|
|
"limit": page_size,
|
|
"page": page,
|
|
"minified": 1,
|
|
"sort": "name",
|
|
},
|
|
)
|
|
response.raise_for_status()
|
|
return response.json()
|
|
|
|
def get_series_by_library_id(self, library_id, page_size=100):
|
|
page = 0
|
|
|
|
while True:
|
|
data = self.__get_library_page(library_id, page, page_size)
|
|
|
|
yield from data["results"]
|
|
|
|
page += 1
|
|
|
|
if data["total"] < page_size * page: # Stop if no more data
|
|
break
|
|
|
|
|
|
class ABSConnectorMock(ABSConnector):
|
|
def __init__(self, abs_url, token=None):
|
|
super().__init__(abs_url, token)
|
|
|
|
self.directory = "dumps/abs"
|
|
if not os.path.exists(self.directory):
|
|
os.makedirs(self.directory)
|
|
|
|
def get_library_ids(self):
|
|
path = f"{self.directory}/libraries.json"
|
|
|
|
try:
|
|
with open(path, "r") as f:
|
|
data = json.load(f)
|
|
return data["libraries"]
|
|
except FileNotFoundError:
|
|
data = ABSConnector.get_library_ids(self)
|
|
with open(path, "w+") as f:
|
|
json.dump({"libraries": data}, f, indent=4)
|
|
return data
|
|
|
|
def get_series_by_library_id(self, library_id, page_size=100):
|
|
page = 0
|
|
|
|
while True:
|
|
path = f"{self.directory}/library_{library_id}.page_{page}.json"
|
|
|
|
try:
|
|
with open(path, "r") as f:
|
|
data = json.load(f)
|
|
except FileNotFoundError:
|
|
data = ABSConnector._get_library_page(self, library_id, page, page_size)
|
|
with open(path, "w+") as f:
|
|
json.dump(data, f, indent=4)
|
|
|
|
yield from data["results"]
|
|
|
|
page += 1
|
|
|
|
if data["total"] < page_size * page: # Stop if no more data
|
|
break
|