def addTorrent(link): method = 'torrent-add' arguments = { 'filename': link, 'download-dir': headphones.DOWNLOAD_TORRENT_DIR } response = torrentAction(method, arguments) if not response: return False if response['result'] == 'success': try: name = response['arguments']['torrent-added']['name'] logger.info(u"Torrent sent to Transmission successfully") if headphones.PROWL_ENABLED and headphones.PROWL_ONSNATCH: logger.info(u"Sending Prowl notification") prowl = notifiers.PROWL() prowl.notify(name, "Download started") if headphones.PUSHOVER_ENABLED and headphones.PUSHOVER_ONSNATCH: logger.info(u"Sending Pushover notification") prowl = notifiers.PUSHOVER() prowl.notify(name, "Download started") if headphones.NMA_ENABLED and headphones.NMA_ONSNATCH: logger.debug(u"Sending NMA notification") nma = notifiers.NMA() nma.notify(snatched_nzb=name) return response['arguments']['torrent-added']['name'], response[ 'arguments']['torrent-added']['hashString'] except KeyError: logger.warn(u"Torrent was not sent to Transmission") return False
def doPostProcessing(albumid, albumpath, release, tracks, downloaded_track_list): logger.info('Starting post-processing for: %s - %s' % (release['ArtistName'], release['AlbumTitle'])) #start encoding if headphones.MUSIC_ENCODER: downloaded_track_list = music_encoder.encode(albumpath) if not downloaded_track_list: return album_art_path = albumart.getAlbumArt(albumid) if headphones.EMBED_ALBUM_ART or headphones.ADD_ALBUM_ART: if album_art_path: artwork = urllib.urlopen(album_art_path).read() if not album_art_path or len(artwork) < 100: logger.info( "No suitable album art found from Amazon. Checking Last.FM...." ) artwork = albumart.getCachedArt(albumid) if not artwork or len(artwork) < 100: artwork = False logger.info( "No suitable album art found from Last.FM. Not adding album art" ) if headphones.EMBED_ALBUM_ART and artwork: embedAlbumArt(artwork, downloaded_track_list) if headphones.CLEANUP_FILES: cleanupFiles(albumpath) if headphones.ADD_ALBUM_ART and artwork: addAlbumArt(artwork, albumpath) if headphones.CORRECT_METADATA: correctMetadata(albumid, release, downloaded_track_list) if headphones.EMBED_LYRICS: embedLyrics(downloaded_track_list) if headphones.RENAME_FILES: renameFiles(albumpath, downloaded_track_list, release) if headphones.MOVE_FILES and not headphones.DESTINATION_DIR: logger.error( 'No DESTINATION_DIR has been set. Set "Destination Directory" to the parent directory you want to move the files to' ) albumpaths = [albumpath] elif headphones.MOVE_FILES and headphones.DESTINATION_DIR: albumpaths = moveFiles(albumpath, release, tracks) else: albumpaths = [albumpath] myDB = db.DBConnection() myDB.action('UPDATE albums SET status = "Downloaded" WHERE AlbumID=?', [albumid]) myDB.action('UPDATE snatched SET status = "Processed" WHERE AlbumID=?', [albumid]) # Update the have tracks for all created dirs: for albumpath in albumpaths: librarysync.libraryScan(dir=albumpath, append=True, ArtistID=release['ArtistID'], ArtistName=release['ArtistName']) logger.info(u'Post-processing for %s - %s complete' % (release['ArtistName'], release['AlbumTitle'])) if headphones.PROWL_ENABLED: pushmessage = release['ArtistName'] + ' - ' + release['AlbumTitle'] logger.info(u"Prowl request") prowl = notifiers.PROWL() prowl.notify(pushmessage, "Download and Postprocessing completed") if headphones.XBMC_ENABLED: xbmc = notifiers.XBMC() if headphones.XBMC_UPDATE: xbmc.update() if headphones.XBMC_NOTIFY: xbmc.notify(release['ArtistName'], release['AlbumTitle'], album_art_path) if headphones.NMA_ENABLED: nma = notifiers.NMA() nma.notify(release['ArtistName'], release['AlbumTitle']) if headphones.SYNOINDEX_ENABLED: syno = notifiers.Synoindex() for albumpath in albumpaths: syno.notify(albumpath)
if headphones.PROWL_ENABLED: pushmessage = release['ArtistName'] + ' - ' + release['AlbumTitle'] logger.info(u"Prowl request") prowl = notifiers.PROWL() prowl.notify(pushmessage, "Download and Postprocessing completed") if headphones.XBMC_ENABLED: xbmc = notifiers.XBMC() if headphones.XBMC_UPDATE: xbmc.update() if headphones.XBMC_NOTIFY: xbmc.notify(release['ArtistName'], release['AlbumTitle'], album_art_path) if headphones.NMA_ENABLED: nma = notifiers.NMA() nma.notify(release['ArtistName'], release['AlbumTitle']) if headphones.SYNOINDEX_ENABLED: syno = notifiers.Synoindex() for albumpath in albumpaths: syno.notify(albumpath) if headphones.PUSHOVER_ENABLED: pushmessage = release['ArtistName'] + ' - ' + release['AlbumTitle'] logger.info(u"Pushover request") pushover = notifiers.PUSHOVER() pushover.notify(pushmessage, "Download and Postprocessing completed") def embedAlbumArt(artwork, downloaded_track_list):
def doPostProcessing(albumid, albumpath, release, tracks, downloaded_track_list): logger.info('Starting post-processing for: %s - %s' % (release['ArtistName'], release['AlbumTitle'])) #start enconding if headphones.MUSIC_ENCODER: downloaded_track_list=music_encoder.encode(albumpath) album_art_path = albumart.getAlbumArt(albumid) if headphones.EMBED_ALBUM_ART or headphones.ADD_ALBUM_ART: artwork = urllib.urlopen(album_art_path).read() if len(artwork) < 100: artwork = False logger.info("No suitable album art found. Not adding album art") if headphones.EMBED_ALBUM_ART and artwork: embedAlbumArt(artwork, downloaded_track_list) if headphones.CLEANUP_FILES: cleanupFiles(albumpath) if headphones.ADD_ALBUM_ART and artwork: addAlbumArt(artwork, albumpath) if headphones.CORRECT_METADATA: correctMetadata(albumid, release, downloaded_track_list) if headphones.EMBED_LYRICS: embedLyrics(downloaded_track_list) if headphones.RENAME_FILES: renameFiles(albumpath, downloaded_track_list, release) if headphones.MOVE_FILES and headphones.DESTINATION_DIR: albumpath = moveFiles(albumpath, release, tracks) if headphones.MOVE_FILES and not headphones.DESTINATION_DIR: logger.error('No DESTINATION_DIR has been set. Set "Destination Directory" to the parent directory you want to move the files to') pass myDB = db.DBConnection() # There's gotta be a better way to update the have tracks - sqlite trackcount = myDB.select('SELECT HaveTracks from artists WHERE ArtistID=?', [release['ArtistID']]) if not trackcount[0][0]: cur_track_count = 0 else: cur_track_count = trackcount[0][0] new_track_count = cur_track_count + len(downloaded_track_list) myDB.action('UPDATE artists SET HaveTracks=? WHERE ArtistID=?', [new_track_count, release['ArtistID']]) myDB.action('UPDATE albums SET status = "Downloaded" WHERE AlbumID=?', [albumid]) myDB.action('UPDATE snatched SET status = "Processed" WHERE AlbumID=?', [albumid]) updateHave(albumpath) logger.info('Post-processing for %s - %s complete' % (release['ArtistName'], release['AlbumTitle'])) if headphones.PROWL_ONSNATCH: pushmessage = release['ArtistName'] + ' - ' + release['AlbumTitle'] logger.info(u"Prowl request") prowl = notifiers.PROWL() prowl.notify(pushmessage,"Download and Postprocessing completed") if headphones.XBMC_ENABLED: xbmc = notifiers.XBMC() if headphones.XBMC_UPDATE: xbmc.update() if headphones.XBMC_NOTIFY: xbmc.notify(release['ArtistName'], release['AlbumTitle'], album_art_path) if headphones.NMA_ENABLED: nma = notifiers.NMA() nma.notify(release['ArtistName'], release['AlbumTitle'])