Merge branch 'dev'

[i_like_pandora.git] / likes_pandora.py

diff --git a/likes_pandora.py b/likes_pandora.py
index 5e376ff..181cda7 100755 (executable)
--- a/likes_pandora.py
+++ b/likes_pandora.py
@@ -4,20 +4,40 @@
 __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
@@ -42,8 +62,7 @@ def fetch_stations(user):
     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:
@@ -64,12 +83,14 @@ def fetch_tracks(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:
@@ -100,14 +121,14 @@ def check_for_existing(video_list):
         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)
@@ -122,7 +143,7 @@ def fetch_videos(videolist):
                 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
@@ -143,15 +164,14 @@ def fetch_videos(videolist):
                     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()