__author__ = ("Dylan Lloyd <dylan@psu.edu>")
__license__ = "BSD"
-# SETTINGS
+default_options = {
+ 'notifications' : 'true',
+ # NOTIFICATIONS must be a string due to issues noted here:
+ # http://bugs.python.org/issue974019
+ # ConfigParser.getboolean fails when falling back to the default value
+ # if the value type is bool.
+ 'youtube-dl' : '/usr/bin/youtube-dl',
+ 'default_icon' : '/usr/share/icons/gnome/48x48/mimetypes/gnome-mime-application-x-shockwave-flash.png',
+ 'youtube-dl_options' : '--no-progress --ignore-errors --continue --max-quality=22 -o "%(stitle)s---%(id)s.%(ext)s"'
+}
-USER = 'alphabethos' # pandora account name http://pandora.com/people<USER>
-DIR = '/home/dylan/pandora/' # where to download the videos - will not be automatically created
-YT_DL = '/usr/bin/youtube-dl' # Path to youtube-dl
-NOTIFICATIONS = True # False
-DEFAULT_ICON ='/usr/share/icons/gnome/48x48/mimetypes/gnome-mime-application-x-shockwave-flash.png' # for notifications
-YT_OPT = '--no-progress --ignore-errors --continue --max-quality=22 -o "%(stitle)s---%(id)s.%(ext)s"'
-# END OF SETTINGS
+import ConfigParser # This module has been renamed to configparser in python 3.0
+import sys
+import os
+
+CONFIG_FILE= os.path.join(os.path.expanduser('~'), '.i_like_pandora.config')
+config = ConfigParser.ConfigParser(default_options)
+loaded_files = config.read(CONFIG_FILE) # config.read returns an empty array if it fails.
+if len(loaded_files) == 0:
+ print 'Can\'t find a configuration file at', CONFIG_FILE
+ sys.exit()
+try:
+ USER = config.get('settings', 'username')
+ DIR = os.path.expanduser(config.get('settings', 'download_folder'))
+ NOTIFICATIONS = config.getboolean('settings', 'notifications')
+ YT_DL = config.get('settings', 'youtube-dl')
+ DEFAULT_ICON = config.get('settings', 'default_icon')
+except:
+ print 'There is a formatting error in the configuration file at', CONFIG_FILE
+ sys.exit()
from BeautifulSoup import BeautifulSoup
import urllib
import urllib2
-import os
import re
import copy
import shlex, subprocess
return stations
def fetch_tracks(stations):
- """ Takes a list of station tokens and returns a list of youtube search urls.
- What this should really do is just return the Title + Artist strings.
+ """ Takes a list of station tokens and returns a list of Title + Artist strings.
"""
search_strings = []
for station in stations:
search_strings.append(search_string)
i += 1
else:
- pass ## ERROR
+ # This would mean something strange has happened: there
+ # aren't the same number of titles and artist names on a
+ # station page.
+ pass
return search_strings
-def fetch_search_video_ids(search_strings):
- """ This takes a list of youtube search urls and tries to find the first result. It returns a list of youtube video ids.
- It really should take a list of ids instead.
+def search_youtube(search_strings):
+ """ This takes a list of search strings and tries to find the first result. It returns a list of the youtube video ids of those results.
"""
video_list = []
for search_string in search_strings:
page = urllib.urlopen(search_url)
page = BeautifulSoup(page)
result = page.find('div', attrs={'class':'video-main-content'})
- print result
if result == None:
print 'odd feedback for search, could not find div at ', search_url
continue
i += 1
return video_list
-def fetch_videos(videolist):
+def fetch_videos(video_list):
""" Uses subprocess to trigger a download using youtube-dl of the list created earlier, and triggers notifications if enabled. """
os.chdir(DIR)
args = shlex.split(YT_DL + ' ' + YT_OPT)
if NOTIFICATIONS: regex = re.compile("\[download\] Destination: (.+)")
- for item in videolist:
- if item:
- thread = subprocess.Popen(args + ["http://youtube.com/watch?v=" + item], stdout=subprocess.PIPE)
+ for video in video_list:
+ if video:
+ thread = subprocess.Popen(args + ["http://youtube.com/watch?v=" + video], stdout=subprocess.PIPE)
output = thread.stdout.read()
if NOTIFICATIONS:
video_file = regex.findall(output)
if not os.path.isfile(thumbnail):
opener = urllib2.build_opener()
try:
- page = opener.open('http://img.youtube.com/vi/' + item + '/1.jpg')
+ page = opener.open('http://img.youtube.com/vi/' + video + '/1.jpg')
thumb = page.read()
# The thumbnail really should be saved to
# ~/.thumbnails/normal (Thumbnail Managing
note = pynotify.Notification(title, 'video downloaded', thumbnail)
note.show()
-
def main():
stations = fetch_stations(USER)
if len(stations) == 0:
- print 'are you sure your pandora profile is public?'
- search_urls = fetch_tracks(stations)
- video_list = fetch_search_video_ids(search_urls)
- video_list = check_for_existing(video_list)
- fetch_videos(video_list)
+ print 'Are you sure your pandora profile is public? Can\'t seem to find any stations listed with your account.'
+ search_strings = fetch_tracks(stations)
+ videos = search_youtube(search_strings)
+ videos = check_for_existing(videos)
+ fetch_videos(videos)
if __name__ == "__main__":
main()