/sickbeard/metadata/tivo.py

# Author: Nic Wolfe <nic@wolfeden.ca>
# Author: Gordon Turner <gordonturner@gordonturner.ca>
# URL: http://code.google.com/p/sickbeard/
#
# This file is part of Sick Beard.
#
# Sick Beard is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Sick Beard is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Sick Beard.  If not, see <http://www.gnu.org/licenses/>.

import datetime
import os

import sickbeard

#from sickbeard.common import *
from sickbeard import logger, exceptions, helpers
from sickbeard.metadata import generic
from sickbeard import encodingKludge as ek
from sickbeard import config

from lib.tvdb_api import tvdb_api, tvdb_exceptions

class TIVOMetadata(generic.GenericMetadata):
    """
    Metadata generation class for TIVO

    The following file structure is used:

    show_root/Season 01/show - 1x01 - episode.avi.txt       (* existing episode)
    show_root/Season 01/.meta/show - 1x01 - episode.avi.txt (episode metadata)
    
    This class only generates episode specific metadata files, it does NOT generated a default.txt file.
    """
    
    def __init__(self,
                 show_metadata=False,
                 episode_metadata=False,
                 poster=False,
                 fanart=False,
                 episode_thumbnails=False,
                 season_thumbnails=False):

        generic.GenericMetadata.__init__(self,
                                         show_metadata,
                                         episode_metadata,
                                         poster,
                                         fanart,
                                         episode_thumbnails,
                                         season_thumbnails)
        
        self._ep_nfo_extension = "txt"
        
        self.generate_ep_metadata = True
        
        self.name = 'TIVO'

        self.eg_show_metadata = "<i>not supported</i>"
        self.eg_episode_metadata = "Season##\\.meta\\<i>filename</i>.txt"
        self.eg_fanart = "<i>not supported</i>"
        self.eg_poster = "<i>not supported</i>"
        self.eg_episode_thumbnails = "<i>not supported</i>"
        self.eg_season_thumbnails = "<i>not supported</i>"
    
    # Override with empty methods for unsupported features.
    def create_show_metadata(self, show_obj):
        pass
    
    def create_fanart(self, show_obj):
        pass
    
    def get_episode_thumb_path(self, ep_obj):
        pass
    
    def get_season_thumb_path(self, show_obj, season):
        pass

    def retrieveShowMetadata(self, dir):
        return (None, None)
        
    # Override and implement features for Tivo.
    def get_episode_file_path(self, ep_obj):
        """
        Returns a full show dir/.meta/episode.txt path for Tivo
        episode metadata files.
        
        Note, that pyTivo requires the metadata filename to include the original extention.
        
        ie If the episode name is foo.avi, the metadata name is foo.avi.txt
        
        ep_obj: a TVEpisode object to get the path for
        """
        if ek.ek(os.path.isfile, ep_obj.location):
            metadata_file_name = ek.ek(os.path.basename, ep_obj.location) + "." + self._ep_nfo_extension
            metadata_dir_name = ek.ek(os.path.join, ek.ek(os.path.dirname, ep_obj.location), '.meta')
            metadata_file_path = ek.ek(os.path.join, metadata_dir_name, metadata_file_name)
        else:
            logger.log(u"Episode location doesn't exist: "+str(ep_obj.location), logger.DEBUG)
            return ''
        return metadata_file_path

    def _ep_data(self, ep_obj):
        """
        Creates a key value structure for a Tivo episode metadata file and
        returns the resulting data object.
        
        ep_obj: a TVEpisode instance to create the metadata file for.
        
        Lookup the show in http://thetvdb.com/ using the python library:
        
        https://github.com/dbr/tvdb_api/
        
        The results are saved in the object myShow.
            
        The key values for the tivo metadata file are from:
            
        http://pytivo.sourceforge.net/wiki/index.php/Metadata
        """
        
        data = "";
                
        eps_to_write = [ep_obj] + ep_obj.relatedEps
        
        tvdb_lang = ep_obj.show.lang
    
        try:
            # There's gotta be a better way of doing this but we don't wanna
            # change the language value elsewhere
            ltvdb_api_parms = sickbeard.TVDB_API_PARMS.copy()

            if tvdb_lang and not tvdb_lang == 'en':
                ltvdb_api_parms['language'] = tvdb_lang

            t = tvdb_api.Tvdb(actors=True, **ltvdb_api_parms)
            myShow = t[ep_obj.show.tvdbid]
        except tvdb_exceptions.tvdb_shownotfound, e:
            raise exceptions.ShowNotFoundException(str(e))
        except tvdb_exceptions.tvdb_error, e:
            logger.log("Unable to connect to TVDB while creating meta files - skipping - "+str(e), logger.ERROR)
            return False
        
        for curEpToWrite in eps_to_write:
        
            try:
                myEp = myShow[curEpToWrite.season][curEpToWrite.episode]
            except (tvdb_exceptions.tvdb_episodenotfound, tvdb_exceptions.tvdb_seasonnotfound):
                logger.log("Unable to find episode " + str(curEpToWrite.season) + "x" + str(curEpToWrite.episode) + " on tvdb... has it been removed? Should I delete from db?")
                return None
            
            if myEp["firstaired"] == None and ep_obj.season == 0:
                myEp["firstaired"] = str(datetime.date.fromordinal(1))
            
            if myEp["episodename"] == None or myEp["firstaired"] == None:
                return None
                
            if myShow["seriesname"] != None: 
            
                # Title of the series (The Simpsons, Seinfeld, etc.) or title of the movie (The Mummy, Spiderman, etc).
                data += ("title : " + myShow["seriesname"] + "\n")
                
                
                # Name of series (The Simpsons, Seinfeld, etc.). This should be included if the show is episodic. 
                # For movies, you may repeat the name of the movie (The Mummy, Spiderman, etc), leave blank, or omit. 
                data += ("seriesTitle : " + myShow["seriesname"] + "\n")
                
            
            # Title of the episode (Pilot, Homer's Night Out, Episode 02, etc.) Should be included for episodic shows. 
            # Leave blank or omit for movies.
            #
            # Added season episode to title, so that the shows will sort correctly, as often the date information is wrong.
            data += ("episodeTitle : " + config.naming_ep_type[sickbeard.NAMING_EP_TYPE] % {'seasonnumber': curEpToWrite.season, 'episodenumber': curEpToWrite.episode}  + " " + curEpToWrite.name + "\n")
            
            
            # This should be entered for episodic shows and omitted for movies. The standard tivo format is to enter 
            # the season number followed by the episode number for that season. For example, enter 201 for season 2 
            # episode 01. 

            # This only shows up if you go into the Details from the Program screen. 

            # This seems to disappear once the video is transferred to TiVo.

            # NOTE: May not be correct format, missing season, but based on description from wiki leaving as is.

            data += ("episodeNumber : " + str(curEpToWrite.episode) + "\n")
            
            
            # Must be entered as true or false. If true, the year from originalAirDate will be shown in parentheses 
            # after the episode's title and before the description on the Program screen.
 
            # FIXME: Hardcode isEpisode to true for now, not sure how to handle movies

            data += ("isEpisode : true\n")
            
            
            # Write the synopsis of the video here. 
            
            # Micrsoft Word's smartquotes can die in a fire.
            sanitizedDescription = curEpToWrite.description
            # Replace double curly quotes
            sanitizedDescription = sanitizedDescription.replace(u"\u201c", "\"").replace(u"\u201d", "\"")
            # Replace single curly quotes
            sanitizedDescription = sanitizedDescription.replace(u"\u2018", "'").replace(u"\u2019", "'").replace(u"\u02BC", "'")

            data += ("description : " + sanitizedDescription + "\n")
            
            
            # Usually starts with "SH" and followed by 6-8 digits.
            # Tivo uses zap2it for thier data, so the series id is the zap2it_id.
            if myShow["zap2it_id"] != None:
                data += ("seriesId : " + myShow["zap2it_id"] + "\n")
                

            # This is the call sign of the channel the episode was recorded from.
            if myShow["network"] != None:
                data += ("callsign : " + myShow["network"] + "\n")
            
                    
            # This must be entered as yyyy-mm-ddThh:mm:ssZ (the t is capitalized and never changes, the Z is also 
            # capitalized and never changes). This is the original air date of the episode. 
            # NOTE: Hard coded the time to T00:00:00Z as we really don't know when during the day the first run happened.
            if curEpToWrite.airdate != datetime.date.fromordinal(1):
                data += ("originalAirDate : " + str(curEpToWrite.airdate) + "T00:00:00Z\n")
               
                    
            # This shows up at the beginning of the description on the Program screen and on the Details screen.
            if myShow["actors"]:
                for actor in myShow["actors"].split('|'):
                    if actor:
                        data += ("vActor : " + actor + "\n")
                    
               
            # This is shown on both the Program screen and the Details screen. It uses a single digit to determine the 
            # number of stars: 1 for 1 star, 7 for 4 stars
            if myShow["rating"] != None:
                try:
                    rating = float(myShow['rating'])
                except ValueError:
                    rating = 0.0
                rating = rating / 10 * 4
                data += ("starRating : " + str(rating) + "\n")

                    
            # This is shown on both the Program screen and the Details screen.
            # It uses the standard TV rating system of: TV-Y7, TV-Y, TV-G, TV-PG, TV-14, TV-MA and TV-NR.
            if myShow["contentrating"]:
                data += ("tvRating : " + str(myShow["contentrating"]) + "\n")
            
                    
            # This field can be repeated as many times as necessary or omitted completely.
            if ep_obj.show.genre:
                for genre in ep_obj.show.genre.split('|'):
                    if genre:
                        data += ("vProgramGenre : " + str(genre) + "\n")

                        
            # NOTE: The following are metadata keywords are not used
            # displayMajorNumber
            # showingBits
            # displayMinorNumber
            # colorCode
            # vSeriesGenre
            # vGuestStar, vDirector, vExecProducer, vProducer, vWriter, vHost, vChoreographer
            # partCount
            # partIndex
            
        
        return data

    def write_ep_file(self, ep_obj):
        """
        Generates and writes ep_obj's metadata under the given path with the
        given filename root. Uses the episode's name with the extension in
        _ep_nfo_extension.
        
        ep_obj: TVEpisode object for which to create the metadata
        
        file_name_path: The file name to use for this metadata. Note that the extension
                will be automatically added based on _ep_nfo_extension. This should
                include an absolute path.
        """       
        data = self._ep_data(ep_obj)
        
        if not data:
            return False
        
        nfo_file_path = self.get_episode_file_path(ep_obj)
        nfo_file_dir = ek.ek(os.path.dirname, nfo_file_path)
        
        try:
            if not ek.ek(os.path.isdir, nfo_file_dir):
                logger.log("Metadata dir didn't exist, creating it at "+nfo_file_dir, logger.DEBUG)
                ek.ek(os.makedirs, nfo_file_dir)
                helpers.chmodAsParent(nfo_file_dir)
            
            logger.log(u"Writing episode nfo file to "+nfo_file_path)
            nfo_file = ek.ek(open, nfo_file_path, 'w')
            
            # Calling encode directly, b/c often descriptions have wonky characters.
            nfo_file.write( data.encode( "utf-8" ) )
            
            nfo_file.close()
            helpers.chmodAsParent(nfo_file_path)
        except IOError, e:
            logger.log(u"Unable to write file to "+nfo_file_path+" - are you sure the folder is writable? "+str(e).decode('utf-8'), logger.ERROR)
            return False
        
        return True

# present a standard "interface"
metadata_class = TIVOMetadata