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)
def testNMA(self, **data): nma = NMA() nma.test(data.get('NMA.apikey'), data.get('NMA.devkey'), data.get('NMA.priority')) return ''
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
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
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)
def testNMA(self, **data): nma = NMA() nma.test(data.get("NMA.apikey"), data.get("NMA.devkey"), data.get("NMA.priority")) return ""
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)