Ticket #3313: episode_title_db.diff

mythvideo/mythvideo/scripts/imdbpy.py

@@ -9 +9 @@
 
 This wrapper script is written by
 Pekka JÀÀskelÀinen (gmail: pekka.jaaskelainen).
-"""  
+"""
 
+import os
 import sys
 import optparse
 import re
+import socket
+import shlex
 
 try:
         import imdb
@@ -22 +25 @@
                 "from (http://imdbpy.sourceforge.net/?page=download)"
         sys.exit(1)
 
+try:
+        import MySQLdb
+        db_support = True
+except:
+        db_support = False
 
+db = None
+episode_title_format = '%(series_title)s S%(season)02d E%(episode)02d %(episode_title)s'
 
+def init_db():
+        global db
+        try:
+                config = shlex.shlex(open(os.path.expanduser('~/.mythtv/mysql.txt')))
+        except:
+                print "Error opening ~/.mythtv/mysql.txt"
+                return False
+                
+        
+        token = config.get_token()
+        db_host = db_user = db_password = None
+        while  token != config.eof and (db_host == None or db_user == None or db_password == None):
+                if token == "DBHostName":
+                        if config.get_token() == "=":
+                                db_host = config.get_token()
+                elif token == "DBUserName":
+                        if config.get_token() == "=":
+                                db_user = config.get_token()
+                elif token == "DBPassword":
+                        if config.get_token() == "=":
+                                db_password = config.get_token()
+                                                                
+                token = config.get_token()
+        db = MySQLdb.connect(user=db_user, host=db_host, passwd=db_password, 
+                db="mythconverg")
+        return True
+
+def mythtv_setting(value, hostname = '%'):
+        """
+        Returns the value for the given MythTV setting.
+        
+        Returns None if the settings was not found. If multiple rows are
+        found (multiple hostnames), returns the value of the first one.
+        """
+        global db
+        c = db.cursor()
+        c.execute("""
+                SELECT data
+                FROM settings
+                WHERE value LIKE(%s) AND hostname LIKE(%s) LIMIT 1""", 
+                (value, hostname))
+        row = c.fetchone()
+        c.close()
+        
+        if row is not None:
+                return row[0]
+        else:
+                return None
+
 def detect_series_title(search_string):
         """
         Detects a series episode title.
@@ -175 +234 @@
         genres = None
         countries = None
         
-        episode_title_format = '%(series_title)s S%(season)02d E%(episode)02d %(episode_title)s'
-        
         def __init__(self):
                 pass
         
@@ -190 +247 @@
                 metadata = ""
                 if self.series_episode == True and self.season is not None and \
                         self.episode is not None:
-                        metadata += 'Title:' + self.episode_title_format % \
+                        metadata += 'Title:' + episode_title_format % \
                                 {
                                         'series_title': self.series_title,
                                         'season': int(self.season),
@@ -316 +373 @@
         return None
         
 def main():
+        global episode_title_format
+        
         p = optparse.OptionParser()
         p.add_option('--version', '-v', action="store_true", default=False,
                 help="display 1-line describing name, version, author etc")
@@ -345 +404 @@
 episodes."""
                 sys.exit(0)
 
+        if db_support:
+                if init_db():
+                        episode_title_format_setting = mythtv_setting('VideoEpisodeTitleFormat', socket.gethostname())
+                        if episode_title_format_setting:
+                                episode_title_format = episode_title_format_setting
+
         if options.movie_search is not None:
                 results = title_search(options.movie_search.decode("utf8"))
                 for result in results: