massive refactoring, and pylint cleanup
All checks were successful
Pylint / build (3.12) (push) Successful in 7s

This commit is contained in:
Greg Gauthier 2024-07-22 21:28:45 +01:00
parent 520569ab48
commit 0d48269790
7 changed files with 94 additions and 84 deletions

15
Pipfile
View File

@ -1,15 +0,0 @@
[[source]]
url = "https://pypi.org/simple"
verify_ssl = true
name = "pypi"
[packages]
requests = "*"
py3dns = "*"
python-console-menu = "*"
[dev-packages]
pytest = "*"
[requires]
python_version = "3.8"

View File

@ -1,6 +1,7 @@
# pystations python script
Simple python tool to grab radio stations from radio-browser.info, put them into a menu, and then use your local installation of mpg123 to play the station for you.
Simple python tool to grab radio stations from radio-browser.info, put them into a menu, and then use your local
installation of mpg123 to play the station for you.
### Requirements:
@ -9,13 +10,14 @@ Simple python tool to grab radio stations from radio-browser.info, put them into
* Pipenv
* mpv (or mpg123, or mplayer, if you want to swap it in yourself)
### Install
```shell
pipenv install
```
### Execute
```shell
pipenv run python ./stations.py -h
@ -33,6 +35,7 @@ optional arguments:
```
Example (using mpg123):
```
> pipenv run python ./stations.py -n "wfmt"
@ -54,6 +57,7 @@ A: 00:00:27 / 00:00:51 (53%) Cache: 23s/833KB
```
When I hit `Q`, mpv exits, and sends me back to my menu:
```
Exiting... (Quit)
@ -68,11 +72,22 @@ Select Option:
### CAVEATS
1. This is a VERY ROUGH script. If your search criteria are too broad, you'll get hundreds and hundres of menu items. I set the limit to 9999, and it seems to work ok. But don't push your luck; try to be as precise with your searches as possible.
2. ~~mpg123 and mplayer are primitive at best.~~ `mpv` is a better player than mpg123, and more streams will work out of the box. What's more, via homebrew, `mpv` is available on mac or linux. However, it's still the case that not all the stations on the list will play correctly. This is partly due to the player, but mostly due to problems with the streams themselves.
3. if you supply multiple tags in a comma-separated list, you may unintentionally filter out results. Unfortunately, the api at radio-info is such that the tag list you search for, has to be in precisely the order it is returned from the host. So, for example, if you search for "classical,chicago", your search will filter out WFMT, because their tags are "chicago,classical". So, best to keep your tags to a minimum (meaning 1 lol)
4. The country search is by country NAME, not CODE. So, "United States" will work, but "US" will not. Likewise for the United Kingdom.
5. It seems many of the stations put their city in the tag list. So, you can reduce the size of your results by doing something like this: `-c "United States" -t "atlanta"`, which makes more sense for radio stations anyway. Eg:
1. This is a VERY ROUGH script. If your search criteria are too broad, you'll get hundreds and hundres of menu items. I
set the limit to 9999, and it seems to work ok. But don't push your luck; try to be as precise with your searches as
possible.
2. ~~mpg123 and mplayer are primitive at best.~~ `mpv` is a better player than mpg123, and more streams will work out of
the box. What's more, via homebrew, `mpv` is available on mac or linux. However, it's still the case that not all the
stations on the list will play correctly. This is partly due to the player, but mostly due to problems with the
streams themselves.
3. if you supply multiple tags in a comma-separated list, you may unintentionally filter out results. Unfortunately, the
api at radio-info is such that the tag list you search for, has to be in precisely the order it is returned from the
host. So, for example, if you search for "classical,chicago", your search will filter out WFMT, because their tags
are "chicago,classical". So, best to keep your tags to a minimum (meaning 1 lol)
4. The country search is by country NAME, not CODE. So, "United States" will work, but "US" will not. Likewise for the
United Kingdom.
5. It seems many of the stations put their city in the tag list. So, you can reduce the size of your results by doing
something like this: `-c "United States" -t "atlanta"`, which makes more sense for radio stations anyway. Eg:
```
> pipenv run python ./stations.py -c "United States" -t "atlanta"
@ -89,6 +104,10 @@ Radio Player Menu
Select Option:
```
### Potential next steps:
* Searching by tags would be powerful, if it didn't matter how many or which order they were in. So, one approach might be to do some sort of preprocessing on them, after capture. But this requires getting an unfiltered list to begin with. The server code itself is open source on github. So, one could potentially suggest an improvement to the tag searching, via pull request. You can find it here: https://github.com/segler-alex/radiobrowser-api-rust
* Searching by tags would be powerful, if it didn't matter how many or which order they were in. So, one approach might
be to do some sort of preprocessing on them, after capture. But this requires getting an unfiltered list to begin
with. The server code itself is open source on github. So, one could potentially suggest an improvement to the tag
searching, via pull request. You can find it here: https://github.com/segler-alex/radiobrowser-api-rust

View File

@ -18,4 +18,3 @@ def options():
def maxitems():
return int(config.get('DEFAULT', 'menu_items.max'))

View File

@ -1,6 +1,7 @@
from python_console_menu import AbstractMenu, MenuItem
import subprocess
from python_console_menu import AbstractMenu, MenuItem
from config import player, options, maxitems
@ -10,20 +11,22 @@ class RadioMenu(AbstractMenu):
if station_list is None:
station_list = []
for i in range(len(station_list)):
if i < maxitems(): # The very first item is the exit option, so not "<=".
for i, station in enumerate(station_list, start=1):
self.add_menu_item(
MenuItem(
i,
"{:<30}".format(station_list[i]["name"][:30]) + " " + # force 35 character fixed length
"{:<5}".format(station_list[i]["codec"][:5]) + " " + # force 5 character fixed length
"{:<5}".format(station_list[i]["bitrate"][:5]) + " " +
station_list[i]["url"],
lambda url=station_list[i]["url"]: subprocess.run([player(), options(), url])
(
f"{station['name'][:30]:<30} "
f"{station['codec'][:5]:<5} "
f"{station['bitrate'][:5]:<5} "
f"{station['url']}"
),
lambda url=station["url"]: subprocess.run(
[player(), options(), url],
capture_output=True, text=True, check=True
).stdout
)
)
else:
break
def initialise(self):
self.add_menu_item(MenuItem(maxitems(), "Exit menu").set_as_exit_option())

View File

@ -1,5 +1,5 @@
[DEFAULT]
radio_browser.api=all.api.radio-browser.info
player.command=mpv
player.options=--no-video
menu_items.max=9999
radio_browser.api = all.api.radio-browser.info
player.command = mpv
player.options = --no-video
menu_items.max = 9999

View File

@ -1,9 +1,10 @@
import argparse
import DNS
import json
import requests
import secrets
import DNS
import requests
from config import api
from radiomenu import RadioMenu
@ -23,56 +24,48 @@ def get_host():
def get_stations(qstring, host):
resp = requests.get(f"https://{host}/json/stations/search?{qstring}&limit=100000")
resp = requests.get(
f"https://{host}/json/stations/search?{qstring}&limit=100000", timeout=10
)
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 = ""
def search_stations(query_dict=None):
if query_dict is None:
query_dict = {}
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
query_string = ""
for key, value in query_dict.items():
if key == "tags":
query_string += f"{key}={','.join(value)}&"
query_string += f"{key}={value}&"
stations = get_stations(query, get_host())
stations = get_stations(query_string, 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.
for station in stations:
station["status"] = "down" # assume down
# if last check and lastcheckok match, the station is up
if str(station["lastchecktime"]) == str(station["lastcheckoktime"]):
station["status"] = "up"
if len(query_dict["tags"]) > 1 and station["tags"] == "":
continue
if station_stat == status: # only add the entry if it matches the up/down status specified.
if station["status"] == "up":
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"),
"favicon": str(station["favicon"])\
.replace(" ", "%20").replace("(", "%28").replace(")", "%29"),
"tags": str(station["tags"].split(",")).replace(",", ";"),
"status": str(station_stat)
"status": str(station["status"])
}
filtered_list.append(station_entry)
return filtered_list
@ -80,13 +73,24 @@ def search_stations(name=None, country=None, state=None, tags=None, status="up")
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="")
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(","))
query = {
"name": args.name,
"country": args.country,
"state": args.state,
"tags": args.tags.split(","),
}
station_list = search_stations(query)
mainMenu = RadioMenu(station_list)
mainMenu.display()