86 lines
2.5 KiB
Python
Executable file
86 lines
2.5 KiB
Python
Executable file
#!/usr/bin/env python3
|
||
# PYTHON_ARGCOMPLETE_OK
|
||
import argcomplete, argparse
|
||
import json as JSON
|
||
import requests
|
||
import sys
|
||
|
||
from datetime import datetime, timedelta
|
||
|
||
# ===========================================================================
|
||
|
||
def getOldStations():
|
||
params = {"json": JSON.dumps({"filters": FILTERS, "sort": SORT, "size": COUNT})}
|
||
json = requests.post(APIURL, params).json()
|
||
|
||
ret =""
|
||
for station in json["results"]:
|
||
if args.short:
|
||
ret += "{}\n".format(station["system_name"])
|
||
else:
|
||
ret += "{}: {} ({})\n".format(station["system_name"], station["name"],
|
||
station["updated_at"])
|
||
|
||
return ret[:-1]
|
||
|
||
def getOldStationsInSystem(system):
|
||
FILTERS.update(system_name={"value": system})
|
||
params = {"json": JSON.dumps({"filters": FILTERS, "sort": SORT})}
|
||
json = requests.post(APIURL, params).json()
|
||
|
||
ret =""
|
||
for station in json["results"]:
|
||
if args.short:
|
||
ret += "{}\n".format(station["name"])
|
||
else:
|
||
ret += "{} ({})\n".format(station["name"], station["updated_at"])
|
||
|
||
return ret[:-1]
|
||
|
||
# ===========================================================================
|
||
|
||
parser = argparse.ArgumentParser(description="Script for interfacing with "
|
||
+ "Spansh’s API.")
|
||
subparsers = parser.add_subparsers(title="subcommands", help="sub-command help",
|
||
dest="subcommand", required=True)
|
||
|
||
parser_oldstations = subparsers.add_parser("oldstations",
|
||
help="Searches for stations with old data (>1 year without an update.")
|
||
parser_oldstations.add_argument("--system", nargs="?",
|
||
help="a single system to query. If not present, get the oldest stations "
|
||
+ "overall.")
|
||
parser_oldstations.add_argument("--count", nargs="?", type=int, default=50,
|
||
help="how many stations to output. Defaults to 50.")
|
||
parser_oldstations.add_argument("--short", action='store_true',
|
||
help="short output format (system/station names only)")
|
||
|
||
argcomplete.autocomplete(parser)
|
||
args = parser.parse_args()
|
||
|
||
# ===========================================================================
|
||
|
||
APIURL = "http://backend3.spansh.co.uk:3000/api/stations/search"
|
||
|
||
FILTERS = {"updated_at":
|
||
{"value":
|
||
["2017-11-06",
|
||
(datetime.now() - timedelta(days=365)).strftime("%Y-%m-%d")],
|
||
"comparison": "<=>"}}
|
||
SORT = {"updated_at": {"direction": "asc"}}
|
||
COUNT = args.count
|
||
|
||
out =""
|
||
if args.subcommand == "oldstations":
|
||
if args.system:
|
||
out = getOldStationsInSystem(args.system)
|
||
else:
|
||
out = getOldStations()
|
||
else:
|
||
parser.print_usage(sys.stderr)
|
||
sys.exit(1)
|
||
|
||
if out == "":
|
||
sys.exit(3)
|
||
else:
|
||
print(out)
|
||
sys.exit(0)
|