tavianator/Makefile

## Makefile
partymix_ui.py: partymix.ui
	pyuic4 partymix.ui >partymix_ui.py

## partymix.py
#!/usr/bin/env python3

import sys
import re
import os
import os.path
import time
import subprocess
import urllib.request
from html.parser import HTMLParser
from PyQt4.Qt import *
from PyQt4.QtCore import *
from PyQt4.QtGui import *
from PyQt4.QtWebKit import *
from partymix_ui import *
from multiprocessing import *

class YouTubeHTMLParser(HTMLParser):
  title = ""
  _title_seen = False

  def handle_starttag(self, tag, attrs):
    if tag == "title":
      self._title_seen = True
  def handle_endtag(self, tag):
    if self._title_seen:
      self._title_seen = False
      self.title = re.sub(r"\\n", "", self.title)
      self.title = re.sub(r" +", " ", self.title)
      self.title = re.sub(r"- *YouTube", "", self.title)
      self.title = self.title.strip()

  def handle_data(self, data):
    if self._title_seen:
      self.title += data

class PartyMix(QMainWindow):
  """Main window for the party mix!."""
  def __init__(self, items, to_download):
    self.items = items
    self.ui_items = []
    self.to_download = to_download

    QMainWindow.__init__(self)
    self.ui = Ui_PartyMix()
    self.ui.setupUi(self)

    self.ui.back.setIcon(QIcon.fromTheme("go-previous"))
    self.ui.youtube.setIcon(QIcon.fromTheme("go-home"))
    self.ui.google.setIcon(QIcon.fromTheme("edit-find"))

    self.connect(self.ui.youtube, SIGNAL("clicked()"), self.goto_youtube)
    self.connect(self.ui.google, SIGNAL("clicked()"), self.goto_google)

    self.ui.webView.page().setLinkDelegationPolicy(QWebPage.DelegateAllLinks)
    self.connect(self.ui.webView, SIGNAL("linkClicked(QUrl)"), self.link_clicked)

    self.timer = QTimer()
    self.connect(self.timer, SIGNAL("timeout()"), self.timer_action)
    self.timer.start(1000)

  def goto_youtube(self):
    self.ui.webView.load(QUrl("http://www.youtube.com/"))
  def goto_google(self):
    self.ui.webView.load(QUrl("http://www.google.com/"))

  def link_clicked(self, url):
    urlstr = url.toString()
    if re.match(r"http://www.youtube.com/watch.*", urlstr):
      if QMessageBox.question(self, "Add to playlist?",
           "Do you want to queue this song onto the playlist?",
           QMessageBox.Yes | QMessageBox.No, QMessageBox.Yes) == QMessageBox.Yes:
        self.queue(urlstr)
    else:
      self.ui.webView.load(url)

  def queue(self, url):
    f = urllib.request.urlopen(url)
    html = f.read()
    f.close()

    parser = YouTubeHTMLParser()
    try:
      parser.feed(str(html))
      parser.close()
    except:
      pass

    v = re.search(r"v=([^&]*)", url).group(1)

    layout = self.ui.playlistContents.layout()
    self.make_playlist_item(parser.title, v, layout)

    self.items.append({
      "v" : v,
      "state" : "queued"
    })
    self.to_download.put(len(self.items) - 1)

  def make_playlist_item(self, title, v, parent):
    i = len(self.ui_items)

    widget = QWidget()
    widget.setObjectName(v)

    item = QGridLayout(widget)
    item.setObjectName(v + "Layout")

    itemTitle = QLabel()
    itemTitle.setText(title)
    itemTitle.setObjectName(v + "Title")
    itemTitle.setWordWrap(True)
    item.addWidget(itemTitle, 0, 0, 1, 1)
    item.setColumnStretch(0, 1)

    def hide_playlist_item():
      widget.hide()
      item = self.items[i]
      item["state"] = "deleted"
      self.items[i] = item

    itemDelete = QToolButton()
    itemDelete.setText("X")
    itemDelete.setIcon(QIcon.fromTheme("edit-delete"))
    itemDelete.setObjectName(v + "Delete")
    self.connect(itemDelete, SIGNAL("clicked()"), hide_playlist_item)
    item.addWidget(itemDelete, 0, 1, 1, 1)

    itemProgress = QProgressBar()
    itemProgress.setObjectName(v + "Progress")
    item.addWidget(itemProgress, 1, 0, 1, 2)

    parent.insertWidget(parent.count() - 1, widget)
    self.ui.playlist.repaint()
    self.ui.playlist.ensureWidgetVisible(widget)
    self.ui_items.append({
      "widget" : widget,
      "progress" : itemProgress,
      "value" : 0
    })

  def timer_action(self):
    for i in range(len(self.ui_items)):
      item = self.items[i]
      ui_item = self.ui_items[i]
      if item["state"] == "downloading":
        ui_item["value"] = ui_item["value"] + (100 - ui_item["value"])/10
        ui_item["progress"].setValue(ui_item["value"])
      elif item["state"] == "downloaded" or item["state"] == "playing":
        ui_item["progress"].hide()
        if item["state"] == "playing":
          ui_item["widget"].setEnabled(False)

# Downloader process
def download(items, to_download, to_play):
  while True:
    i = to_download.get()
    if i == "done":
      return

    item = items[i]
    if item["state"] == "deleted":
      continue

    v = item["v"]
    path = "./downloads/" + v
    if not os.path.exists(path):
      item["state"] = "downloading"
      items[i] = item

      url = "http://www.youtube.com/watch?v=" + v
      formats = subprocess.getoutput("clive --query-formats " + url)
      match = re.search(r"^.*(fmt.*) : .*$", formats, re.MULTILINE)
      os.system("clive -f {} -O {} {}".format(match.group(1), path, url))

    item["state"] = "downloaded"
    items[i] = item

    to_play.put(i)

# Player process
def play(items, to_play):
  while True:
    i = to_play.get()
    if i == "done":
      return

    item = items[i]
    if item["state"] == "deleted":
      continue

    v = item["v"]
    item["state"] = "playing"
    items[i] = item

    os.system("vlc --one-instance --no-loop --playlist-enqueue ./downloads/" + v + " &")

    metadata = ""
    while v not in metadata:
      metadata = subprocess.getoutput("qdbus org.mpris.vlc /Player org.freedesktop.MediaPlayer.GetMetadata")
      time.sleep(1)

# Entry point
def main():
  manager = Manager()
  items = manager.list()

  to_download = Queue()
  to_play = Queue()

  downloader = Process(target = download, args = (items, to_download, to_play))
  downloader.start()

  player = Process(target = play, args = (items, to_play))
  player.start()

  try:
    app = QApplication(sys.argv)
    main = PartyMix(items, to_download)
    main.show()
    # main.setWindowState(main.windowState() | Qt.WindowMaximized)
    app.exec()
  finally:
    to_download.put("done")
    to_play.put("done")

    player.join()
    downloader.join()

if __name__ == "__main__":
  main()

## partymix.ui
<?xml version="1.0" encoding="UTF-8"?>
<ui version="4.0">
 <class>PartyMix</class>
 <widget class="QMainWindow" name="PartyMix">
  <property name="geometry">
   <rect>
    <x>0</x>
    <y>0</y>
    <width>800</width>
    <height>600</height>
   </rect>
  </property>
  <property name="windowTitle">
   <string>Party Mixer</string>
  </property>
  <widget class="QWidget" name="centralwidget">
   <property name="sizePolicy">
    <sizepolicy hsizetype="Preferred" vsizetype="Preferred">
     <horstretch>0</horstretch>
     <verstretch>0</verstretch>
    </sizepolicy>
   </property>
   <layout class="QHBoxLayout" name="horizontalLayout">
    <item>
     <widget class="QGroupBox" name="playlistBox">
      <property name="enabled">
       <bool>true</bool>
      </property>
      <property name="sizePolicy">
       <sizepolicy hsizetype="Preferred" vsizetype="Preferred">
        <horstretch>1</horstretch>
        <verstretch>0</verstretch>
       </sizepolicy>
      </property>
      <property name="title">
       <string>Playlist</string>
      </property>
      <layout class="QVBoxLayout" name="verticalLayout_2">
       <item>
        <widget class="QScrollArea" name="playlist">
         <property name="horizontalScrollBarPolicy">
          <enum>Qt::ScrollBarAlwaysOff</enum>
         </property>
         <property name="widgetResizable">
          <bool>true</bool>
         </property>
         <widget class="QWidget" name="playlistContents">
          <property name="geometry">
           <rect>
            <x>0</x>
            <y>0</y>
            <width>177</width>
            <height>550</height>
           </rect>
          </property>
          <layout class="QVBoxLayout" name="verticalLayout_3">
           <item>
            <spacer name="verticalSpacer">
             <property name="orientation">
              <enum>Qt::Vertical</enum>
             </property>
             <property name="sizeHint" stdset="0">
              <size>
               <width>20</width>
               <height>40</height>
              </size>
             </property>
            </spacer>
           </item>
          </layout>
         </widget>
        </widget>
       </item>
      </layout>
     </widget>
    </item>
    <item>
     <widget class="QGroupBox" name="searchBox">
      <property name="sizePolicy">
       <sizepolicy hsizetype="Preferred" vsizetype="Preferred">
        <horstretch>3</horstretch>
        <verstretch>0</verstretch>
       </sizepolicy>
      </property>
      <property name="title">
       <string>Search</string>
      </property>
      <layout class="QVBoxLayout" name="verticalLayout">
       <item>
        <layout class="QHBoxLayout" name="navbar" stretch="0,0,0,0">
         <property name="sizeConstraint">
          <enum>QLayout::SetDefaultConstraint</enum>
         </property>
         <property name="leftMargin">
          <number>0</number>
         </property>
         <item>
          <widget class="QPushButton" name="back">
           <property name="sizePolicy">
            <sizepolicy hsizetype="Fixed" vsizetype="Fixed">
             <horstretch>0</horstretch>
             <verstretch>0</verstretch>
            </sizepolicy>
           </property>
           <property name="styleSheet">
            <string notr="true"/>
           </property>
           <property name="text">
            <string>Back</string>
           </property>
          </widget>
         </item>
         <item>
          <widget class="QPushButton" name="youtube">
           <property name="sizePolicy">
            <sizepolicy hsizetype="Fixed" vsizetype="Fixed">
             <horstretch>0</horstretch>
             <verstretch>0</verstretch>
            </sizepolicy>
           </property>
           <property name="text">
            <string>YouTube</string>
           </property>
          </widget>
         </item>
         <item>
          <widget class="QPushButton" name="google">
           <property name="sizePolicy">
            <sizepolicy hsizetype="Fixed" vsizetype="Fixed">
             <horstretch>0</horstretch>
             <verstretch>0</verstretch>
            </sizepolicy>
           </property>
           <property name="text">
            <string>Google</string>
           </property>
          </widget>
         </item>
         <item>
          <spacer name="horizontalSpacer">
           <property name="orientation">
            <enum>Qt::Horizontal</enum>
           </property>
           <property name="sizeHint" stdset="0">
            <size>
             <width>40</width>
             <height>20</height>
            </size>
           </property>
          </spacer>
         </item>
        </layout>
       </item>
       <item>
        <widget class="QWebView" name="webView">
         <property name="url">
          <url>
           <string>http://www.youtube.com/</string>
          </url>
         </property>
        </widget>
       </item>
      </layout>
     </widget>
    </item>
   </layout>
  </widget>
 </widget>
 <customwidgets>
  <customwidget>
   <class>QWebView</class>
   <extends>QWidget</extends>
   <header>QtWebKit/QWebView</header>
  </customwidget>
 </customwidgets>
 <tabstops>
  <tabstop>webView</tabstop>
  <tabstop>back</tabstop>
 </tabstops>
 <resources/>
 <connections>
  <connection>
   <sender>back</sender>
   <signal>clicked()</signal>
   <receiver>webView</receiver>
   <slot>back()</slot>
   <hints>
    <hint type="sourcelabel">
     <x>295</x>
     <y>58</y>
    </hint>
    <hint type="destinationlabel">
     <x>356</x>
     <y>146</y>
    </hint>
   </hints>
  </connection>
 </connections>
</ui>
	partymix_ui.py: partymix.ui
	pyuic4 partymix.ui >partymix_ui.py
	#!/usr/bin/env python3

	import sys
	import re
	import os
	import os.path
	import time
	import subprocess
	import urllib.request
	from html.parser import HTMLParser
	from PyQt4.Qt import *
	from PyQt4.QtCore import *
	from PyQt4.QtGui import *
	from PyQt4.QtWebKit import *
	from partymix_ui import *
	from multiprocessing import *

	class YouTubeHTMLParser(HTMLParser):
	title = ""
	_title_seen = False

	def handle_starttag(self, tag, attrs):
	if tag == "title":
	self._title_seen = True
	def handle_endtag(self, tag):
	if self._title_seen:
	self._title_seen = False
	self.title = re.sub(r"\\n", "", self.title)
	self.title = re.sub(r" +", " ", self.title)
	self.title = re.sub(r"- *YouTube", "", self.title)
	self.title = self.title.strip()

	def handle_data(self, data):
	if self._title_seen:
	self.title += data

	class PartyMix(QMainWindow):
	"""Main window for the party mix!."""
	def __init__(self, items, to_download):
	self.items = items
	self.ui_items = []
	self.to_download = to_download

	QMainWindow.__init__(self)
	self.ui = Ui_PartyMix()
	self.ui.setupUi(self)

	self.ui.back.setIcon(QIcon.fromTheme("go-previous"))
	self.ui.youtube.setIcon(QIcon.fromTheme("go-home"))
	self.ui.google.setIcon(QIcon.fromTheme("edit-find"))

	self.connect(self.ui.youtube, SIGNAL("clicked()"), self.goto_youtube)
	self.connect(self.ui.google, SIGNAL("clicked()"), self.goto_google)

	self.ui.webView.page().setLinkDelegationPolicy(QWebPage.DelegateAllLinks)
	self.connect(self.ui.webView, SIGNAL("linkClicked(QUrl)"), self.link_clicked)

	self.timer = QTimer()
	self.connect(self.timer, SIGNAL("timeout()"), self.timer_action)
	self.timer.start(1000)

	def goto_youtube(self):
	self.ui.webView.load(QUrl("http://www.youtube.com/"))
	def goto_google(self):
	self.ui.webView.load(QUrl("http://www.google.com/"))

	def link_clicked(self, url):
	urlstr = url.toString()
	if re.match(r"http://www.youtube.com/watch.*", urlstr):
	if QMessageBox.question(self, "Add to playlist?",
	"Do you want to queue this song onto the playlist?",
	QMessageBox.Yes \| QMessageBox.No, QMessageBox.Yes) == QMessageBox.Yes:
	self.queue(urlstr)
	else:
	self.ui.webView.load(url)

	def queue(self, url):
	f = urllib.request.urlopen(url)
	html = f.read()
	f.close()

	parser = YouTubeHTMLParser()
	try:
	parser.feed(str(html))
	parser.close()
	except:
	pass

	v = re.search(r"v=([^&]*)", url).group(1)

	layout = self.ui.playlistContents.layout()
	self.make_playlist_item(parser.title, v, layout)

	self.items.append({
	"v" : v,
	"state" : "queued"
	})
	self.to_download.put(len(self.items) - 1)

	def make_playlist_item(self, title, v, parent):
	i = len(self.ui_items)

	widget = QWidget()
	widget.setObjectName(v)

	item = QGridLayout(widget)
	item.setObjectName(v + "Layout")

	itemTitle = QLabel()
	itemTitle.setText(title)
	itemTitle.setObjectName(v + "Title")
	itemTitle.setWordWrap(True)
	item.addWidget(itemTitle, 0, 0, 1, 1)
	item.setColumnStretch(0, 1)

	def hide_playlist_item():
	widget.hide()
	item = self.items[i]
	item["state"] = "deleted"
	self.items[i] = item

	itemDelete = QToolButton()
	itemDelete.setText("X")
	itemDelete.setIcon(QIcon.fromTheme("edit-delete"))
	itemDelete.setObjectName(v + "Delete")
	self.connect(itemDelete, SIGNAL("clicked()"), hide_playlist_item)
	item.addWidget(itemDelete, 0, 1, 1, 1)

	itemProgress = QProgressBar()
	itemProgress.setObjectName(v + "Progress")
	item.addWidget(itemProgress, 1, 0, 1, 2)

	parent.insertWidget(parent.count() - 1, widget)
	self.ui.playlist.repaint()
	self.ui.playlist.ensureWidgetVisible(widget)
	self.ui_items.append({
	"widget" : widget,
	"progress" : itemProgress,
	"value" : 0
	})

	def timer_action(self):
	for i in range(len(self.ui_items)):
	item = self.items[i]
	ui_item = self.ui_items[i]
	if item["state"] == "downloading":
	ui_item["value"] = ui_item["value"] + (100 - ui_item["value"])/10
	ui_item["progress"].setValue(ui_item["value"])
	elif item["state"] == "downloaded" or item["state"] == "playing":
	ui_item["progress"].hide()
	if item["state"] == "playing":
	ui_item["widget"].setEnabled(False)

	# Downloader process
	def download(items, to_download, to_play):
	while True:
	i = to_download.get()
	if i == "done":
	return

	item = items[i]
	if item["state"] == "deleted":
	continue

	v = item["v"]
	path = "./downloads/" + v
	if not os.path.exists(path):
	item["state"] = "downloading"
	items[i] = item

	url = "http://www.youtube.com/watch?v=" + v
	formats = subprocess.getoutput("clive --query-formats " + url)
	match = re.search(r"^.(fmt.) : .*$", formats, re.MULTILINE)
	os.system("clive -f {} -O {} {}".format(match.group(1), path, url))

	item["state"] = "downloaded"
	items[i] = item

	to_play.put(i)

	# Player process
	def play(items, to_play):
	while True:
	i = to_play.get()
	if i == "done":
	return

	item = items[i]
	if item["state"] == "deleted":
	continue

	v = item["v"]
	item["state"] = "playing"
	items[i] = item

	os.system("vlc --one-instance --no-loop --playlist-enqueue ./downloads/" + v + " &")

	metadata = ""
	while v not in metadata:
	metadata = subprocess.getoutput("qdbus org.mpris.vlc /Player org.freedesktop.MediaPlayer.GetMetadata")
	time.sleep(1)

	# Entry point
	def main():
	manager = Manager()
	items = manager.list()

	to_download = Queue()
	to_play = Queue()

	downloader = Process(target = download, args = (items, to_download, to_play))
	downloader.start()

	player = Process(target = play, args = (items, to_play))
	player.start()

	try:
	app = QApplication(sys.argv)
	main = PartyMix(items, to_download)
	main.show()
	# main.setWindowState(main.windowState() \| Qt.WindowMaximized)
	app.exec()
	finally:
	to_download.put("done")
	to_play.put("done")

	player.join()
	downloader.join()

	if __name__ == "__main__":
	main()
	<?xml version="1.0" encoding="UTF-8"?>
	<ui version="4.0">
	<class>PartyMix</class>
	<widget class="QMainWindow" name="PartyMix">
	<property name="geometry">
	<rect>
	<x>0</x>
	<y>0</y>
	<width>800</width>
	<height>600</height>
	</rect>
	</property>
	<property name="windowTitle">
	<string>Party Mixer</string>
	</property>
	<widget class="QWidget" name="centralwidget">
	<property name="sizePolicy">
	<sizepolicy hsizetype="Preferred" vsizetype="Preferred">
	<horstretch>0</horstretch>
	<verstretch>0</verstretch>
	</sizepolicy>
	</property>
	<layout class="QHBoxLayout" name="horizontalLayout">
	<item>
	<widget class="QGroupBox" name="playlistBox">
	<property name="enabled">
	<bool>true</bool>
	</property>
	<property name="sizePolicy">
	<sizepolicy hsizetype="Preferred" vsizetype="Preferred">
	<horstretch>1</horstretch>
	<verstretch>0</verstretch>
	</sizepolicy>
	</property>
	<property name="title">
	<string>Playlist</string>
	</property>
	<layout class="QVBoxLayout" name="verticalLayout_2">
	<item>
	<widget class="QScrollArea" name="playlist">
	<property name="horizontalScrollBarPolicy">
	<enum>Qt::ScrollBarAlwaysOff</enum>
	</property>
	<property name="widgetResizable">
	<bool>true</bool>
	</property>
	<widget class="QWidget" name="playlistContents">
	<property name="geometry">
	<rect>
	<x>0</x>
	<y>0</y>
	<width>177</width>
	<height>550</height>
	</rect>
	</property>
	<layout class="QVBoxLayout" name="verticalLayout_3">
	<item>
	<spacer name="verticalSpacer">
	<property name="orientation">
	<enum>Qt::Vertical</enum>
	</property>
	<property name="sizeHint" stdset="0">
	<size>
	<width>20</width>
	<height>40</height>
	</size>
	</property>
	</spacer>
	</item>
	</layout>
	</widget>
	</widget>
	</item>
	</layout>
	</widget>
	</item>
	<item>
	<widget class="QGroupBox" name="searchBox">
	<property name="sizePolicy">
	<sizepolicy hsizetype="Preferred" vsizetype="Preferred">
	<horstretch>3</horstretch>
	<verstretch>0</verstretch>
	</sizepolicy>
	</property>
	<property name="title">
	<string>Search</string>
	</property>
	<layout class="QVBoxLayout" name="verticalLayout">
	<item>
	<layout class="QHBoxLayout" name="navbar" stretch="0,0,0,0">
	<property name="sizeConstraint">
	<enum>QLayout::SetDefaultConstraint</enum>
	</property>
	<property name="leftMargin">
	<number>0</number>
	</property>
	<item>
	<widget class="QPushButton" name="back">
	<property name="sizePolicy">
	<sizepolicy hsizetype="Fixed" vsizetype="Fixed">
	<horstretch>0</horstretch>
	<verstretch>0</verstretch>
	</sizepolicy>
	</property>
	<property name="styleSheet">
	<string notr="true"/>
	</property>
	<property name="text">
	<string>Back</string>
	</property>
	</widget>
	</item>
	<item>
	<widget class="QPushButton" name="youtube">
	<property name="sizePolicy">
	<sizepolicy hsizetype="Fixed" vsizetype="Fixed">
	<horstretch>0</horstretch>
	<verstretch>0</verstretch>
	</sizepolicy>
	</property>
	<property name="text">
	<string>YouTube</string>
	</property>
	</widget>
	</item>
	<item>
	<widget class="QPushButton" name="google">
	<property name="sizePolicy">
	<sizepolicy hsizetype="Fixed" vsizetype="Fixed">
	<horstretch>0</horstretch>
	<verstretch>0</verstretch>
	</sizepolicy>
	</property>
	<property name="text">
	<string>Google</string>
	</property>
	</widget>
	</item>
	<item>
	<spacer name="horizontalSpacer">
	<property name="orientation">
	<enum>Qt::Horizontal</enum>
	</property>
	<property name="sizeHint" stdset="0">
	<size>
	<width>40</width>
	<height>20</height>
	</size>
	</property>
	</spacer>
	</item>
	</layout>
	</item>
	<item>
	<widget class="QWebView" name="webView">
	<property name="url">
	<url>
	<string>http://www.youtube.com/</string>
	</url>
	</property>
	</widget>
	</item>
	</layout>
	</widget>
	</item>
	</layout>
	</widget>
	</widget>
	<customwidgets>
	<customwidget>
	<class>QWebView</class>
	<extends>QWidget</extends>
	<header>QtWebKit/QWebView</header>
	</customwidget>
	</customwidgets>
	<tabstops>
	<tabstop>webView</tabstop>
	<tabstop>back</tabstop>
	</tabstops>
	<resources/>
	<connections>
	<connection>
	<sender>back</sender>
	<signal>clicked()</signal>
	<receiver>webView</receiver>
	<slot>back()</slot>
	<hints>
	<hint type="sourcelabel">
	<x>295</x>
	<y>58</y>
	</hint>
	<hint type="destinationlabel">
	<x>356</x>
	<y>146</y>
	</hint>
	</hints>
	</connection>
	</connections>
	</ui>