pystations/stations.py
2021-03-16 20:19:11 +00:00

93 lines
3.1 KiB
Python

import argparse
import DNS
import json
import requests
import secrets
from config import api
from radiomenu import RadioMenu
def ipaddr(hostname, rectype='A'):
return DNS.dnslookup(hostname, rectype, 3)
def nsname(ipaddrs):
return DNS.revlookup(ipaddrs, 3)
def get_host():
hosts = ipaddr(api())
preferred_host = secrets.choice(hosts)
return nsname(preferred_host)
def get_stations(qstring, host):
resp = requests.get(f"https://{host}/json/stations/search?{qstring}&limit=100000")
if resp.status_code == 200:
return json.loads(resp.text)
else:
return [{"response_code": resp.status_code, "reason": resp.reason}]
def search_stations(name=None, country=None, state=None, tags=None, status="up"):
if tags is None:
tags = []
query = ""
if name:
if query != "":
query = query + "&"
query = query + "name=" + name
if country:
if query != "":
query = query + "&"
query = query + "country=" + country
if state:
if query != "":
query = query + "&"
query = query + "state=" + state
if len(tags) > 0:
if query != "":
query = query + "&"
tag_string = ','.join(tags) # Be careful here! the tag list is order-dependent in the http call :(
query = query + "tag=" + tag_string
stations = get_stations(query, get_host())
filtered_list = []
for station in stations:
station_stat = "down"
if str(station["lastchecktime"]) == str(station["lastcheckoktime"]):
station_stat = "up"
if len(tags) > 1 and station["tags"] == "": # If searching with tags, but no tags, don't include the station.
continue
if station_stat == status: # only add the entry if it matches the up/down status specified.
station_entry = {
"name": str(station["name"]).replace('"', "'").replace(",", " - "),
"url": str(station["url"].replace(",", "%2C")),
"codec": str(station["codec"]),
"bitrate": str(station["bitrate"]),
"countrycode": str(station["countrycode"]),
"favicon": str(station["favicon"]).replace(" ", "%20").replace("(", "%28").replace(")", "%29"),
"tags": str(station["tags"].split(",")).replace(",", ";"),
"status": str(station_stat)
}
filtered_list.append(station_entry)
return filtered_list
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument("-n", "--name", type=str, help="Name of station", default=None)
parser.add_argument("-c", "--country", type=str, help="Station country", default=None)
parser.add_argument("-s", "--state", type=str, help="Station state (if in US)", default=None)
parser.add_argument("-t", "--tags", type=str, help="search tag or tags (if more than one, comma-separated", default="")
args = parser.parse_args()
station_list = search_stations(name=args.name, country=args.country, state=args.state, tags=args.tags.split(","))
mainMenu = RadioMenu(station_list)
mainMenu.display()