From 91e6b33a4733fbe622ce0e09460a75cd377bee7a Mon Sep 17 00:00:00 2001 From: Bryan Newbold Date: Tue, 14 Jan 2020 17:00:59 -0800 Subject: small fixups to SandcrawlerPostgrestClient --- python/sandcrawler/db.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) (limited to 'python/sandcrawler/db.py') diff --git a/python/sandcrawler/db.py b/python/sandcrawler/db.py index eb1a922..1a47b0b 100644 --- a/python/sandcrawler/db.py +++ b/python/sandcrawler/db.py @@ -9,7 +9,7 @@ import requests class SandcrawlerPostgrestClient: def __init__(self, api_url="http://aitio.us.archive.org:3030", **kwargs): - self.api_uri = api_url + self.api_url = api_url def get_cdx(self, url): resp = requests.get(self.api_url + "/cdx", params=dict(url='eq.'+url)) @@ -34,6 +34,15 @@ class SandcrawlerPostgrestClient: else: return None + def get_ingest_file_result(self, url): + resp = requests.get(self.api_url + "/ingest_file_result", params=dict(base_url='eq.'+url)) + resp.raise_for_status() + resp = resp.json() + if resp: + return resp[0] + else: + return None + class SandcrawlerPostgresClient: def __init__(self, db_url, **kwargs): -- cgit v1.2.3