getRadioList.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-
##############
"""
made in October 2019 by Axel Schneider
https://github.com/Axel-Erfurt/
Credits: André P. Santos (andreztz) for pyradios
https://github.com/andreztz/pyradios
Copyright (c) 2018 André P. Santos
radio-browser
http://www.radio-browser.info/webservice
"""
##############
from PyQt5.QtWidgets import (QApplication, QMainWindow, QPlainTextEdit, QLineEdit, QComboBox, QPushButton, QFileDialog,
                             QWidget, QButtonGroup, QHBoxLayout, QVBoxLayout, QGroupBox, QAction, QMenu, QMessageBox,
                             QLabel)
from PyQt5.QtGui import QIcon, QTextCursor, QTextOption, QPixmap
from PyQt5.QtCore import Qt, QUrl
from radios import RadioBrowser
from PyQt5.QtMultimedia import QMediaPlayer, QMediaContent
# from PyQt5.Qt import QClipboard
from urllib import request
import yaml

genres = """Americana
Bluegrass
Country
New Country
Classic Country
Country Rock
Cowboy / Western
Folk
Folk Rock
Grunge
Hard Rock
Classic Rock
Blues
Oldies
Pop
Rock
Classic
Beat
Metal
"""


class MainWindow(QMainWindow):
    def __init__(self):
        try:
            with open('favs.yaml', 'r') as yaml_stream:
                self.fav_list = yaml.load(yaml_stream, Loader=yaml.SafeLoader)
        except:
            print("Cannot read yaml config file, check formatting.")
            self.fav_list = None

        super(MainWindow, self).__init__()
        self.setGeometry(0, 0, 700, 400)
        self.setContentsMargins(6, 6, 6, 6)
        self.setWindowTitle("pyRadioQt")
        self.setWindowIcon(QIcon('./icon/icon.png'))

        self.uiGenreCombo()
        self.uiSearchField()

        self.path = './icon/icon.png'
        self.pix = QPixmap(self.path)
        self.label_image = QLabel()
        self.label_image.setPixmap(QPixmap(self.pix))

        self.field = QPlainTextEdit()
        self.field.setContextMenuPolicy(Qt.CustomContextMenu)
        self.field.customContextMenuRequested.connect(self.contextMenuRequested)
        self.field.cursorPositionChanged.connect(self.selectLine)
        self.field.setWordWrapMode(QTextOption.NoWrap)

        self.saveButton = QPushButton("Save as txt")
        self.saveButton.setIcon(QIcon.fromTheme("document-save"))
        self.saveButton.clicked.connect(self.saveStations)
        self.savePlaylistButton = QPushButton("Save as m3u")
        self.savePlaylistButton.setIcon(QIcon.fromTheme("document-save"))
        self.savePlaylistButton.clicked.connect(self.savePlaylist)

        # Toolbar
        self.tb = self.addToolBar("tools")
        self.tb.setContextMenuPolicy(Qt.PreventContextMenu)
        self.tb.setMovable(False)
        self.tb.addWidget(self.searchField)
        self.tb.addWidget(self.saveButton)
        self.tb.addWidget(self.savePlaylistButton)
        self.tb.addSeparator()
        self.tb.addWidget(self.genreCombo)

        # Main Layout
        self.createFavoriteLayout()
        self.mainWidget = QWidget(self)
        self.mainLayout = QVBoxLayout(self.mainWidget)

        self.centerLayout = QHBoxLayout()
        self.centerLayout.addWidget(self.label_image)
        self.centerLayout.addWidget(self.field)
        self.mainLayout.addLayout(self.centerLayout)

        self.mainLayout.addWidget(self.horizontalGroupBox)
        self.mainWidget.setLayout(self.mainLayout)

        self.setCentralWidget(self.mainWidget)

        # player ###
        self.player = QMediaPlayer()
        self.player.metaDataChanged.connect(self.metaDataChanged)
        self.startButton = QPushButton("Play")
        self.startButton.setIcon(QIcon.fromTheme("media-playback-start"))
        self.startButton.clicked.connect(self.getURLtoPlay)
        self.stopButton = QPushButton("Stop")
        self.stopButton.setIcon(QIcon.fromTheme("media-playback-stop"))
        self.stopButton.clicked.connect(self.stopPlayer)
        self.statusBar().addPermanentWidget(self.startButton)
        self.statusBar().addPermanentWidget(self.stopButton)
        # actions
        self.getNameAction = QAction(QIcon.fromTheme("edit-copy"), "copy Station Name", self, triggered=self.getName)
        self.getUrlAction = QAction(QIcon.fromTheme("edit-copy"), "copy Station URL", self, triggered=self.getURL)
        self.getNameAndUrlAction = QAction(QIcon.fromTheme("edit-copy"), "copy Station Name,URL", self, triggered=self.getNameAndUrl)
        self.getURLtoPlayAction = QAction(QIcon.fromTheme("media-playback-start"), "play Station", self, shortcut="F6", triggered=self.getURLtoPlay)
        self.addAction(self.getURLtoPlayAction)
        self.stopPlayerAction = QAction(QIcon.fromTheme("media-playback-stop"), "stop playing", self, shortcut="F7", triggered=self.stopPlayer)
        self.addAction(self.stopPlayerAction)
        self.helpAction = QAction(QIcon.fromTheme("help-info"), "Help", self, shortcut="F1", triggered=self.showHelp)
        self.addAction(self.helpAction)
        self.statusBar().showMessage("Welcome", 0)

    def uiGenreCombo(self):
        self.genreList = genres.splitlines()

        self.genreCombo = QComboBox()
        self.genreCombo.setFixedWidth(150)
        self.genreCombo.currentIndexChanged.connect(self.genreSearch)

        self.genreCombo.addItem("choose Genre")
        for m in self.genreList:
            self.genreCombo.addItem(m)

    def uiSearchField(self):
        self.searchField = QLineEdit()
        self.searchField.setFixedWidth(250)
        self.searchField.addAction(QIcon.fromTheme("edit-find"), 0)
        self.searchField.setPlaceholderText("type search term and press RETURN ")
        self.searchField.returnPressed.connect(self.findStations)

    def createFavoriteLayout(self):
        self.horizontalGroupBox = QGroupBox("Favorites")
        layout = QHBoxLayout()
        if self.fav_list is not None:
            self.buttongroup = QButtonGroup()
            self.buttongroup.buttonClicked[int].connect(self.handleButtonClicked)

            i = 1
            for station in self.fav_list:
                #print(station)
                self.button = QPushButton(station, self)
                self.buttongroup.addButton(self.button, i)
                i = i + 1
                layout.addWidget(self.button)
        else:
            l1 = QLabel()
            l1.setText("No Favorites")
            layout.addWidget(l1)

        self.horizontalGroupBox.setLayout(layout)

    def handleButtonClicked(self, id):
        for button in self.buttongroup.buttons():
            if button is self.buttongroup.button(id):
                #print(button.text() + " Was Clicked ")
                for station, url in self.fav_list.items():
                    if station == button.text():
                        #print(url[0])
                        self.getURLtoPlay(True, station, url[0])

    def genreSearch(self):
        if self.genreCombo.currentIndex() > 0:
            self.searchField.setText(self.genreCombo.currentText())
            self.findStations()

    def getName(self):
        t = self.field.textCursor().selectedText().partition(",")[0]
        clip = QApplication.clipboard()
        clip.setText(t)

    def getURL(self):
        t = self.field.textCursor().selectedText().partition(",")[2]
        clip = QApplication.clipboard()
        clip.setText(t)

    def getNameAndUrl(self):
        t = self.field.textCursor().selectedText()
        clip = QApplication.clipboard()
        clip.setText(t)

    def selectLine(self):
        tc = self.field.textCursor()
        tc.select(QTextCursor.LineUnderCursor)
        tc.movePosition(QTextCursor.StartOfLine, QTextCursor.MoveAnchor)
        tc.movePosition(QTextCursor.EndOfBlock, QTextCursor.KeepAnchor)
        self.field.setTextCursor(tc)

    def showHelp(self):
        QMessageBox.information(self, "Information", "F6 -> play Station (from line where cursor is)\n\
                                                      F7 -> stop playing")

    def stopPlayer(self):
        self.player.stop()
        self.statusBar().showMessage("Player stopped", 0)

    # QPlainTextEdit contextMenu
    def contextMenuRequested(self, point):
        cmenu = QMenu()
        if not self.field.toPlainText() == "":
            cmenu.addAction(self.getNameAction)
            cmenu.addAction(self.getUrlAction)
            cmenu.addAction(self.getNameAndUrlAction)
            cmenu.addSeparator()
            cmenu.addAction(self.getURLtoPlayAction)
            cmenu.addAction(self.stopPlayerAction)
            cmenu.addSeparator()
            cmenu.addAction(self.helpAction)
        cmenu.exec_(self.field.mapToGlobal(point))

    def getURLtoPlay(self, fav=False, name="", url_fav=""):
        url = ""
        stext = ""
        if fav:
            # print("url_fav=",url_fav)
            # print("name in func:", name)
            stext = name
            url = url_fav
        else:
            tc = self.field.textCursor()
            rtext = tc.selectedText().partition(",")[2]
            stext = tc.selectedText().partition(",")[0]
            if rtext.endswith(".pls"):
                url = self.getURLfromPLS(rtext)
            elif rtext.endswith(".m3u"):
                url = self.getURLfromM3U(rtext)
            else:
                url = rtext
        # print("stream url=", url)

        self.player.setMedia(QMediaContent(QUrl(url)))
        self.player.play()
        self.statusBar().showMessage("%s %s" % ("playing", stext), 0)

    def metaDataChanged(self):
        if self.player.isMetaDataAvailable():
            trackInfo = (self.player.metaData("Title"))
            trackInfo2 = (self.player.metaData("Comment"))
            if trackInfo is not None:
                self.statusBar().showMessage(trackInfo, 0)
                if trackInfo2 is not None:
                    self.statusBar().showMessage("%s %s" % (trackInfo, trackInfo2))

    def getURLfromPLS(self, inURL):
        if "&" in inURL:
            inURL = inURL.partition("&")[0]
        response = request.urlopen(inURL)
        html = response.read().decode("utf-8").splitlines()
        if len(html) > 3:
            if "http" in str(html[1]):
                t = str(html[1])
            elif "http" in str(html[2]):
                t = str(html[2])
            elif "http" in str(html[3]):
                t = str(html[3])
        elif len(html) > 2:
            if "http" in str(html[1]):
                t = str(html[1])
            elif "http" in str(html[2]):
                t = str(html[2])
        else:
            t = str(html[0])
        url = t.partition("=")[2].partition("'")[0]
#        print(url)
        return (url)

    def getURLfromM3U(self, inURL):
        if "?u=" in inURL:
            inURL = inURL.partition("?u=")[2]
        if "&" in inURL:
            inURL = inURL.partition("&")[0]
        response = request.urlopen(inURL)
        html = response.read().splitlines()
        if len(html) > 1:
            if "http" in str(html[1]):
                t = str(html[1])
            else:
                t = str(html[0])
        else:
            t = str(html[0])
        url = t.partition("'")[2].partition("'")[0]
#        print(url)

        return (url)

    def findStations(self):
        self.field.setPlainText("")
        mysearch = self.searchField.text()
        self.statusBar().showMessage("searching ...")
        rb = RadioBrowser()
        myparams = {'name': 'search', 'nameExact': 'false'}

        for key in myparams.keys():
            if key == "name":
                myparams[key] = mysearch

        self.r = rb.station_search(params=myparams)

        n = ""
        m = ""
        for i in range(len(self.r)):
            for key, value in self.r[i].items():
                if str(key) == "favicon":
                    self.path = value
                    print(self.path)
                if str(key) == "name":
                    n = value.replace(",", " ")
        #            print (n)
                if str(key) == "url":
                    m = value
                    self.field.appendPlainText("%s,%s" % (n, m))
#        self.combo.setCurrentIndex(0)
        if not self.field.toPlainText() == "":
            self.statusBar().showMessage("found " + str(self.field.toPlainText().count('\n')+1) + " '" +
                                         self.searchField.text() + "' Stations")
        else:
            self.statusBar().showMessage("nothing found", 0)
#        self.field.textCursor().movePosition(QTextCursor.Start, Qt.MoveAnchor)

    def saveStations(self):
        if not self.field.toPlainText() == "":
            path, _ = QFileDialog.getSaveFileName(None, "RadioStations", self.searchField.text() +
                                                  ".txt", "Text Files (*.txt)")
            if path:
                s = self.field.toPlainText()
                with open(path, 'w') as f:
                    f.write(s)
                    f.close()
                    self.statusBar().showMessage("saved!", 0)

    def savePlaylist(self):
        if not self.field.toPlainText() == "":
            path, _ = QFileDialog.getSaveFileName(None, "RadioStations", self.searchField.text() + ".m3u",
                                                  "Playlist Files (*.m3u)")
            if path:
                result = ""
                s = self.field.toPlainText()
                st = []
                for line in s.splitlines():
                    st.append(line)
                result += "#EXTM3U"
                result += '\n'
                for x in range(len(st)):
                    result += "#EXTINF:" + str(x) + "," + st[x].partition(",")[0]
                    result += '\n'
                    result += st[x].partition(",")[2]
                    result += '\n'
                with open(path, 'w') as f:
                    f.write(result)
                    f.close()
                    self.statusBar().showMessage("saved!", 0)


if __name__ == '__main__':

    import sys

    app = QApplication(sys.argv)

    mainWin = MainWindow()
    
    mainWin.show()
    mainWin.searchField.setFocus()
    sys.exit(app.exec_())