예제 #1
0
    def doRename(self):
        '''
        Go find files and rename them!
        '''

        log.debug('Starting renaming')

        if self.isDisabled():
            log.debug('Renaming is disabled')
            return

        allMovies = self.getMovies(self.conf('download'))
        log.debug('Movies: %s' % allMovies)

        if allMovies:
            log.debug("Ready to rename some files.")

        for movie in allMovies:

            if movie.get('match'):
                log.debug('self.renameFiles(movie)')
                finalDestination = self.renameFiles(movie)

                # Search for trailer & subtitles
                log.debug('crons')
                cherrypy.config['cron']['trailer'].forDirectory(finalDestination['directory'])
                cherrypy.config['cron']['subtitle'].forDirectory(finalDestination['directory'])

                # Update Metadata
                if self.config.get('Meta', 'enabled'):
                    log.debug('metadata')
                    nfoFileName = self.config.get('Meta', 'nfoFileName')
                    fanartFileNaming = self.config.get('Meta', 'fanartFileName')
                    fanartMinHeight = self.config.get('Meta', 'fanartMinHeight')
                    fanartMinWidth = self.config.get('Meta', 'fanartMinWidth')
                    posterFileNaming = self.config.get('Meta', 'posterFileName')
                    posterMinHeight = self.config.get('Meta', 'posterMinHeight')
                    posterMinWidth = self.config.get('Meta', 'posterMinWidth')

                    try:
                        x = xmg.MetaGen(movie['movie'].imdb)
                        fanartOrigExt = os.path.splitext(x.get_fanart_url(fanartMinHeight, fanartMinWidth))[-1][1:]
                        posterOrigExt = os.path.splitext(x.get_poster_url(posterMinHeight, posterMinWidth))[-1][1:]

                        nfo_location = os.path.join(finalDestination['directory'],
                                                    self.genMetaFileName(movie, nfoFileName))
                        fanart_filename = self.genMetaFileName(movie,
                                                               fanartFileNaming,
                                                               add_tags = {'orig_ext': fanartOrigExt})
                        poster_filename = self.genMetaFileName(movie,
                                                               posterFileNaming,
                                                               add_tags = {'orig_ext': posterOrigExt})

                        x.write_nfo(nfo_location)

                        x.write_fanart(fanart_filename,
                                       finalDestination['directory'],
                                       fanartMinHeight,
                                       fanartMinWidth)

                        x.write_poster(poster_filename,
                                       finalDestination['directory'],
                                       posterMinHeight,
                                       posterMinWidth)

                        log.info('XBMC metainfo for imdbid, %s, generated' % movie['movie'].imdb)
                    except Exception, e:
                        log.error('XMG TMDB API failure.  Please report to developers. API returned: %s' % e)
                        log.error(traceback.format_exc())

                # Notify XBMC
                log.debug('XBMC')
                xbmc = XBMC()
                xbmc.notify('Downloaded %s (%s)' % (movie['movie'].name, movie['movie'].year))
                xbmc.updateLibrary()

                # Notify NMJ
                log.debug('NMJ')
                nmj = NMJ()
                nmj.updateLibrary()

                # Notify PLEX
                log.debug('PLEX')
                plex = PLEX()
                plex.updateLibrary()

                # Notify PROWL
                log.debug('PROWL')
                prowl = PROWL()
                prowl.notify('Downloaded %s (%s)' % (movie['movie'].name, movie['movie'].year), 'Download Complete')

                # Notify GROWL
                log.debug('GROWL')
                growl = GROWL()
                growl.notify('Downloaded %s (%s)' % (movie['movie'].name, movie['movie'].year), 'Download Complete')

                # Notify Notifo
                log.debug('Notifo')
                notifo = Notifo()
                notifo.notify('%s (%s)' % (movie['movie'].name, movie['movie'].year), "Downloaded:")

                # Notify Boxcar
                log.debug('Boxcar')
                boxcar = Boxcar()
                boxcar.notify('%s (%s)' % (movie['movie'].name, movie['movie'].year), "Downloaded:")

                #Notify NotifyMyAndroid
                log.debug('NotifyMyAndroid')
                nma = NMA()
                nma.notify('Download Complete', 'Downloaded %s (%s)' % (movie['movie'].name, movie['movie'].year))

                # Notify Twitter
                log.debug('Twitter')
                twitter = Twitter()
                twitter.notify('Download Finished', 'Downloaded %s (%s)' % (movie['movie'].name, movie['movie'].year))

                # Notify Synoindex
                log.debug('Synoindex')
                synoindex = Synoindex()
                synoindex.addToLibrary(finalDestination['directory'])

            else:
                path = movie['path'].split(os.sep)
                path.extend(['_UNKNOWN_' + path.pop()])
                target = os.sep.join(path)
                _move(movie['path'], target)

                log.info('No Match found for: %s' % str(movie['info']['name']))

            #Cleanup
            if self.conf('cleanup'):
                log.debug('cleanup')
                path = movie['path']

#                if self.conf('destination') == path:
#                    log.error('Download folder and movie destination shouldn\'t be the same. Change it in Settings >> Renaming.')
#                    return

                for root, subfiles, filenames in os.walk(path):
                    skip = False

                    # Stop if something is in ignored list
                    for ignore in self.ignoredInPath:
                        if ignore in root.lower():
                            skip = True

                    # Ignore full directory names
                    for dir in os.path.split(root):
                        if dir in self.ignoreNames:
                            skip = True

                    # ignore if the current dir is the blackhole
                    if root in self.conf('download'):
                        skip = True

                    if skip: continue

                    for filename in filenames:
                        fullFilePath = os.path.join(root, filename)
                        fileSize = os.path.getsize(fullFilePath)

                        if fileSize < 157286400:
                            try:
                                os.remove(fullFilePath)
                                log.info('Removing file %s.' % fullFilePath)
                            except OSError:
                                log.error('Couldn\'t remove file %s. Too large.' % fullFilePath)

                    try:
                        os.rmdir(root)
                        log.info('Removing dir: %s in download dir.' % root)
                    except OSError:
                        log.error("Tried to clean-up download folder, but '%s' isn't empty." % root)
예제 #2
0
    def doRename(self):
        """
        Go find files and rename them!
        """

        if self.isDisabled():
            return

        allMovies = self.getMovies(self.conf("download"))

        if allMovies:
            log.debug("Ready to rename some files.")

        for movie in allMovies:

            if movie.get("match"):
                finalDestination = self.renameFiles(movie)

                # Search for trailer & subtitles
                cherrypy.config["cron"]["trailer"].forDirectory(finalDestination["directory"])
                cherrypy.config["cron"]["subtitle"].forDirectory(finalDestination["directory"])

                # Notify XBMC
                xbmc = XBMC()
                xbmc.notify("Downloaded %s (%s)" % (movie["movie"].name, movie["movie"].year))
                xbmc.updateLibrary()

                # Update Metadata
                if self.config.get("Meta", "enabled"):
                    nfoFileName = self.config.get("Meta", "nfoFileName")
                    fanartFileNaming = self.config.get("Meta", "fanartFileName")
                    fanartMinHeight = self.config.get("Meta", "fanartMinHeight")
                    fanartMinWidth = self.config.get("Meta", "fanartMinWidth")
                    posterFileNaming = self.config.get("Meta", "posterFileName")
                    posterMinHeight = self.config.get("Meta", "posterMinHeight")
                    posterMinWidth = self.config.get("Meta", "posterMinWidth")

                    x = xmg.MetaGen(movie["movie"].imdb)

                    fanartOrigExt = os.path.splitext(x.get_fanart_url(fanartMinHeight, fanartMinWidth))[-1][1:]
                    posterOrigExt = os.path.splitext(x.get_poster_url(posterMinHeight, posterMinWidth))[-1][1:]

                    nfo_location = os.path.join(finalDestination["directory"], self.genMetaFileName(movie, nfoFileName))
                    fanart_filename = self.genMetaFileName(
                        movie, fanartFileNaming, add_tags={"orig_ext": fanartOrigExt}
                    )
                    poster_filename = self.genMetaFileName(
                        movie, posterFileNaming, add_tags={"orig_ext": posterOrigExt}
                    )

                    x.write_nfo(nfo_location)

                    x.write_fanart(fanart_filename, finalDestination["directory"], fanartMinHeight, fanartMinWidth)

                    x.write_poster(poster_filename, finalDestination["directory"], posterMinHeight, posterMinWidth)

                    log.info("XBMC metainfo for imdbid, %s, generated" % movie["movie"].imdb)

            else:
                try:
                    path = movie["path"].split(os.sep)
                    path.extend(["_UNKNOWN_" + path.pop()])
                    shutil.move(movie["path"], os.sep.join(path))
                except IOError:
                    pass

                log.info("No Match found for: %s" % str(movie["info"]["name"]))

        # Cleanup
        if self.conf("cleanup"):
            path = self.conf("download")

            if self.conf("destination") == path:
                log.error(
                    "Download folder and movie destination shouldn't be the same. Change it in Settings >> Renaming."
                )
                return

            for root, subfiles, filenames in os.walk(path):
                skip = False

                # Stop if something is in ignored list
                for ignore in self.ignoredInPath:
                    if ignore in root.lower():
                        skip = True
                if skip:
                    continue

                for filename in filenames:
                    fullFilePath = os.path.join(root, filename)
                    fileSize = os.path.getsize(fullFilePath)

                    if fileSize < 157286400:
                        try:
                            os.remove(fullFilePath)
                            log.info("Removing file %s." % fullFilePath)
                        except OSError:
                            log.error("Couldn't remove file %s." % fullFilePath)

                if not root in path:
                    try:
                        os.rmdir(root)
                        log.info("Removing dir: %s in download dir." % root)
                    except OSError:
                        log.error("Tried to clean-up download folder, but '%s' isn't empty." % root)
예제 #3
0
파일: yarr.py 프로젝트: Ne0tr0n/CouchPotato
    def _search(self, movie, force = False):

        # Stop caching ffs!
        Db.expire_all()

        # Check release date and search for appropriate qualities
        preReleaseSearch = False
        dvdReleaseSearch = False
        now = int(time.time())

        # Search all if ETA is unknow, but try update ETA for next time.
        log.debug('Calculate ETA')
        checkETA = False
        if not movie.eta:
            checkETA = True
            preReleaseSearch = True
            dvdReleaseSearch = True
        else:
            # Prerelease 1 week before theaters
            if movie.eta.theater <= now + 604800:
                preReleaseSearch = True

            # dvdRelease 6 weeks before dvd release
            if movie.eta.dvd <= now + 3628800:
                preReleaseSearch = True
                dvdReleaseSearch = True

            # Dvd date is unknown but movie is in theater already
            if movie.eta.dvd == 0 and movie.eta.theater > now:
                dvdReleaseSearch = False

            # Force ETA check
            if movie.eta.lastCheck < now:
                checkETA = True

        # Minimal week interval for ETA check
        if checkETA and self.config.get('MovieETA', 'enabled'):
            cherrypy.config.get('searchers').get('etaQueue').put({'id':movie.id})

        for queue in movie.queue:

            # Movie already found, don't search further
            if queue.completed:
                log.debug('%s already completed for "%s". Not searching for any qualities below.' % (queue.qualityType, movie.name))
                return True

            # only search for active and not completed, minimal 1 min since last search
            if queue.active and not queue.completed and not self.abort and not self.stop:

                #skip if no search is set
                log.debug('Needs a search?')
                if (not ((preReleaseSearch and queue.qualityType in Qualities.preReleases) or (dvdReleaseSearch and not queue.qualityType in Qualities.preReleases))) and not queue.lastCheck < (now - int(self.config.get('Intervals', 'search')) * 7200):
                    continue

                log.debug('Start searching for movie: %s' % movie.name)
                highest = self.provider.find(movie, queue)
                log.debug('End searching for movie: %s' % movie.name)

                #send highest to SABnzbd & mark as snatched
                if highest:
                    log.debug('Found highest')

                    #update what I found
                    queue.name = latinToAscii(highest.name)
                    queue.link = highest.detailUrl
                    Db.flush()

                    waitFor = queue.waitFor * (60 * 60 * 24)

                    if queue.markComplete or (not queue.markComplete and highest.date + waitFor < time.time()):
                        time.sleep(10) # Give these APIs air!
                        if self.config.get('NZB', 'sendTo') == 'Sabnzbd' and highest.type == 'nzb':
                            success = self.sabNzbd.send(highest, movie.imdb)
                        elif self.config.get('NZB', 'sendTo') == 'Nzbget' and highest.type == 'nzb':
                            success = self.nzbGet.send(highest)
                        elif self.config.get('Torrents', 'sendTo') == 'Transmission' and highest.type == 'torrent':
                            success = self.transmission.send(highest, movie.imdb)
                        else:
                            success = self.blackHole(highest)

                    else:
                        success = False
                        log.info('Found %s but waiting for %d hours.' % (highest.name, ((highest.date + waitFor) - time.time()) / (60 * 60)))

                    # Set status
                    if success:
                        log.debug('Success')
                        movie.status = u'snatched' if queue.markComplete else u'waiting'
                        movie.dateChanged = datetime.datetime.now()
                        queue.lastCheck = now
                        queue.completed = True
                        Db.flush()

                        # Add to history
                        h = History()
                        h.movie = movie.id
                        h.value = str(highest.id) + '-' + str(highest.size)
                        h.status = u'snatched'
                        Db.add(h)
                        Db.flush()

                        # Notify PROWL
                        if self.config.get('PROWL', 'onSnatch'):
                            log.debug('PROWL')
                            prowl = PROWL()
                            prowl.notify(highest.name, 'Download Started')

                        # Notify XBMC
                        if self.config.get('XBMC', 'onSnatch'):
                            log.debug('XBMC')
                            xbmc = XBMC()
                            xbmc.notify('Snatched %s' % highest.name)

                        # Notify GROWL
                        if self.config.get('GROWL', 'onSnatch'):
                            log.debug('GROWL')
                            growl = GROWL()
                            growl.notify('Snatched %s' % highest.name, 'Download Started')

                        # Notify Notifo
                        if self.config.get('Notifo', 'onSnatch'):
                            log.debug('Notifo')
                            notifo = Notifo()
                            notifo.notify('%s' % highest.name, "Snatched:")

                        # Notify NotifyMyAndroid
                        if self.config.get('NMA', 'onSnatch'):
                            log.debug('NotifyMyAndroid')
                            nma = NMA()
                            nma.notify('Download Started', 'Snatched %s' % highest.name)

                        # Notify Twitter
                        if self.config.get('Twitter', 'onSnatch'):
                            log.debug('Twitter')
                            twitter = Twitter()
                            twitter.notify('Download Started', 'Snatched %s' % highest.name)

                    return True

                queue.lastCheck = now
                Db.flush()

        return False
예제 #4
0
    def _search(self, movie, force=False):

        # Stop caching ffs!
        Db.expire_all()

        # Check release date and search for appropriate qualities
        preReleaseSearch = False
        dvdReleaseSearch = False
        now = int(time.time())

        # Search all if ETA is unknow, but try update ETA for next time.
        log.debug('Calculate ETA')
        checkETA = False
        if not movie.eta:
            checkETA = True
            preReleaseSearch = True
            dvdReleaseSearch = True
        else:
            # Prerelease 1 week before theaters
            if movie.eta.theater <= now + 604800:
                preReleaseSearch = True

            # dvdRelease 6 weeks before dvd release
            if movie.eta.dvd <= now + 3628800:
                preReleaseSearch = True
                dvdReleaseSearch = True

            # Dvd date is unknown but movie is in theater already
            if movie.eta.dvd == 0 and movie.eta.theater > now:
                dvdReleaseSearch = False

            # Force ETA check
            if movie.eta.lastCheck < now:
                checkETA = True

        # Minimal week interval for ETA check
        if checkETA and self.config.get('MovieETA', 'enabled'):
            cherrypy.config.get('searchers').get('etaQueue').put(
                {'id': movie.id})

        for queue in movie.queue:

            # Movie already found, don't search further
            if queue.completed:
                log.debug(
                    '%s already completed for "%s". Not searching for any qualities below.'
                    % (queue.qualityType, movie.name))
                return True

            # only search for active and not completed, minimal 1 min since last search
            if queue.active and not queue.completed and not self.abort and not self.stop:

                #skip if no search is set
                log.debug('Needs a search?')
                if (not ((preReleaseSearch
                          and queue.qualityType in Qualities.preReleases) or
                         (dvdReleaseSearch
                          and not queue.qualityType in Qualities.preReleases))
                    ) and not queue.lastCheck < (now - int(
                        self.config.get('Intervals', 'search')) * 7200):
                    continue

                log.debug('Start searching for movie: %s' % movie.name)
                highest = self.provider.find(movie, queue)
                log.debug('End searching for movie: %s' % movie.name)

                #send highest to SABnzbd & mark as snatched
                if highest:
                    log.debug('Found highest')

                    #update what I found
                    queue.name = latinToAscii(highest.name)
                    queue.link = highest.detailUrl
                    Db.flush()

                    waitFor = queue.waitFor * (60 * 60 * 24)

                    if queue.markComplete or (
                            not queue.markComplete
                            and highest.date + waitFor < time.time()):
                        time.sleep(10)  # Give these APIs air!
                        if self.config.get(
                                'NZB', 'sendTo'
                        ) == 'Sabnzbd' and highest.type == 'nzb':
                            success = self.sabNzbd.send(highest, movie.imdb)
                        elif self.config.get(
                                'NZB', 'sendTo'
                        ) == 'Nzbget' and highest.type == 'nzb':
                            success = self.nzbGet.send(highest)
                        elif self.config.get(
                                'Torrents', 'sendTo'
                        ) == 'Transmission' and highest.type == 'torrent':
                            success = self.transmission.send(
                                highest, movie.imdb)
                        else:
                            success = self.blackHole(highest)

                    else:
                        success = False
                        log.info('Found %s but waiting for %d hours.' %
                                 (highest.name,
                                  ((highest.date + waitFor) - time.time()) /
                                  (60 * 60)))

                    # Set status
                    if success:
                        log.debug('Success')
                        movie.status = u'snatched' if queue.markComplete else u'waiting'
                        movie.dateChanged = datetime.datetime.now()
                        queue.lastCheck = now
                        queue.completed = True
                        Db.flush()

                        # Add to history
                        h = History()
                        h.movie = movie.id
                        h.value = str(highest.id) + '-' + str(highest.size)
                        h.status = u'snatched'
                        Db.add(h)
                        Db.flush()

                        # Notify PROWL
                        if self.config.get('PROWL', 'onSnatch'):
                            log.debug('PROWL')
                            prowl = PROWL()
                            prowl.notify(highest.name, 'Download Started')

                        # Notify XBMC
                        if self.config.get('XBMC', 'onSnatch'):
                            log.debug('XBMC')
                            xbmc = XBMC()
                            xbmc.notify('Snatched %s' % highest.name)

                        # Notify GROWL
                        if self.config.get('GROWL', 'onSnatch'):
                            log.debug('GROWL')
                            growl = GROWL()
                            growl.notify('Snatched %s' % highest.name,
                                         'Download Started')

                        # Notify Notifo
                        if self.config.get('Notifo', 'onSnatch'):
                            log.debug('Notifo')
                            notifo = Notifo()
                            notifo.notify('%s' % highest.name, "Snatched:")

                        # Notify SNS
                        if self.config.get('SNS', 'onSnatch'):
                            log.debug('SNS')
                            sns = SNS()
                            sns.notify('%s' % highest.name, "Snatched:")

                        # Notify Boxcar
                        if self.config.get('Boxcar', 'onSnatch'):
                            log.debug('Boxcar')
                            boxcar = Boxcar()
                            boxcar.notify('%s' % highest.name, "Snatched:")

                        # Notify NotifyMyAndroid
                        if self.config.get('NMA', 'onSnatch'):
                            log.debug('NotifyMyAndroid')
                            nma = NMA()
                            nma.notify('Download Started',
                                       'Snatched %s' % highest.name)

                        # Notify NotifyMyWindowsPhone
                        if self.config.get('NMWP', 'onSnatch'):
                            log.debug('NotifyMyWindowsPhone')
                            nmwp = NMWP()
                            nmwp.notify('Download Started',
                                        'Snatched %s' % highest.name)

                        # Notify Twitter
                        if self.config.get('Twitter', 'onSnatch'):
                            log.debug('Twitter')
                            twitter = Twitter()
                            twitter.notify('Download Started',
                                           'Snatched %s' % highest.name)

                    return True

                queue.lastCheck = now
                Db.flush()

        return False
예제 #5
0
    def doRename(self):
        '''
        Go find files and rename them!
        '''

        log.debug('Starting renaming')

        if self.isDisabled():
            log.debug('Renaming is disabled')
            return

        allMovies = self.getMovies(self.conf('download'))
        log.debug('Movies: %s' % allMovies)

        if allMovies:
            log.debug("Ready to rename some files.")

        for movie in allMovies:

            if movie.get('match'):
                log.debug('self.renameFiles(movie)')
                finalDestination = self.renameFiles(movie)

                # Search for trailer & subtitles
                log.debug('crons')
                cherrypy.config['cron']['trailer'].forDirectory(finalDestination['directory'])
                cherrypy.config['cron']['subtitle'].forDirectory(finalDestination['directory'])

                # Notify XBMC
                log.debug('XBMC')
                xbmc = XBMC()
                xbmc.notify('Downloaded %s (%s)' % (movie['movie'].name, movie['movie'].year))
                xbmc.updateLibrary()

                # Update Metadata
                if self.config.get('Meta', 'enabled'):
                    log.debug('metadata')
                    nfoFileName = self.config.get('Meta', 'nfoFileName')
                    fanartFileNaming = self.config.get('Meta', 'fanartFileName')
                    fanartMinHeight = self.config.get('Meta', 'fanartMinHeight')
                    fanartMinWidth = self.config.get('Meta', 'fanartMinWidth')
                    posterFileNaming = self.config.get('Meta', 'posterFileName')
                    posterMinHeight = self.config.get('Meta', 'posterMinHeight')
                    posterMinWidth = self.config.get('Meta', 'posterMinWidth')
                    
                    try:
                        x = xmg.MetaGen(movie['movie'].imdb)
                        fail = False
                    except xmg.ApiError, e:
                        log.info('XMG TMDB API failure.  Please report to developers. API returned: %s' % e)
                        fail = True
                    if not fail:
                        fanartOrigExt = os.path.splitext(x.get_fanart_url(fanartMinHeight, fanartMinWidth))[-1][1:]
                        posterOrigExt = os.path.splitext(x.get_poster_url(posterMinHeight, posterMinWidth))[-1][1:]
    
                        nfo_location = os.path.join(finalDestination['directory'],
                                                    self.genMetaFileName(movie, nfoFileName))
                        fanart_filename = self.genMetaFileName(movie,
                                                               fanartFileNaming,
                                                               add_tags = {'orig_ext': fanartOrigExt})
                        poster_filename = self.genMetaFileName(movie,
                                                               posterFileNaming,
                                                               add_tags = {'orig_ext': posterOrigExt})
    
                        x.write_nfo(nfo_location)
    
                        x.write_fanart(fanart_filename,
                                       finalDestination['directory'],
                                       fanartMinHeight,
                                       fanartMinWidth)
    
                        x.write_poster(poster_filename,
                                       finalDestination['directory'],
                                       posterMinHeight,
                                       posterMinWidth)
    
                        log.info('XBMC metainfo for imdbid, %s, generated' % movie['movie'].imdb)

            else:
                try:
                    path = movie['path'].split(os.sep)
                    path.extend(['_UNKNOWN_' + path.pop()])
                    shutil.move(movie['path'], os.sep.join(path))
                except IOError:
                    pass

                log.info('No Match found for: %s' % str(movie['info']['name']))
예제 #6
0
    def doRename(self):
        """
        Go find files and rename them!
        """

        log.debug("Starting renaming")

        if self.isDisabled():
            log.debug("Renaming is disabled")
            return

        allMovies = self.getMovies(self.conf("download"))
        log.debug("Movies: %s" % allMovies)

        if allMovies:
            log.debug("Ready to rename some files.")

        for movie in allMovies:
            if movie.get("match"):
                log.debug("self.renameFiles(movie)")
                finalDestination = self.renameFiles(movie)

                # Search for trailer & subtitles
                log.debug("crons")
                cherrypy.config["cron"]["trailer"].forDirectory(finalDestination["directory"])
                cherrypy.config["cron"]["subtitle"].forDirectory(finalDestination["directory"])

                # Update Metadata
                if self.config.get("Meta", "enabled"):
                    log.debug("metadata")
                    nfoFileName = self.config.get("Meta", "nfoFileName")
                    fanartFileNaming = self.config.get("Meta", "fanartFileName")
                    fanartMinHeight = self.config.get("Meta", "fanartMinHeight")
                    fanartMinWidth = self.config.get("Meta", "fanartMinWidth")
                    posterFileNaming = self.config.get("Meta", "posterFileName")
                    posterMinHeight = self.config.get("Meta", "posterMinHeight")
                    posterMinWidth = self.config.get("Meta", "posterMinWidth")

                    try:
                        x = xmg.MetaGen(movie["movie"].imdb)
                        fanartOrigExt = os.path.splitext(x.get_fanart_url(fanartMinHeight, fanartMinWidth))[-1][1:]
                        posterOrigExt = os.path.splitext(x.get_poster_url(posterMinHeight, posterMinWidth))[-1][1:]

                        nfo_location = os.path.join(
                            finalDestination["directory"], self.genMetaFileName(movie, nfoFileName)
                        )
                        fanart_filename = self.genMetaFileName(
                            movie, fanartFileNaming, add_tags={"orig_ext": fanartOrigExt}
                        )
                        poster_filename = self.genMetaFileName(
                            movie, posterFileNaming, add_tags={"orig_ext": posterOrigExt}
                        )

                        urlOnly = self.config.get("Meta", "urlOnly")
                        x.write_nfo(nfo_location, url=True, xml=False)

                        x.write_fanart(fanart_filename, finalDestination["directory"], fanartMinHeight, fanartMinWidth)

                        x.write_poster(poster_filename, finalDestination["directory"], posterMinHeight, posterMinWidth)

                        log.info("XBMC metainfo for imdbid, %s, generated" % movie["movie"].imdb)
                    except Exception, e:
                        log.error("XMG TMDB API failure.  Please report to developers. API returned: %s" % e)
                        log.error(traceback.format_exc())

                # Run post-processing Scripts
                self._run_extra_script(finalDestination)

                # Notify XBMC
                log.debug("XBMC")
                xbmc = XBMC()
                xbmc.notify("Downloaded %s (%s)" % (movie["movie"].name, movie["movie"].year))
                xbmc.updateLibrary()

                # Notify NMJ
                log.debug("NMJ")
                nmj = NMJ()
                nmj.updateLibrary()

                # Notify PLEX
                log.debug("PLEX")
                plex = PLEX()
                plex.updateLibrary()

                # Notify PROWL
                log.debug("PROWL")
                prowl = PROWL()
                prowl.notify("Downloaded %s (%s)" % (movie["movie"].name, movie["movie"].year), "Download Complete")

                # Notify GROWL
                log.debug("GROWL")
                growl = GROWL()
                growl.notify("Downloaded %s (%s)" % (movie["movie"].name, movie["movie"].year), "Download Complete")

                # Notify Notifo
                log.debug("Notifo")
                notifo = Notifo()
                notifo.notify("%s (%s)" % (movie["movie"].name, movie["movie"].year), "Downloaded:")

                # Notify Boxcar
                log.debug("Boxcar")
                boxcar = Boxcar()
                boxcar.notify("%s (%s)" % (movie["movie"].name, movie["movie"].year), "Downloaded:")

                # Notify NotifyMyAndroid
                log.debug("NotifyMyAndroid")
                nma = NMA()
                nma.notify("Download Complete", "Downloaded %s (%s)" % (movie["movie"].name, movie["movie"].year))

                # Notify Twitter
                log.debug("Twitter")
                twitter = Twitter()
                twitter.notify("Download Finished", "Downloaded %s (%s)" % (movie["movie"].name, movie["movie"].year))

                # Notify Synoindex
                log.debug("Synoindex")
                synoindex = Synoindex()
                synoindex.addToLibrary(finalDestination["directory"])

                # Notify Trakt
                log.debug("Trakt")
                trakt = Trakt()
                trakt.notify(movie["movie"].name, movie["movie"].year, movie["movie"].imdb)

            else:
                path = movie["path"].split(os.sep)
                path.extend(["_UNKNOWN_" + path.pop()])
                target = os.sep.join(path)
                _move(movie["path"], target)

                log.info("No Match found for: %s" % str(movie["info"]["name"]))

            # Cleanup
            if self.conf("cleanup"):
                log.debug("cleanup")
                path = movie["path"]

                #                if self.conf('destination') == path:
                #                    log.error('Download folder and movie destination shouldn\'t be the same. Change it in Settings >> Renaming.')
                #                    return

                for root, subfiles, filenames in os.walk(path):
                    skip = False

                    # Stop if something is in ignored list
                    for ignore in self.ignoredInPath:
                        if ignore in root.lower():
                            skip = True

                    # Ignore full directory names
                    for dir in os.path.split(root):
                        if dir in self.ignoreNames:
                            skip = True

                    # ignore if the current dir is the blackhole
                    if root in self.conf("download"):
                        skip = True

                    if skip:
                        continue

                    for filename in filenames:
                        fullFilePath = os.path.join(root, filename)
                        fileSize = os.path.getsize(fullFilePath)

                        if fileSize < 157286400:
                            try:
                                os.remove(fullFilePath)
                                log.info("Removing file %s." % fullFilePath)
                            except OSError:
                                log.error("Couldn't remove file %s. Too large." % fullFilePath)

                    try:
                        os.rmdir(root)
                        log.info("Removing dir: %s in download dir." % root)
                    except OSError:
                        log.error("Tried to clean-up download folder, but '%s' isn't empty." % root)
예제 #7
0
    def doRename(self):
        '''
        Go find files and rename them!
        '''

        log.debug('Starting renaming')

        if self.isDisabled():
            log.debug('Renaming is disabled')
            return

        allMovies = self.getMovies(self.conf('download'))
        log.debug('Movies: %s' % allMovies)

        if allMovies:
            log.debug("Ready to rename some files.")

        for movie in allMovies:

            if movie.get('match'):
                log.debug('self.renameFiles(movie)')
                finalDestination = self.renameFiles(movie)

                # Search for trailer & subtitles
                log.debug('crons')
                cherrypy.config['cron']['trailer'].forDirectory(finalDestination['directory'])
                cherrypy.config['cron']['subtitle'].forDirectory(finalDestination['directory'])

                # Update Metadata
                if self.config.get('Meta', 'enabled'):
                    log.debug('metadata')
                    nfoFileName = self.config.get('Meta', 'nfoFileName')
                    fanartFileNaming = self.config.get('Meta', 'fanartFileName')
                    fanartMinHeight = self.config.get('Meta', 'fanartMinHeight')
                    fanartMinWidth = self.config.get('Meta', 'fanartMinWidth')
                    posterFileNaming = self.config.get('Meta', 'posterFileName')
                    posterMinHeight = self.config.get('Meta', 'posterMinHeight')
                    posterMinWidth = self.config.get('Meta', 'posterMinWidth')

                    try:
                        x = xmg.MetaGen(movie['movie'].imdb)
                        fanartOrigExt = os.path.splitext(x.get_fanart_url(fanartMinHeight, fanartMinWidth))[-1][1:]
                        posterOrigExt = os.path.splitext(x.get_poster_url(posterMinHeight, posterMinWidth))[-1][1:]

                        nfo_location = os.path.join(finalDestination['directory'],
                                                    self.genMetaFileName(movie, nfoFileName))
                        fanart_filename = self.genMetaFileName(movie,
                                                               fanartFileNaming,
                                                               add_tags = {'orig_ext': fanartOrigExt})
                        poster_filename = self.genMetaFileName(movie,
                                                               posterFileNaming,
                                                               add_tags = {'orig_ext': posterOrigExt})

                        x.write_nfo(nfo_location)

                        x.write_fanart(fanart_filename,
                                       finalDestination['directory'],
                                       fanartMinHeight,
                                       fanartMinWidth)

                        x.write_poster(poster_filename,
                                       finalDestination['directory'],
                                       posterMinHeight,
                                       posterMinWidth)

                        log.info('XBMC metainfo for imdbid, %s, generated' % movie['movie'].imdb)
                    except Exception, e:
                        log.error('XMG TMDB API failure.  Please report to developers. API returned: %s' % e)
                        log.error(traceback.format_exc())

                # Notify XBMC
                log.debug('XBMC')
                xbmc = XBMC()
                xbmc.notify('Downloaded %s (%s)' % (movie['movie'].name, movie['movie'].year))
                xbmc.updateLibrary()

                # Notify NMJ
                log.debug('NMJ')
                nmj = NMJ()
                nmj.updateLibrary()

                # Notify PLEX
                log.debug('PLEX')
                plex = PLEX()
                plex.updateLibrary()

                # Notify PROWL
                log.debug('PROWL')
                prowl = PROWL()
                prowl.notify('Downloaded %s (%s)' % (movie['movie'].name, movie['movie'].year), 'Download Complete')

                # Notify GROWL
                log.debug('GROWL')
                growl = GROWL()
                growl.notify('Downloaded %s (%s)' % (movie['movie'].name, movie['movie'].year), 'Download Complete')

                # Notify Notifo
                log.debug('Notifo')
                notifo = Notifo()
                notifo.notify('%s (%s)' % (movie['movie'].name, movie['movie'].year), "Downloaded:")

                #Notify NotifyMyAndroid
                log.debug('NotifyMyAndroid')
                nma = NMA()
                nma.notify('Download Complete', 'Downloaded %s (%s)' % (movie['movie'].name, movie['movie'].year))

                # Notify Twitter
                log.debug('Twitter')
                twitter = Twitter()
                twitter.notify('Download Finished', 'Downloaded %s (%s)' % (movie['movie'].name, movie['movie'].year))

                # Notify Synoindex
                log.debug('Synoindex')
                synoindex = Synoindex()
                synoindex.addToLibrary(finalDestination['directory'])

            else:
                path = movie['path'].split(os.sep)
                path.extend(['_UNKNOWN_' + path.pop()])
                target = os.sep.join(path)
                _move(movie['path'], target)

                log.info('No Match found for: %s' % str(movie['info']['name']))

            #Cleanup
            if self.conf('cleanup'):
                log.debug('cleanup')
                path = movie['path']

#                if self.conf('destination') == path:
#                    log.error('Download folder and movie destination shouldn\'t be the same. Change it in Settings >> Renaming.')
#                    return

                for root, subfiles, filenames in os.walk(path):
                    skip = False

                    # Stop if something is in ignored list
                    for ignore in self.ignoredInPath:
                        if ignore in root.lower():
                            skip = True

                    # Ignore full directory names
                    for dir in os.path.split(root):
                        if dir in self.ignoreNames:
                            skip = True

                    # ignore if the current dir is the blackhole
                    if root in self.conf('download'):
                        skip = True

                    if skip: continue

                    for filename in filenames:
                        fullFilePath = os.path.join(root, filename)
                        fileSize = os.path.getsize(fullFilePath)

                        if fileSize < 157286400:
                            try:
                                os.remove(fullFilePath)
                                log.info('Removing file %s.' % fullFilePath)
                            except OSError:
                                log.error('Couldn\'t remove file %s. Too large.' % fullFilePath)

                    try:
                        os.rmdir(root)
                        log.info('Removing dir: %s in download dir.' % root)
                    except OSError:
                        log.error("Tried to clean-up download folder, but '%s' isn't empty." % root)
예제 #8
0
    def doRename(self):
        """
        Go find files and rename them!
        """

        log.debug("Starting renaming")

        if self.isDisabled():
            log.debug("Renaming is disabled")
            return

        allMovies = self.getMovies(self.conf("download"))
        log.debug("Movies: %s" % allMovies)

        if allMovies:
            log.debug("Ready to rename some files.")

        for movie in allMovies:

            if movie.get("match"):
                log.debug("self.renameFiles(movie)")
                finalDestination = self.renameFiles(movie)

                # Search for trailer & subtitles
                log.debug("crons")
                cherrypy.config["cron"]["trailer"].forDirectory(finalDestination["directory"])
                cherrypy.config["cron"]["subtitle"].forDirectory(finalDestination["directory"])

                # Update Metadata
                if self.config.get("Meta", "enabled"):
                    log.debug("metadata")
                    nfoFileName = self.config.get("Meta", "nfoFileName")
                    fanartFileNaming = self.config.get("Meta", "fanartFileName")
                    fanartMinHeight = self.config.get("Meta", "fanartMinHeight")
                    fanartMinWidth = self.config.get("Meta", "fanartMinWidth")
                    posterFileNaming = self.config.get("Meta", "posterFileName")
                    posterMinHeight = self.config.get("Meta", "posterMinHeight")
                    posterMinWidth = self.config.get("Meta", "posterMinWidth")

                    try:
                        x = xmg.MetaGen(movie["movie"].imdb)
                        fanartOrigExt = os.path.splitext(x.get_fanart_url(fanartMinHeight, fanartMinWidth))[-1][1:]
                        posterOrigExt = os.path.splitext(x.get_poster_url(posterMinHeight, posterMinWidth))[-1][1:]

                        nfo_location = os.path.join(
                            finalDestination["directory"], self.genMetaFileName(movie, nfoFileName)
                        )
                        fanart_filename = self.genMetaFileName(
                            movie, fanartFileNaming, add_tags={"orig_ext": fanartOrigExt}
                        )
                        poster_filename = self.genMetaFileName(
                            movie, posterFileNaming, add_tags={"orig_ext": posterOrigExt}
                        )

                        x.write_nfo(nfo_location)

                        x.write_fanart(fanart_filename, finalDestination["directory"], fanartMinHeight, fanartMinWidth)

                        x.write_poster(poster_filename, finalDestination["directory"], posterMinHeight, posterMinWidth)

                        log.info("XBMC metainfo for imdbid, %s, generated" % movie["movie"].imdb)
                    except xmg.ApiError, e:
                        log.error("XMG TMDB API failure.  Please report to developers. API returned: %s" % e)
                        log.error(traceback.format_exc())

                # Notify XBMC
                log.debug("XBMC")
                xbmc = XBMC()
                xbmc.notify("Downloaded %s (%s)" % (movie["movie"].name, movie["movie"].year))
                xbmc.updateLibrary()

            else:
                path = movie["path"].split(os.sep)
                path.extend(["_UNKNOWN_" + path.pop()])
                target = os.sep.join(path)
                _move(movie["path"], target)

                log.info("No Match found for: %s" % str(movie["info"]["name"]))