github/nzbToMedia
1
0
Fork 0
mirror of https://github.com/clinton-hall/nzbToMedia.git synced 2025-08-21 22:03:13 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

more manual run fixes.

Browse source
This commit is contained in:
clinton-hall 2014-07-19 21:55:19 +09:30
commit 0268718074
1 changed files with 39 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

79
nzbtomedia/nzbToMediaUtil.py
View file

@ -480,56 +480,55 @@ def getDirs(section, subsection):
logger.info("Searching %s for mediafiles to post-process ..." % (path)) logger.info("Searching %s for mediafiles to post-process ..." % (path))
# search for single files and move them into there own folder for post-processing # search for single files and move them into their own folder for post-processing
for mediafile in listMediaFiles(path): for mediafile in [ os.path.join(path, o) for o in os.listdir(path) if
parentDir = os.path.dirname(mediafile) os.path.isfile(os.path.join(path, o)) ]:
if parentDir == path: logger.debug("Found file %s in root directory %s." % (mediafile, path))
logger.debug("Found file %s in root directory %s." % (mediafile, path)) newPath = None
newPath = None fileExt = os.path.splitext(os.path.basename(mediafile))[1]
fileExt = os.path.splitext(os.path.basename(mediafile))[1]
try: try:
if fileExt in nzbtomedia.AUDIOCONTAINER: if fileExt in nzbtomedia.AUDIOCONTAINER:
f = beets.mediafile.MediaFile(mediafile) f = beets.mediafile.MediaFile(mediafile)
# get artist and album info # get artist and album info
artist = f.artist artist = f.artist
album = f.album album = f.album
# create new path # create new path
newPath = os.path.join(parentDir, "%s - %s" % (sanitizeName(artist), sanitizeName(album))) newPath = os.path.join(path, "%s - %s" % (sanitizeName(artist), sanitizeName(album)))
elif fileExt in nzbtomedia.MEDIACONTAINER: elif fileExt in nzbtomedia.MEDIACONTAINER:
f = guessit.guess_video_info(mediafile) f = guessit.guess_video_info(mediafile)
# get title # get title
title = None title = None
try: try:
title = f['series'] title = f['series']
except: except:
title = f['title'] title = f['title']
if not title: if not title:
title = os.path.splitext(os.path.basename(mediafile))[0] title = os.path.splitext(os.path.basename(mediafile))[0]
newPath = os.path.join(parentDir, sanitizeName(title)) newPath = os.path.join(path, sanitizeName(title))
except Exception, e: except Exception, e:
logger.error("Exception parsing name for media file: %s: %s" % (mediafile, e)) logger.error("Exception parsing name for media file: %s: %s" % (mediafile, e))
if not newPath: if not newPath:
title = os.path.splitext(os.path.basename(mediafile))[0] title = os.path.splitext(os.path.basename(mediafile))[0]
newPath = os.path.join(parentDir, sanitizeName(title)) newPath = os.path.join(path, sanitizeName(title))
# Just fail-safe incase we already have afile with this clean-name (was actually a bug from earlier code, but let's be safe). # Just fail-safe incase we already have afile with this clean-name (was actually a bug from earlier code, but let's be safe).
if os.path.isfile(newPath): if os.path.isfile(newPath):
newPath2 = os.path.join(os.path.join(os.path.split(newPath)[0], 'new'), os.path.split(newPath)[1]) newPath2 = os.path.join(os.path.join(os.path.split(newPath)[0], 'new'), os.path.split(newPath)[1])
newPath = newPath2 newPath = newPath2
# create new path if it does not exist # create new path if it does not exist
if not os.path.exists(newPath): if not os.path.exists(newPath):
makeDir(newPath) makeDir(newPath)
# move file to its new path # move file to its new path
copy_link(mediafile, os.path.join(newPath, os.path.split(mediafile)[1]), 'hard') copy_link(mediafile, os.path.join(newPath, os.path.split(mediafile)[1]), 'hard')
removeEmptyFolders(path, removeRoot=False) removeEmptyFolders(path, removeRoot=False)