136 lines
4.9 KiB
Python
136 lines
4.9 KiB
Python
# coding=utf-8
|
|
|
|
import json
|
|
|
|
import requests
|
|
from PySide6.QtCore import QUrl
|
|
from PySide6.QtMultimedia import QAudioOutput, QMediaPlayer
|
|
from PySide6.QtWidgets import QMainWindow, QMessageBox, QStyle
|
|
|
|
from . import API_URL, DEFAULT_VOLUME, Ui_MainWindow
|
|
|
|
from __feature__ import snake_case, true_property # isort: skip
|
|
|
|
|
|
class MainWindow(QMainWindow):
|
|
"""MainWindow Class."""
|
|
|
|
def __init__(self):
|
|
"""Initialize UI, audio player and handlers."""
|
|
super().__init__()
|
|
self.ui = Ui_MainWindow()
|
|
self.ui.setupUi(self)
|
|
|
|
class __icons(object):
|
|
volume = self.style().standard_icon(QStyle.SP_MediaVolume)
|
|
muted = self.style().standard_icon(QStyle.SP_MediaVolumeMuted)
|
|
play = self.style().standard_icon(QStyle.SP_MediaPlay)
|
|
stop = self.style().standard_icon(QStyle.SP_MediaStop)
|
|
|
|
self.__icons = __icons
|
|
|
|
self.ui.volumeToolButton.icon = self.__icons.volume
|
|
self.ui.playbackToolButton.icon = self.__icons.play
|
|
|
|
self.__stations_data = self.get_stations_data()
|
|
self.__player = QMediaPlayer()
|
|
self.__audio = QAudioOutput()
|
|
self.__player.audio_output = self.__audio
|
|
self.print_groups()
|
|
|
|
self.set_volume(DEFAULT_VOLUME)
|
|
self.ui.volumeHorizontalSlider.value = DEFAULT_VOLUME
|
|
|
|
self.ui.groupsListWidget.itemClicked.connect(self.print_stations)
|
|
self.ui.stationsListWidget.itemClicked.connect(self.play_radio)
|
|
self.ui.playbackToolButton.clicked.connect(self.toggle_player)
|
|
self.ui.volumeHorizontalSlider.valueChanged.connect(self.set_volume)
|
|
self.ui.volumeToolButton.clicked.connect(self.toggle_mute)
|
|
|
|
def get_stations_data(self) -> dict:
|
|
"""Get JSON data from API and convert it to dict."""
|
|
try:
|
|
resp = requests.get(API_URL)
|
|
if resp.status_code not in range(200, 299 + 1):
|
|
raise requests.exceptions.ConnectionError()
|
|
else:
|
|
return json.loads(resp.text)
|
|
except requests.exceptions.ConnectionError:
|
|
error_box = QMessageBox.critical(
|
|
self,
|
|
"Błąd",
|
|
"Brak połączenia z serwerem.",
|
|
QMessageBox.Cancel,
|
|
)
|
|
error_box.exec()
|
|
|
|
def print_groups(self) -> None:
|
|
"""Print groups (categories) in groupsListWidget."""
|
|
self.ui.groupsListWidget.add_items(
|
|
[e["name"] for e in self.__stations_data["groups"]])
|
|
|
|
def print_stations(self) -> None:
|
|
"""Print stations (channels) in stationsListWidget."""
|
|
group = self.ui.groupsListWidget.selected_items()[0].text()
|
|
group_id = None
|
|
for e in self.__stations_data["groups"]:
|
|
if e["name"] == group:
|
|
group_id = e["id"]
|
|
|
|
self.ui.stationsListWidget.clear()
|
|
self.ui.stationsListWidget.add_items([
|
|
e["name"] for e in self.__stations_data["channels"]
|
|
if e["group_id"] == group_id
|
|
])
|
|
|
|
def set_volume(self, volume: int = None) -> None:
|
|
"""Set playback volume to given number or slider value."""
|
|
if not volume:
|
|
volume = self.ui.volumeHorizontalSlider.value
|
|
self.__audio.volume = volume / 100
|
|
|
|
def toggle_mute(self) -> None:
|
|
"""Toggle playback volume between 0 and DEFAULT_VOLUME."""
|
|
if self.ui.volumeHorizontalSlider.value == 0:
|
|
self.ui.volumeHorizontalSlider.value = self.previous_volume
|
|
icon = self.__icons.volume
|
|
self.set_volume(self.previous_volume)
|
|
else:
|
|
self.previous_volume = self.__audio.volume * 100
|
|
self.ui.volumeHorizontalSlider.value = 0
|
|
icon = self.__icons.muted
|
|
self.set_volume(0)
|
|
|
|
self.ui.volumeToolButton.icon = icon
|
|
|
|
def play_radio(self) -> None:
|
|
"""Play station selected by user."""
|
|
station = self.ui.stationsListWidget.selected_items()[0].text()
|
|
stream_url = None
|
|
for e in self.__stations_data["channels"]:
|
|
if e["name"] == station:
|
|
stream_url = f"http://stream.open.fm/{e['id']}"
|
|
|
|
self.__player.source = QUrl(stream_url)
|
|
self.window_title = f"Open FM - {station}"
|
|
|
|
# Required to avoid crashing. For some reason if you want to change
|
|
# the station for the first time, you need to stop and resume playback.
|
|
# If you won't, application would crash.
|
|
for _ in range(3):
|
|
self.toggle_player()
|
|
|
|
def toggle_player(self) -> None:
|
|
"""Toggle playback (play/stop)."""
|
|
pb_state = QMediaPlayer.PlaybackState
|
|
if self.__player.playback_state == pb_state.PlayingState:
|
|
self.__player.stop()
|
|
icon = self.__icons.play
|
|
elif self.__player.playback_state == pb_state.StoppedState:
|
|
self.__player.play()
|
|
icon = self.__icons.stop
|
|
else:
|
|
pass
|
|
|
|
self.ui.playbackToolButton.icon = icon
|