cmtagmylar.py

def run(dirName, nzbName=None):
  #!/usr/bin/python

  ##    IMPORTANT    ##
  ##  YOU MUST PLACE THIS SCRIPT IN YOUR SABNZBD SCRIPTS FOLDER ##
  ##  AND YOU MUST SET THE COMICTAGGER DIRECTORY/PATH BELOW ##
  ##  FINALLY, YOU MUST MAKE THIS FILE EXECUTABLE (sudo chmod a+x /path/to/cmtagmylar.py) ##

  import os
  import sys
  import glob
  import platform
  import shutil
  import zipfile
  import subprocess

  ## Set the directory in which comictagger and other external commands are located - IMPORTANT - ##
  # ( User may have to modify, depending on their setup, but these are some guesses for now )

  if platform.system() == "Windows":
    (x, y) = platform.architecture()
    if x == "64bit":
      comictagger_cmd = "C:\Program Files (x86)\ComicTagger\comictagger.exe"
      # http://www.win-rar.com/download.html
    else:
      comictagger_cmd = "C:\Program Files\ComicTagger\comictagger.exe"
    unrar_cmd =       "C:\Program Files\WinRAR\UnRAR.exe"
    
  elif platform.system() == "Darwin":  #Mac OS X
    comictagger_cmd = "/Applications/ComicTagger.app/Contents/MacOS/ComicTagger"
    unrar_cmd =       "/usr/local/bin/unrar"
    
  else:
    comictagger_cmd = "/path/to/comictagger/comictagger.py"
    unrar_cmd =       "/usr/bin/unrar"

  if not os.path.exists( comictagger_cmd ):
    print "ERROR:  can't find the ComicTagger program: {0}".format( comictagger_cmd )
    print "        You probably need to edit this script!"
    sys.exit( 1 )

  file_conversion = True
  file_extension_fixing = True
  if not os.path.exists( unrar_cmd ):
    print "WARNING:  can't find the unrar command."
    print "          File conversion and extension fixing not available"
    print "          You probably need to edit this script, or install the missing tool, or both!"
    file_conversion = False
    file_extension_fixing = False


  ## Sets up other directories ##
  scriptname = os.path.basename( sys.argv[0] )
  downloadpath = os.path.abspath( dirName ) 
  sabnzbdscriptpath = os.path.dirname( sys.argv[0] )
  comicpath = os.path.join( downloadpath , "temp" )
  unrar_folder = os.path.join( comicpath , "unrard" )

  print "----------------------------------------------"
  print "Running {0}, the Post-SabNZBd/Mylar script".format( scriptname )
  print "----------------------------------------------"

  if os.path.exists( comicpath ):
    shutil.rmtree( comicpath )
  os.makedirs( comicpath )

  # make a list of all CBR and CBZ files in downloadpath
  filename_list = glob.glob( os.path.join( downloadpath, "*.cbz" ) )
  filename_list.extend( glob.glob( os.path.join( downloadpath, "*.cbr" ) ) )

  ## Takes all .cbr and .cbz files and dumps them to processing directory ##
  for f in filename_list:
    shutil.move( f, comicpath)

  ## Changes filetype extensions when needed ##
  cbr_list = glob.glob( os.path.join( comicpath, "*.cbr" ) )
  for f in cbr_list:
    if zipfile.is_zipfile( f ):        
      base = os.path.splitext( f )[0]
      shutil.move( f, base + ".cbz" )
      print "{0}: renaming {1} to be a cbz".format( scriptname, os.path.basename( f ) )

  if file_extension_fixing:
    cbz_list = glob.glob( os.path.join( comicpath, "*.cbz" ) )
    for f in cbz_list:
      try:
        rar_test_cmd_output = "is not RAR archive" #default, in case of error
        rar_test_cmd_output = subprocess.check_output( [ unrar_cmd, "t", f ] )
      except:
        pass
      if not "is not RAR archive" in rar_test_cmd_output:
        base = os.path.splitext( f )[0]
        shutil.move( f, base + ".cbr" )
        print "{0}: renaming {1} to be a cbr".format( scriptname, os.path.basename( f ) )

  # Now rename all CBR files to RAR
  cbr_list = glob.glob( os.path.join( comicpath, "*.cbr" ) )
  for f in cbr_list:
    base = os.path.splitext( f )[0]
    shutil.move( f, base + ".rar" )

  ## Changes any cbr files to cbz files for insertion of metadata ##
  if file_conversion:
    rar_list = glob.glob( os.path.join( comicpath, "*.rar" ) )
    for f in rar_list:
      print "{0}: converting {1} to be zip format".format( scriptname, os.path.basename( f ) )
      basename = os.path.splitext( f )[0]
      zipname = basename + ".cbz"

      # Move into the folder where we will be unrar-ing things
      os.makedirs( unrar_folder )
      os.chdir( unrar_folder )

      # Extract and zip up
      subprocess.Popen( [ unrar_cmd, "x", f ] ).communicate()
      shutil.make_archive( basename, "zip", unrar_folder )

      # get out of unrar folder and clean up
      os.chdir( comicpath )
      #shutil.rmtree( unrar_folder )
      
      ## Changes zip to cbz
    zip_list = glob.glob( os.path.join( comicpath, "*.zip" ) )
    for f in zip_list:
      base = os.path.splitext( f )[0]
      shutil.move( f, base + ".cbz" )

  ## Tag each CBZ, and move it back to original directory ##
  cbz_list = glob.glob( os.path.join( comicpath, "*.cbz" ) )
  for f in cbz_list:
      subprocess.Popen( [ comictagger_cmd, "-s", "-t", "cr", "-f", "-o", "--verbose", "--nooverwrite", f ] ).communicate()
      subprocess.Popen( [ comictagger_cmd, "-s", "-t", "cbl", "-f", "-o", "--verbose", "--nooverwrite", f ] ).communicate()
      shutil.move( f, downloadpath )

  ## Clean up temp directory  ##
  os.chdir( sabnzbdscriptpath )
  shutil.rmtree( comicpath )

  ## Will Run Mylar Post=processing In Future ##