Metadata base

couchpotato/core/loader.py

@@ -28,6 +28,7 @@ class Loader:
             'torrent_provider' : (20, 'couchpotato.core.providers.torrent', os.path.join(providers, 'torrent')),
             'trailer_provider' : (20, 'couchpotato.core.providers.trailer', os.path.join(providers, 'trailer')),
             'subtitle_provider' : (20, 'couchpotato.core.providers.subtitle', os.path.join(providers, 'subtitle')),
+            'metadata_provider' : (25, 'couchpotato.core.providers.metadata', os.path.join(providers, 'metadata')),
         }
 
         for type, tuple in self.paths.iteritems():

couchpotato/core/plugins/metadata/__init__.py

@@ -0,0 +1,6 @@
+from .main import MetaData
+
+def start():
+    return MetaData()
+
+config = []

couchpotato/core/plugins/metadata/main.py

@@ -0,0 +1,22 @@
+from couchpotato.core.event import addEvent, fireEvent
+from couchpotato.core.logger import CPLog
+from couchpotato.core.plugins.base import Plugin
+
+log = CPLog(__name__)
+
+
+class MetaData(Plugin):
+
+    def __init__(self):
+        addEvent('renaming.after', self.add)
+
+        self.registerStatic(__file__)
+
+        def test():
+            fireEvent('metadata.create')
+
+        addEvent('app.load', test)
+
+    def add(self, data = {}):
+        log.info('Getting meta data')
+

couchpotato/core/providers/metadata/base.py

@@ -0,0 +1,26 @@
+from couchpotato.core.event import addEvent
+from couchpotato.core.logger import CPLog
+from couchpotato.core.plugins.base import Plugin
+
+log = CPLog(__name__)
+
+
+class MetaDataBase(Plugin):
+
+    def __init__(self):
+        addEvent('metadata.create', self.create)
+
+    def create(self):
+        print 'create metadata %s' % __name__
+
+    def getFanartName(self):
+        return
+
+    def getThumbnailName(self):
+        return
+
+    def getNfoName(self):
+        return
+
+    def getNfo(self):
+        return

couchpotato/core/providers/metadata/mediabrowser/__init__.py

@@ -0,0 +1,6 @@
+from .main import MediaBrowser
+
+def start():
+    return MediaBrowser()
+
+config = []

couchpotato/core/providers/metadata/mediabrowser/main.py

@@ -0,0 +1,5 @@
+from couchpotato.core.providers.metadata.base import MetaDataBase
+
+class MediaBrowser(MetaDataBase):
+
+    pass

couchpotato/core/providers/metadata/sonyps3/__init__.py

@@ -0,0 +1,6 @@
+from .main import SonyPS3
+
+def start():
+    return SonyPS3()
+
+config = []

couchpotato/core/providers/metadata/sonyps3/main.py

@@ -0,0 +1,5 @@
+from couchpotato.core.providers.metadata.base import MetaDataBase
+
+class SonyPS3(MetaDataBase):
+
+    pass

couchpotato/core/providers/metadata/wdtv/__init__.py

@@ -0,0 +1,6 @@
+from .main import WDTV
+
+def start():
+    return WDTV()
+
+config = []

couchpotato/core/providers/metadata/wdtv/main.py

@@ -0,0 +1,5 @@
+from couchpotato.core.providers.metadata.base import MetaDataBase
+
+class WDTV(MetaDataBase):
+
+    pass

couchpotato/core/providers/metadata/xbmc/__init__.py

@@ -0,0 +1,6 @@
+from .main import XBMC
+
+def start():
+    return XBMC()
+
+config = []

couchpotato/core/providers/metadata/xbmc/main.py

@@ -0,0 +1,238 @@
+from couchpotato.core.providers.metadata.base import MetaDataBase
+from xml.etree.ElementTree import Element, SubElement, Comment, tostring
+import xml.dom.minidom
+
+class XBMC(MetaDataBase):
+
+    def getFanartName(self, root):
+        return '%s-fanart.jpg' % root
+
+    def getThumbnailName(self, root):
+        return '%s.tbn' % root
+
+    def getNfoName(self, root):
+        return '%s.nfo' % root
+
+    def getNfo(self):
+        pass
+
+
+"""
+    def write_nfo(self, path, url = True, xml = True):
+
+        self.out_string = ''
+
+        if xml:
+            self.out_string = self._generate_nfo_xml()
+
+        if url:
+            self.out_string = self.out_string + self.nfo_string
+
+        try:
+            f = open(path, 'w')
+            f.write(self.out_string)
+            f.close()
+        except:
+            raise NfoError("Couldn't write nfo")
+
+    def _generate_nfo_xml(self):
+        nfoxml = Element('movie')
+
+        try:
+            title = SubElement(nfoxml, 'title')
+            title.text = self.tmdb_data['name']
+        except:
+            pass
+
+        try:
+            originaltitle = SubElement(nfoxml, 'originaltitel')
+            originaltitle.text = self.tmdb_data['original_name']
+        except:
+            pass
+
+        try:
+            rating = SubElement(nfoxml, 'rating')
+            rating.text = str(self.tmdb_data['rating'])
+        except:
+            pass
+
+        try:
+            year = SubElement(nfoxml, 'year')
+            year.text = self.tmdb_data['released'][:4]
+        except:
+            pass
+
+        try:
+            votes = SubElement(nfoxml, 'votes')
+            votes.text = str(self.tmdb_data['votes'])
+        except:
+            pass
+
+        try:
+            plot = SubElement(nfoxml, 'outline')
+            plot.text = self.tmdb_data['overview']
+        except:
+            pass
+
+        for genre in self.tmdb_data['genres']:
+            genres = SubElement(nfoxml, 'genre')
+            genres.text = genre['name']
+
+        try:
+            runtime = SubElement(nfoxml, 'runtime')
+            runtime.text = str(self.tmdb_data['runtime']) + " min"
+        except:
+            pass
+
+        try:
+            premiered = SubElement(nfoxml, 'premiered')
+            premiered.text = self.tmdb_data['released']
+        except:
+            pass
+
+        try:
+            mpaa = SubElement(nfoxml, 'mpaa')
+            mpaa.text = self.tmdb_data['certification']
+        except:
+            pass
+
+        try:
+            id = SubElement(nfoxml, 'id')
+            id.text = self.tmdb_data['imdb_id']
+        except:
+            pass
+
+        # Clean up the xml and return it
+        nfoxml = xml.dom.minidom.parseString(tostring(nfoxml))
+        xml_string = nfoxml.toprettyxml(indent = '  ')
+        text_re = re.compile('>\n\s+([^<>\s].*?)\n\s+</', re.DOTALL)
+        xml_string = text_re.sub('>\g<1></', xml_string)
+
+        return xml_string.encode('utf-8')
+
+    def _get_fanart(self, min_height, min_width):
+        '''  Fetches the fanart for the specified imdb_id and saves it to dir.
+        Arguments
+
+        min_height/width: Sets lowest acceptable resolution fanart.  0 means
+        disregard.  If no fanart available at specified resolution or greater, then
+        we disregard.
+        '''
+        images = [image['image'] for image in self.tmdb_data['backdrops'] if image['image'].get('size') == 'original']
+        if len(images) == 0:
+            return
+
+        return self._get_image(images, min_height, min_width)
+
+    def get_fanart_url(self, min_height, min_width):
+        return self._get_fanart(min_height, min_width)['url']
+
+    def write_fanart(self, filename_root, path, min_height, min_width):
+        fanart_url = self.get_fanart_url(min_height, min_width)
+        #fetch and write to disk
+        dest = os.path.join(path, filename_root)
+        try:
+            f = open(dest, 'wb')
+        except:
+            raise IOError("Can't open for writing: %s" % dest)
+
+        response = urllib2.urlopen(fanart_url)
+        f.write(response.read())
+        f.close()
+
+        return True
+
+    def _get_poster(self, min_height, min_width):
+        '''  Fetches the poster for the specified imdb_id and saves it to dir.
+        Arguments
+
+        min_height/width: Sets lowest acceptable resolution poster.  0 means
+        disregard.  If no poster available at specified resolution or greater, then
+        we disregard.
+        '''
+        images = [image['image'] for image in self.tmdb_data['posters'] if image['image'].get('size') == 'original']
+        if len(images) == 0:
+            return
+
+        return self._get_image(images, min_height, min_width)
+
+    def get_poster_url(self, min_height, min_width):
+        return self._get_poster(min_height, min_width)['url']
+
+    def write_poster(self, filename_root, path, min_height, min_width):
+        poster_url = self.get_poster_url(min_height, min_width)
+        dest = os.path.join(path, filename_root)
+
+        try:
+            f = open(dest, 'wb')
+        except:
+            raise IOError("Can't open for writing: %s" % dest)
+
+        response = urllib2.urlopen(poster_url)
+        f.write(response.read())
+        f.close()
+
+        return True
+
+    def _get_tmdb_imdb(self):
+        url = "http://api.themoviedb.org/2.1/Movie.imdbLookup/en/json/%s/%s" % (__tmdb_apikey__, self.imdbid)
+
+        count = 0
+        while 1:
+            count += 1
+            response = urllib2.urlopen(url)
+            json_string = response.read()
+            try:
+                tmdb_data = json.loads(json_string)[0]
+                return tmdb_data
+            except ValueError, e:
+                if count < 3:
+                    continue
+                else:
+                    raise ApiError("Invalid JSON: %s: %s" % (e, json_string))
+            except:
+                ApiError("JSON error with: %s" % json_string)
+
+
+    def _get_image(self, image_list, min_height, min_width):
+        #Select image
+        images = []
+        for image in image_list:
+            if not min_height or min_width:
+                    images.append(image)
+                    break
+            elif min_height and not min_width:
+                if image['height'] >= min_height:
+                    images.append(image)
+                    break
+            elif min_width and not min_height:
+                if image['width'] >= min_width:
+                    images.append(image)
+                    break
+            elif min_width and min_height:
+                if image['width'] >= min_width and image['height'] >= min_height:
+                    images.append(image)
+                    break
+
+        #No image meets our resolution requirements, so disregard those requirements
+        if len(images) == 0 and min_height or min_width:
+            images.append(image_list[0])
+
+        return images[0]
+
+if __name__ == "__main__":
+    import sys
+    try:
+        id = sys.argv[1]
+    except:
+        id = 'tt0111161'
+
+    x = MetaGen(id)
+    x.write_nfo("movie.nfo")
+    try:
+        x.write_fanart("fanart.jpg", ".", 0, 0)
+    except: pass
+    try:
+        x.write_poster("movie.tbn", ".", 0, 0)
+    except: pass
+"""