From ddcab9f3f44fd921bb70021674fb4052efd604f0 Mon Sep 17 00:00:00 2001 From: Bryan Newbold Date: Tue, 26 Apr 2022 15:16:27 -0700 Subject: ingest_tool: spn-status command to check user's quota --- python/ingest_tool.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) (limited to 'python') diff --git a/python/ingest_tool.py b/python/ingest_tool.py index a155842..ac9ece1 100755 --- a/python/ingest_tool.py +++ b/python/ingest_tool.py @@ -115,6 +115,20 @@ def run_file_requests_backfill(args): pusher.run() +def run_spn_status(args): + worker = IngestFileWorker( + sink=None, + try_spn2=False, + ) + + resp = worker.spn_client.v2_session.get("https://web.archive.org/save/status/system") + resp.raise_for_status() + print(f"System status: {json.dumps(resp.json(), sort_keys=True)}") + resp = worker.spn_client.v2_session.get("https://web.archive.org/save/status/user") + resp.raise_for_status() + print(f"User status: {json.dumps(resp.json(), sort_keys=True)}") + + def run_api(args): port = 8083 print("Listening on localhost:{}".format(port)) @@ -204,6 +218,11 @@ def main(): "--grobid-host", default="https://grobid.qa.fatcat.wiki", help="GROBID API host/port" ) + sub_spn_status = subparsers.add_parser( + "spn-status", help="checks save-page-now v2 API status for bot user" + ) + sub_spn_status.set_defaults(func=run_spn_status) + args = parser.parse_args() if not args.__dict__.get("func"): parser.print_help(file=sys.stderr) -- cgit v1.2.3