def __init__(self, directory, filename):
# Default settings for SickBeard
sb_defaults = {
'host': 'localhost',
'port': '8081',
'ssl': "False",
'api_key': '',
}
# Default MP4 conversion settings
mp4_defaults = {
'ffmpeg': 'ffmpeg.exe',
'ffprobe': 'ffprobe.exe',
'output_directory': '',
'copy_to': '',
'move_to': '',
'output_extension': 'mp4',
'output_format': 'mp4',
'delete_original': 'True',
'relocate_moov': 'True',
'ios-audio': 'True',
'ios-first-track-only': 'False',
'max-audio-channels': '',
'audio-language': '',
'audio-codec': 'ac3',
'audio-channel-bitrate': '256',
'video-codec': 'h264, x264',
'video-bitrate': '',
'subtitle-language': '',
'audio-default-language': '',
'subtitle-default-language': '',
'convert-mp4': 'False',
'fullpathguess': 'True',
'tagfile': 'True',
'tag-language': 'en',
'download-artwork': 'True',
'download-subs': 'False',
'embed-subs': 'True',
'sub-providers': 'addic7ed, podnapisi, thesubdb, opensubtitles'
}
# Default settings for CouchPotato
cp_defaults = {
'host': 'localhost',
'port': '5050',
'username': '',
'password': '',
'apikey': '',
'delay': '65',
'method': 'renamer',
'delete_failed': 'False',
'ssl': 'False',
'web_root': ''
}
# Default uTorrent settings
utorrent_defaults = {
'label': '',
'webui': 'False',
'action_before': 'stop',
'action_after': 'removedata',
'host': 'http://localhost:8080/',
'username': '',
'password': ''
}
defaults = {
'SickBeard': sb_defaults,
'CouchPotato': cp_defaults,
'MP4': mp4_defaults,
'uTorrent': utorrent_defaults
}
write = False # Will be changed to true if a value is missing from the config file and needs to be written
config = ConfigParser.SafeConfigParser()
configFile = os.path.join(directory, filename)
if os.path.isfile(configFile):
fp = open(configFile, "rb")
config.readfp(fp)
fp.close()
else:
print "Error: Config file not found, creating"
#config.filename = filename
write = True
# Make sure all sections and all keys for each section are present
for s in defaults:
if not config.has_section(s):
config.add_section(s)
write = True
for k in defaults[s]:
if not config.has_option(s, k):
config.set(s, k, defaults[s][k])
write = True
# If any keys are missing from the config file, write them
if write:
self.writeConfig(config, configFile)
#Read relevant MP4 section information
section = "MP4"
self.ffmpeg = os.path.normpath(self.raw(config.get(
section, "ffmpeg"))) # Location of FFMPEG.exe
self.ffprobe = os.path.normpath(
self.raw(config.get(section,
"ffprobe"))) # Location of FFPROBE.exe
self.output_dir = config.get(section, "output_directory")
if self.output_dir == '':
self.output_dir = None
else:
self.output_dir = os.path.normpath(self.raw(
self.output_dir)) # Output directory
self.copyto = config.get(
section,
"copy_to") # Directories to make copies of the final product
if self.copyto == '':
self.copyto = None
else:
self.copyto = self.copyto.split('|')
for i in xrange(len(self.copyto)):
self.copyto[i] = os.path.normpath(self.copyto[i])
if not os.path.isdir(self.copyto[i]):
try:
os.makedirs(self.copyto[i])
except:
print "Error making directory %s" % (self.copyto[i])
self.moveto = config.get(
section, "move_to") # Directory to move final product to
if self.moveto == '':
self.moveto = None
else:
self.moveto = os.path.normpath(self.moveto)
if not os.path.isdir(self.moveto):
try:
os.makedirs(self.moveto)
except:
print "Error making directory %s" % (self.moveto)
self.moveto = None
self.output_extension = config.get(
section, "output_extension") # Output extension
self.output_format = config.get(section,
"output_format") # Output format
if self.output_format not in valid_formats:
self.output_format = 'mov'
self.delete = config.getboolean(
section, "delete_original") # Delete original file
self.relocate_moov = config.getboolean(
section, "relocate_moov") # Relocate MOOV atom to start of file
self.acodec = config.get(section, "audio-codec").lower(
) # Gets the desired audio codec, if no valid codec selected, default to AC3
if self.acodec == '':
self.acodec == ['ac3']
else:
self.acodec = self.acodec.lower().replace(' ', '').split(',')
self.abitrate = config.get(section, "audio-channel-bitrate")
try:
self.abitrate = int(self.abitrate)
except:
self.abitrate = 256
print "Audio bitrate was invalid, defaulting to 256 per channel"
if self.abitrate > 256:
print "Warning - audio bitrates >256 may create errors with common codecs"
# !!! Leaving this disabled for now, users will be responsible for knowing whicn codecs do and don't work with mp4 files !!!
#if self.acodec not in valid_audio_codecs:
# self.acodec = 'aac'
# print "Audio codec not valid, defaulting to AAC"
self.iOS = config.get(
section, "ios-audio"
) # Creates a second audio channel if the standard output methods are different from this for iOS compatability
if self.iOS == "" or self.iOS.lower() in ['false', 'no', 'f', '0']:
self.iOS = False
else:
if self.iOS.lower() in ['true', 'yes', 't', '1']:
self.iOS = 'aac'
self.iOSFirst = config.getboolean(
section, "ios-first-track-only"
) # Enables the iOS audio option only for the first track
self.downloadsubs = config.getboolean(
section, "download-subs"
) # Enables downloading of subtitles from the internet sources using subliminal
if self.downloadsubs:
try:
import subliminal
except Exception as e:
print e
self.downloadsubs = False
print "Subliminal is not installed, automatically downloading of subs has been disabled"
self.subproviders = config.get(section, 'sub-providers').lower()
if self.subproviders == '':
self.downloadsubs = False
print "You must specifiy at least one subtitle provider to downlaod subs automatically"
else:
self.subproviders = self.subproviders.lower().replace(
' ', '').split(',')
self.embedsubs = config.getboolean(section, 'embed-subs')
#Setup variable for maximum audio channels
self.maxchannels = config.get(section, 'max-audio-channels')
if self.maxchannels == "":
self.maxchannels = None
else:
try:
self.maxchannels = int(self.maxchannels)
except:
print "Invalid number of audio channels specified"
self.maxchannels = None
if self.maxchannels is not None and self.maxchannels < 1:
print "Must have at least 1 audio channel"
self.maxchannels = None
self.vcodec = config.get(section, "video-codec")
if self.vcodec == '':
self.vcodec == ['h264', 'x264']
else:
self.vcodec = self.vcodec.lower().replace(' ', '').split(',')
self.vbitrate = config.get(section, "video-bitrate")
if self.vbitrate == '':
self.vbitrate = None
else:
try:
self.vbitrate = int(self.vbitrate) * 1000
except:
print "Invalid video bitrate, defaulting to no video bitrate cap"
self.vbitrate = None
self.awl = config.get(section, 'audio-language').strip().lower(
) # List of acceptable languages for audio streams to be carried over from the original file, separated by a comma. Blank for all
if self.awl == '':
self.awl = None
else:
self.awl = self.awl.replace(' ', '').split(',')
self.swl = config.get(section, 'subtitle-language').strip().lower(
) # List of acceptable languages for subtitle streams to be carried over from the original file, separated by a comma. Blank for all
if self.swl == '':
self.swl = None
else:
self.swl = self.swl.replace(' ', '').split(',')
self.adl = config.get(section, 'audio-default-language').strip().lower(
) # What language to default an undefinied audio language tag to. If blank, it will remain undefined. This is useful for single language releases which tend to leave things tagged as und
if self.adl == "" or len(self.adl) > 3:
self.adl = None
self.sdl = config.get(section, 'subtitle-default-language').strip(
).lower(
) # What language to default an undefinied subtitle language tag to. If blank, it will remain undefined. This is useful for single language releases which tend to leave things tagged as und
if self.sdl == "" or len(self.sdl) > 3:
self.sdl = None
# Prevent incompatible combination of settings
if self.output_dir == "" and self.delete is False:
print "Error - you must specify an alternate output directory if you aren't going to delete the original file"
sys.exit()
# Create output directory if it does not exist
if self.output_dir is not None:
if not os.path.isdir(self.output_dir):
os.makedirs(self.output_dir)
self.processMP4 = config.getboolean(
section, "convert-mp4"
) # Determine whether or not to reprocess mp4 files or just tag them
self.fullpathguess = config.getboolean(
section, "fullpathguess") # Guess using the full path or not
self.tagfile = config.getboolean(section,
"tagfile") # Tag files with metadata
self.taglanguage = config.get(
section, "tag-language").strip().lower() # Language to tag files
if len(self.taglanguage) > 2:
try:
babel = Language.fromalpha3(self.taglanguage)
self.taglanguage = babel.alpha2
except:
print "Unable to set tag language, defaulting to English"
self.taglanguage = 'en'
elif len(self.taglanguage) < 2:
print "Unable to set tag language, defaulting to English"
self.taglanguage = 'en'
self.artwork = config.getboolean(
section, "download-artwork") # Download and embed artwork
#Read relevant CouchPotato section information
section = "CouchPotato"
self.CP = {}
self.CP['host'] = config.get(section, "host")
self.CP['port'] = config.get(section, "port")
self.CP['username'] = config.get(section, "username")
self.CP['password'] = config.get(section, "password")
self.CP['apikey'] = config.get(section, "apikey")
self.CP['delay'] = config.get(section, "delay")
self.CP['method'] = config.get(section, "method")
self.CP['web_root'] = config.get(section, "web_root")
try:
self.CP['delay'] = float(self.CP['delay'])
except ValueError:
self.CP['delay'] = 60
try:
self.CP['delete_failed'] = config.getboolean(
section, "delete_failed")
except (ConfigParser.NoOptionError, ValueError):
self.CP['delete_failed'] = False
try:
if config.getboolean(section, 'ssl'):
self.CP['protocol'] = "https://"
else:
self.CP['protocol'] = "http://"
except (ConfigParser.NoOptionError, ValueError):
self.CP['protocol'] = "http://"
#Read relevant uTorrent section information
section = "uTorrent"
self.uTorrentLabel = config.get(section, "label").lower()
self.uTorrentWebUI = config.getboolean(section, "webui")
self.uTorrentActionBefore = config.get(section,
"action_before").lower()
self.uTorrentActionAfter = config.get(section, "action_after").lower()
self.uTorrentHost = config.get(section, "host").lower()
self.uTorrentUsername = config.get(section, "username")
self.uTorrentPassword = config.get(section, "password")
if self.uTorrentLabel == "":
self.uTorrentLabel = None
#Pass the values on
self.config = config
self.configFile = configFile
def __init__(self, directory, filename, logger=None):
# Setup logging
if logger:
log = logger
else:
log = logging.getLogger(__name__)
try:
from babelfish import Language
except:
log.debug("Trying to install SetupTools.")
try:
import setup
setup.ez_setup.main()
from babelfish import Language
except:
log.exception("Please install SetupTools.")
# Default settings for SickBeard
sb_defaults = {
'host': 'localhost',
'port': '8081',
'ssl': "False",
'api_key': '',
'web_root': '',
'username': '',
'password': ''
}
# Default MP4 conversion settings
mp4_defaults = {
'ffmpeg': 'ffmpeg.exe',
'ffprobe': 'ffprobe.exe',
'output_directory': '',
'copy_to': '',
'move_to': '',
'output_extension': 'mp4',
'output_format': 'mp4',
'delete_original': 'True',
'relocate_moov': 'True',
'ios-audio': 'True',
'ios-first-track-only': 'False',
'max-audio-channels': '',
'audio-language': '',
'audio-default-language': '',
'audio-codec': 'ac3',
'audio-channel-bitrate': '256',
'video-codec': 'h264, x264',
'video-bitrate': '',
'video-max-width': '',
'subtitle-codec': 'mov_text',
'subtitle-language': '',
'subtitle-default-language': '',
'convert-mp4': 'False',
'fullpathguess': 'True',
'tagfile': 'True',
'tag-language': 'en',
'download-artwork': 'True',
'download-subs': 'False',
'embed-subs': 'True',
'sub-providers': 'addic7ed, podnapisi, thesubdb, opensubtitles',
'permissions': '777'
}
# Default settings for CouchPotato
cp_defaults = {
'host': 'localhost',
'port': '5050',
'username': '',
'password': '',
'apikey': '',
'delay': '65',
'method': 'renamer',
'delete_failed': 'False',
'ssl': 'False',
'web_root': ''
}
# Default settings for Sonarr
sonarr_defaults = {
'host': 'localhost',
'port': '8989',
'apikey': '',
'ssl': 'False',
'web_root': ''
}
# Default uTorrent settings
utorrent_defaults = {
'couchpotato-label': 'couchpotato',
'sickbeard-label': 'sickbeard',
'sickrage-label': 'sickrage',
'sonarr-label': 'sonarr',
'bypass-label': 'bypass',
'convert': 'True',
'webui': 'False',
'action_before': 'stop',
'action_after': 'removedata',
'host': 'http://localhost:8080/',
'username': '',
'password': ''
}
# Default SAB settings
sab_defaults = {
'convert': 'True',
'Sickbeard-category': 'sickbeard',
'Sickrage-category': 'sickrage',
'Couchpotato-category': 'couchpotato',
'Sonarr-category': 'sonarr',
'Bypass-category': 'bypass'
}
# Default Sickrage Settings
sr_defaults = {
'host': 'localhost',
'port': '8081',
'ssl': "False",
'api_key': '',
'web_root': '',
'username': '',
'password': ''
}
# Default deluge settings
deluge_defaults = {
'couchpotato-label': 'couchpotato',
'sickbeard-label': 'sickbeard',
'sickrage-label': 'sickrage',
'sonarr-label': 'sonarr',
'bypass-label': 'bypass',
'convert': 'True',
'host': 'localhost',
'port': '58846',
'username': '',
'password': ''
}
# Default Plex Settings
plex_defaults = {
'host': 'localhost',
'port': '32400',
'refresh': 'true'
}
defaults = {
'SickBeard': sb_defaults,
'CouchPotato': cp_defaults,
'Sonarr': sonarr_defaults,
'MP4': mp4_defaults,
'uTorrent': utorrent_defaults,
'SABNZBD': sab_defaults,
'Sickrage': sr_defaults,
'Deluge': deluge_defaults,
'Plex': plex_defaults
}
write = False # Will be changed to true if a value is missing from the config file and needs to be written
config = configparser.SafeConfigParser()
configFile = os.path.join(directory, filename)
if os.path.isfile(configFile):
config.read(configFile)
else:
log.error("Config file not found, creating %s." % configFile)
#config.filename = filename
write = True
# Make sure all sections and all keys for each section are present
for s in defaults:
if not config.has_section(s):
config.add_section(s)
write = True
for k in defaults[s]:
if not config.has_option(s, k):
config.set(s, k, defaults[s][k])
write = True
# If any keys are missing from the config file, write them
if write:
self.writeConfig(config, configFile)
#Read relevant MP4 section information
section = "MP4"
self.ffmpeg = os.path.normpath(self.raw(config.get(
section, "ffmpeg"))) # Location of FFMPEG.exe
self.ffprobe = os.path.normpath(
self.raw(config.get(section,
"ffprobe"))) # Location of FFPROBE.exe
self.output_dir = config.get(section, "output_directory")
if self.output_dir == '':
self.output_dir = None
else:
self.output_dir = os.path.normpath(self.raw(
self.output_dir)) # Output directory
self.copyto = config.get(
section,
"copy_to") # Directories to make copies of the final product
if self.copyto == '':
self.copyto = None
else:
self.copyto = self.copyto.split('|')
for i in xrange(len(self.copyto)):
self.copyto[i] = os.path.normpath(self.copyto[i])
if not os.path.isdir(self.copyto[i]):
try:
os.makedirs(self.copyto[i])
except:
log.exception("Error making directory %s." %
(self.copyto[i]))
self.moveto = config.get(
section, "move_to") # Directory to move final product to
if self.moveto == '':
self.moveto = None
else:
self.moveto = os.path.normpath(self.moveto)
if not os.path.isdir(self.moveto):
try:
os.makedirs(self.moveto)
except:
log.exception("Error making directory %s." % (self.moveto))
self.moveto = None
self.output_extension = config.get(
section, "output_extension") # Output extension
self.output_format = config.get(section,
"output_format") # Output format
if self.output_format not in valid_formats:
self.output_format = 'mov'
self.delete = config.getboolean(
section, "delete_original") # Delete original file
self.relocate_moov = config.getboolean(
section, "relocate_moov") # Relocate MOOV atom to start of file
self.acodec = config.get(section, "audio-codec").lower(
) # Gets the desired audio codec, if no valid codec selected, default to AC3
if self.acodec == '':
self.acodec == ['ac3']
else:
self.acodec = self.acodec.lower().replace(' ', '').split(',')
self.abitrate = config.get(section, "audio-channel-bitrate")
try:
self.abitrate = int(self.abitrate)
except:
self.abitrate = 256
log.warning(
"Audio bitrate was invalid, defaulting to 256 per channel.")
if self.abitrate > 256:
log.warning(
"Audio bitrates >256 may create errors with common codecs.")
# !!! Leaving this disabled for now, users will be responsible for knowing whicn codecs do and don't work with mp4 files !!!
#if self.acodec not in valid_audio_codecs:
# self.acodec = 'aac'
# print "Audio codec not valid, defaulting to AAC"
self.iOS = config.get(
section, "ios-audio"
) # Creates a second audio channel if the standard output methods are different from this for iOS compatability
if self.iOS == "" or self.iOS.lower() in ['false', 'no', 'f', '0']:
self.iOS = False
else:
if self.iOS.lower() in ['true', 'yes', 't', '1']:
self.iOS = 'aac'
self.iOSFirst = config.getboolean(
section, "ios-first-track-only"
) # Enables the iOS audio option only for the first track
self.downloadsubs = config.getboolean(
section, "download-subs"
) # Enables downloading of subtitles from the internet sources using subliminal
if self.downloadsubs:
try:
import subliminal
except Exception as e:
self.downloadsubs = False
log.exception(
"Subliminal is not installed, automatically downloading of subs has been disabled."
)
self.subproviders = config.get(section, 'sub-providers').lower()
if self.subproviders == '':
self.downloadsubs = False
log.warning(
"You must specifiy at least one subtitle provider to downlaod subs automatically, subtitle downloading disabled."
)
else:
self.subproviders = self.subproviders.lower().replace(
' ', '').split(',')
self.embedsubs = config.getboolean(section, 'embed-subs')
self.permissions = config.get(section, 'permissions')
try:
self.permissions = int(self.permissions, 8)
except:
self.log.exception("Invalid permissions, defaulting to 777.")
self.permissions = int("0777", 8)
#Setup variable for maximum audio channels
self.maxchannels = config.get(section, 'max-audio-channels')
if self.maxchannels == "":
self.maxchannels = None
else:
try:
self.maxchannels = int(self.maxchannels)
except:
log.exception("Invalid number of audio channels specified.")
self.maxchannels = None
if self.maxchannels is not None and self.maxchannels < 1:
log.warning("Must have at least 1 audio channel.")
self.maxchannels = None
self.vcodec = config.get(section, "video-codec")
if self.vcodec == '':
self.vcodec == ['h264', 'x264']
else:
self.vcodec = self.vcodec.lower().replace(' ', '').split(',')
self.vbitrate = config.get(section, "video-bitrate")
if self.vbitrate == '':
self.vbitrate = None
else:
try:
self.vbitrate = int(self.vbitrate)
except:
log.exception(
"Invalid video bitrate, defaulting to no video bitrate cap."
)
self.vbitrate = None
self.vwidth = config.get(section, "video-max-width")
if self.vwidth == '':
self.vwidth = None
else:
try:
self.vwidth = int(self.vwidth)
except:
log.exception("Invalid video width, defaulting to none.")
self.vwidth = None
self.awl = config.get(section, 'audio-language').strip().lower(
) # List of acceptable languages for audio streams to be carried over from the original file, separated by a comma. Blank for all
if self.awl == '':
self.awl = None
else:
self.awl = self.awl.replace(' ', '').split(',')
self.scodec = config.get(section, 'subtitle-codec').strip().lower()
if not self.scodec or self.scodec == "":
if self.embedsubs:
self.scodec = 'mov_text'
else:
self.scodec = 'srt'
log.warning("Invalid subtitle codec, defaulting to '%s'." %
self.scodec)
if self.embedsubs and self.scodec not in valid_internal_subcodecs:
log.warning(
"Invalid interal subtitle codec %s, defaulting to 'mov_text'."
% self.scodec)
self.scodec = 'mov_text'
if not self.embedsubs and self.scodec not in valid_external_subcodecs:
log.warning(
"Invalid external subtitle codec %s, defaulting to 'srt'." %
self.scodec)
self.scodec = 'srt'
self.swl = config.get(section, 'subtitle-language').strip().lower(
) # List of acceptable languages for subtitle streams to be carried over from the original file, separated by a comma. Blank for all
if self.swl == '':
self.swl = None
else:
self.swl = self.swl.replace(' ', '').split(',')
self.adl = config.get(section, 'audio-default-language').strip().lower(
) # What language to default an undefinied audio language tag to. If blank, it will remain undefined. This is useful for single language releases which tend to leave things tagged as und
if self.adl == "" or len(self.adl) > 3:
self.adl = None
self.sdl = config.get(section, 'subtitle-default-language').strip(
).lower(
) # What language to default an undefinied subtitle language tag to. If blank, it will remain undefined. This is useful for single language releases which tend to leave things tagged as und
if self.sdl == "" or len(self.sdl) > 3:
self.sdl = None
# Prevent incompatible combination of settings
if self.output_dir == "" and self.delete is False:
log.error(
"You must specify an alternate output directory if you aren't going to delete the original file."
)
sys.exit()
# Create output directory if it does not exist
if self.output_dir is not None:
if not os.path.isdir(self.output_dir):
os.makedirs(self.output_dir)
self.processMP4 = config.getboolean(
section, "convert-mp4"
) # Determine whether or not to reprocess mp4 files or just tag them
self.fullpathguess = config.getboolean(
section, "fullpathguess") # Guess using the full path or not
self.tagfile = config.getboolean(section,
"tagfile") # Tag files with metadata
self.taglanguage = config.get(
section, "tag-language").strip().lower() # Language to tag files
if len(self.taglanguage) > 2:
try:
babel = Language.fromalpha3(self.taglanguage)
self.taglanguage = babel.alpha2
except:
log.exception(
"Unable to set tag language, defaulting to English.")
self.taglanguage = 'en'
elif len(self.taglanguage) < 2:
log.exception("Unable to set tag language, defaulting to English.")
self.taglanguage = 'en'
self.artwork = config.getboolean(
section, "download-artwork") # Download and embed artwork
#Read relevant CouchPotato section information
section = "CouchPotato"
self.CP = {}
self.CP['host'] = config.get(section, "host")
self.CP['port'] = config.get(section, "port")
self.CP['username'] = config.get(section, "username")
self.CP['password'] = config.get(section, "password")
self.CP['apikey'] = config.get(section, "apikey")
self.CP['delay'] = config.get(section, "delay")
self.CP['method'] = config.get(section, "method")
self.CP['web_root'] = config.get(section, "web_root")
try:
self.CP['delay'] = float(self.CP['delay'])
except ValueError:
self.CP['delay'] = 60
try:
self.CP['delete_failed'] = config.getboolean(
section, "delete_failed")
except (configparser.NoOptionError, ValueError):
self.CP['delete_failed'] = False
try:
if config.getboolean(section, 'ssl'):
self.CP['protocol'] = "https://"
else:
self.CP['protocol'] = "http://"
except (configparser.NoOptionError, ValueError):
self.CP['protocol'] = "http://"
#Read relevant uTorrent section information
section = "uTorrent"
self.uTorrent = {}
self.uTorrent['cp'] = config.get(section, "couchpotato-label").lower()
self.uTorrent['sb'] = config.get(section, "sickbeard-label").lower()
self.uTorrent['sr'] = config.get(section, "sickrage-label").lower()
self.uTorrent['sonarr'] = config.get(section, "sonarr-label").lower()
self.uTorrent['bypass'] = config.get(section, "bypass-label").lower()
try:
self.uTorrent['convert'] = config.getboolean(section, "convert")
except:
self.uTorrent['convert'] = False
self.uTorrentWebUI = config.getboolean(section, "webui")
self.uTorrentActionBefore = config.get(section,
"action_before").lower()
self.uTorrentActionAfter = config.get(section, "action_after").lower()
self.uTorrentHost = config.get(section, "host").lower()
self.uTorrentUsername = config.get(section, "username")
self.uTorrentPassword = config.get(section, "password")
#Read relevant Deluge section information
section = "Deluge"
self.deluge = {}
self.deluge['cp'] = config.get(section, "couchpotato-label").lower()
self.deluge['sb'] = config.get(section, "sickbeard-label").lower()
self.deluge['sr'] = config.get(section, "sickrage-label").lower()
self.deluge['sonarr'] = config.get(section, "sonarr-label").lower()
self.deluge['bypass'] = config.get(section, "bypass-label").lower()
try:
self.deluge['convert'] = config.getboolean(section, "convert")
except:
self.deluge['convert'] = False
self.deluge['host'] = config.get(section, "host").lower()
self.deluge['port'] = config.get(section, "port")
self.deluge['user'] = config.get(section, "username")
self.deluge['pass'] = config.get(section, "password")
#Read relevant Sonarr section information
section = "Sonarr"
self.Sonarr = {}
self.Sonarr['host'] = config.get(section, "host")
self.Sonarr['port'] = config.get(section, "port")
self.Sonarr['apikey'] = config.get(section, "apikey")
self.Sonarr['ssl'] = config.get(section, "ssl")
self.Sonarr['web_root'] = config.get(section, "web_root")
#Read Sickbeard section information
section = "SickBeard"
self.Sickbeard = {}
self.Sickbeard['host'] = config.get(section, "host") # Server Address
self.Sickbeard['port'] = config.get(section, "port") # Server Port
self.Sickbeard['api_key'] = config.get(section,
"api_key") # Sickbeard API key
self.Sickbeard['web_root'] = config.get(
section, "web_root") # Sickbeard webroot
self.Sickbeard['ssl'] = config.getboolean(section, "ssl") # SSL
self.Sickbeard['user'] = config.get(section, "username")
self.Sickbeard['pass'] = config.get(section, "password")
#Read Sickrage section information
section = "Sickrage"
self.Sickrage = {}
self.Sickrage['host'] = config.get(section, "host") # Server Address
self.Sickrage['port'] = config.get(section, "port") # Server Port
self.Sickrage['api_key'] = config.get(section,
"api_key") # Sickbeard API key
self.Sickrage['web_root'] = config.get(section,
"web_root") # Sickbeard webroot
self.Sickrage['ssl'] = config.getboolean(section, "ssl") # SSL
self.Sickrage['user'] = config.get(section, "username")
self.Sickrage['pass'] = config.get(section, "password")
#Read SAB section information
section = "SABNZBD"
self.SAB = {}
try:
self.SAB['convert'] = config.getboolean(section,
"convert") # Convert
except:
self.SAB['convert'] = False
self.SAB['cp'] = config.get(section, "Couchpotato-category").lower()
self.SAB['sb'] = config.get(section, "Sickbeard-category").lower()
self.SAB['sr'] = config.get(section, "Sickrage-category").lower()
self.SAB['sonarr'] = config.get(section, "Sonarr-category").lower()
self.SAB['bypass'] = config.get(section, "Bypass-category").lower()
#Read Plex section information
section = "Plex"
self.Plex = {}
self.Plex['host'] = config.get(section, "host")
self.Plex['port'] = config.get(section, "port")
try:
self.Plex['refresh'] = config.getboolean(section, "refresh")
except:
self.Plex['refresh'] = False
#Pass the values on
self.config = config
self.configFile = configFile
def __init__(self, directory, filename, logger=None):
# Setup logging
if logger:
log = logger
else:
log = logging.getLogger(__name__)
try:
from babelfish import Language
except:
log.debug("Trying to install SetupTools.")
try:
import setup
setup.ez_setup.main()
from babelfish import Language
except:
log.exception("Please install SetupTools.")
# Default settings for SickBeard
sb_defaults = {'host': 'localhost',
'port': '8081',
'ssl': "False",
'api_key': '',
'web_root': '',
'username': '',
'password': '' }
# Default MP4 conversion settings
mp4_defaults = {'ffmpeg': 'ffmpeg.exe',
'ffprobe': 'ffprobe.exe',
'output_directory': '',
'copy_to': '',
'move_to': '',
'output_extension': 'mp4',
'output_format': 'mp4',
'delete_original': 'True',
'relocate_moov': 'True',
'ios-audio': 'True',
'ios-first-track-only': 'False',
'max-audio-channels': '',
'audio-language': '',
'audio-default-language': '',
'audio-codec': 'ac3',
'audio-channel-bitrate': '256',
'video-codec': 'h264, x264',
'video-bitrate': '',
'video-max-width': '',
'h264-max-level': '',
'subtitle-codec': 'mov_text',
'subtitle-language': '',
'subtitle-default-language': '',
'convert-mp4': 'False',
'fullpathguess': 'True',
'tagfile': 'True',
'tag-language': 'en',
'download-artwork': 'True',
'download-subs': 'False',
'embed-subs': 'True',
'sub-providers': 'addic7ed, podnapisi, thesubdb, opensubtitles',
'permissions': '777',
'post-process': 'False',
'pix-fmt': '' }
# Default settings for CouchPotato
cp_defaults = {'host': 'localhost',
'port': '5050',
'username': '',
'password': '',
'apikey': '',
'delay': '65',
'method': 'renamer',
'delete_failed': 'False',
'ssl': 'False',
'web_root': '' }
# Default settings for Sonarr
sonarr_defaults = {'host': 'localhost',
'port': '8989',
'apikey': '',
'ssl': 'False',
'web_root': ''}
# Default uTorrent settings
utorrent_defaults = { 'couchpotato-label': 'couchpotato',
'sickbeard-label': 'sickbeard',
'sickrage-label': 'sickrage',
'sonarr-label': 'sonarr',
'bypass-label': 'bypass',
'convert': 'True',
'webui': 'False',
'action_before': 'stop',
'action_after': 'removedata',
'host': 'http://localhost:8080/',
'username': '',
'password': ''}
# Default SAB settings
sab_defaults = { 'convert': 'True',
'Sickbeard-category': 'sickbeard',
'Sickrage-category': 'sickrage',
'Couchpotato-category': 'couchpotato',
'Sonarr-category': 'sonarr',
'Bypass-category': 'bypass' }
# Default Sickrage Settings
sr_defaults = {'host': 'localhost',
'port': '8081',
'ssl': "False",
'api_key': '',
'web_root': '',
'username': '',
'password': '' }
# Default deluge settings
deluge_defaults = { 'couchpotato-label': 'couchpotato',
'sickbeard-label': 'sickbeard',
'sickrage-label': 'sickrage',
'sonarr-label': 'sonarr',
'bypass-label': 'bypass',
'convert': 'True',
'host': 'localhost',
'port': '58846',
'username': '',
'password': ''}
# Default Plex Settings
plex_defaults = {'host': 'localhost',
'port': '32400',
'refresh': 'true',
'token': ''}
defaults = {'SickBeard': sb_defaults, 'CouchPotato': cp_defaults, 'Sonarr': sonarr_defaults, 'MP4': mp4_defaults, 'uTorrent': utorrent_defaults, 'SABNZBD': sab_defaults, 'Sickrage': sr_defaults, 'Deluge': deluge_defaults, 'Plex': plex_defaults}
write = False # Will be changed to true if a value is missing from the config file and needs to be written
config = configparser.SafeConfigParser()
configFile = os.path.join(directory, filename)
if os.path.isfile(configFile):
config.read(configFile)
else:
log.error("Config file not found, creating %s." % configFile)
#config.filename = filename
write = True
# Make sure all sections and all keys for each section are present
for s in defaults:
if not config.has_section(s):
config.add_section(s)
write = True
for k in defaults[s]:
if not config.has_option(s, k):
config.set(s, k, defaults[s][k])
write = True
# If any keys are missing from the config file, write them
if write:
self.writeConfig(config, configFile)
#Read relevant MP4 section information
section = "MP4"
self.ffmpeg = os.path.normpath(self.raw(config.get(section, "ffmpeg"))) # Location of FFMPEG.exe
self.ffprobe = os.path.normpath(self.raw(config.get(section, "ffprobe"))) # Location of FFPROBE.exe
self.output_dir = config.get(section, "output_directory")
if self.output_dir == '':
self.output_dir = None
else:
self.output_dir = os.path.normpath(self.raw(self.output_dir)) # Output directory
self.copyto = config.get(section, "copy_to") # Directories to make copies of the final product
if self.copyto == '':
self.copyto = None
else:
self.copyto = self.copyto.split('|')
for i in xrange(len(self.copyto)):
self.copyto[i] = os.path.normpath(self.copyto[i])
if not os.path.isdir(self.copyto[i]):
try:
os.makedirs(self.copyto[i])
except:
log.exception("Error making directory %s." % (self.copyto[i]))
self.moveto = config.get(section, "move_to") # Directory to move final product to
if self.moveto == '':
self.moveto = None
else:
self.moveto = os.path.normpath(self.moveto)
if not os.path.isdir(self.moveto):
try:
os.makedirs(self.moveto)
except:
log.exception("Error making directory %s." % (self.moveto))
self.moveto = None
self.output_extension = config.get(section, "output_extension") # Output extension
self.output_format = config.get(section, "output_format") # Output format
if self.output_format not in valid_formats:
self.output_format = 'mov'
self.delete = config.getboolean(section, "delete_original") # Delete original file
self.relocate_moov = config.getboolean(section, "relocate_moov") # Relocate MOOV atom to start of file
self.acodec = config.get(section, "audio-codec").lower() # Gets the desired audio codec, if no valid codec selected, default to AC3
if self.acodec == '':
self.acodec == ['ac3']
else:
self.acodec = self.acodec.lower().replace(' ', '').split(',')
self.abitrate = config.get(section, "audio-channel-bitrate")
try:
self.abitrate = int(self.abitrate)
except:
self.abitrate = 256
log.warning("Audio bitrate was invalid, defaulting to 256 per channel.")
if self.abitrate > 256:
log.warning("Audio bitrate >256 may create errors with common codecs.")
# !!! Leaving this disabled for now, users will be responsible for knowing whicn codecs do and don't work with mp4 files !!!
#if self.acodec not in valid_audio_codecs:
# self.acodec = 'aac'
# print "Audio codec not valid, defaulting to AAC"
self.iOS = config.get(section, "ios-audio") # Creates a second audio channel if the standard output methods are different from this for iOS compatability
if self.iOS == "" or self.iOS.lower() in ['false', 'no', 'f', '0']:
self.iOS = False
else:
if self.iOS.lower() in ['true', 'yes', 't', '1']:
self.iOS = 'aac'
self.iOSFirst = config.getboolean(section, "ios-first-track-only") # Enables the iOS audio option only for the first track
self.downloadsubs = config.getboolean(section, "download-subs") # Enables downloading of subtitles from the internet sources using subliminal
if self.downloadsubs:
try:
import subliminal
except Exception as e:
self.downloadsubs = False
log.exception("Subliminal is not installed, automatically downloading of subs has been disabled.")
self.subproviders = config.get(section, 'sub-providers').lower()
if self.subproviders == '':
self.downloadsubs = False
log.warning("You must specifiy at least one subtitle provider to downlaod subs automatically, subtitle downloading disabled.")
else:
self.subproviders = self.subproviders.lower().replace(' ', '').split(',')
self.embedsubs = config.getboolean(section, 'embed-subs')
self.permissions = config.get(section, 'permissions')
try:
self.permissions = int(self.permissions, 8)
except:
self.log.exception("Invalid permissions, defaulting to 777.")
self.permissions = int("0777", 8)
try:
self.postprocess = config.getboolean(section, 'post-process')
except:
self.postprocess = False
#Setup variable for maximum audio channels
self.maxchannels = config.get(section, 'max-audio-channels')
if self.maxchannels == "":
self.maxchannels = None
else:
try:
self.maxchannels = int(self.maxchannels)
except:
log.exception("Invalid number of audio channels specified.")
self.maxchannels = None
if self.maxchannels is not None and self.maxchannels < 1:
log.warning("Must have at least 1 audio channel.")
self.maxchannels = None
self.vcodec = config.get(section, "video-codec")
if self.vcodec == '':
self.vcodec == ['h264', 'x264']
else:
self.vcodec = self.vcodec.lower().replace(' ', '').split(',')
self.vbitrate = config.get(section, "video-bitrate")
if self.vbitrate == '':
self.vbitrate = None
else:
try:
self.vbitrate = int(self.vbitrate)
if not (self.vbitrate > 0):
self.vbitrate = None
log.warning("Video bitrate must be greater than 0, defaulting to no video bitrate cap.")
except:
log.exception("Invalid video bitrate, defaulting to no video bitrate cap.")
self.vbitrate = None
self.vwidth = config.get(section, "video-max-width")
if self.vwidth == '':
self.vwidth = None
else:
try:
self.vwidth = int(self.vwidth)
except:
log.exception("Invalid video width, defaulting to none.")
self.vwidth = None
self.h264_level = config.get(section, "h264-max-level")
if self.h264_level == '':
self.h264_level = None
else:
try:
self.h264_level = float(self.h264_level)
except:
log.exception("Invalid h264 level, defaulting to none.")
self.h264_level = None
self.pix_fmt = config.get(section, "pix-fmt").strip().lower()
if self.pix_fmt == '':
self.pix_fmt = None
self.awl = config.get(section, 'audio-language').strip().lower() # List of acceptable languages for audio streams to be carried over from the original file, separated by a comma. Blank for all
if self.awl == '':
self.awl = None
else:
self.awl = self.awl.replace(' ', '').split(',')
self.scodec = config.get(section, 'subtitle-codec').strip().lower()
if not self.scodec or self.scodec == "":
if self.embedsubs:
self.scodec = 'mov_text'
else:
self.scodec = 'srt'
log.warning("Invalid subtitle codec, defaulting to '%s'." % self.scodec)
if self.embedsubs and self.scodec not in valid_internal_subcodecs:
log.warning("Invalid interal subtitle codec %s, defaulting to 'mov_text'." % self.scodec)
self.scodec = 'mov_text'
if not self.embedsubs and self.scodec not in valid_external_subcodecs:
log.warning("Invalid external subtitle codec %s, defaulting to 'srt'." % self.scodec)
self.scodec = 'srt'
self.swl = config.get(section, 'subtitle-language').strip().lower() # List of acceptable languages for subtitle streams to be carried over from the original file, separated by a comma. Blank for all
if self.swl == '':
self.swl = None
else:
self.swl = self.swl.replace(' ', '').split(',')
self.adl = config.get(section, 'audio-default-language').strip().lower() # What language to default an undefinied audio language tag to. If blank, it will remain undefined. This is useful for single language releases which tend to leave things tagged as und
if self.adl == "" or len(self.adl) > 3:
self.adl = None
self.sdl = config.get(section, 'subtitle-default-language').strip().lower() # What language to default an undefinied subtitle language tag to. If blank, it will remain undefined. This is useful for single language releases which tend to leave things tagged as und
if self.sdl == ""or len(self.sdl) > 3:
self.sdl = None
# Prevent incompatible combination of settings
if self.output_dir == "" and self.delete is False:
log.error("You must specify an alternate output directory if you aren't going to delete the original file.")
sys.exit()
# Create output directory if it does not exist
if self.output_dir is not None:
if not os.path.isdir(self.output_dir):
os.makedirs(self.output_dir)
self.processMP4 = config.getboolean(section, "convert-mp4") # Determine whether or not to reprocess mp4 files or just tag them
self.fullpathguess = config.getboolean(section, "fullpathguess") # Guess using the full path or not
self.tagfile = config.getboolean(section, "tagfile") # Tag files with metadata
self.taglanguage = config.get(section, "tag-language").strip().lower() # Language to tag files
if len(self.taglanguage) > 2:
try:
babel = Language.fromalpha3(self.taglanguage)
self.taglanguage = babel.alpha2
except:
log.exception("Unable to set tag language, defaulting to English.")
self.taglanguage = 'en'
elif len(self.taglanguage) < 2:
log.exception("Unable to set tag language, defaulting to English.")
self.taglanguage = 'en'
self.artwork = config.getboolean(section, "download-artwork") # Download and embed artwork
#Read relevant CouchPotato section information
section = "CouchPotato"
self.CP = {}
self.CP['host'] = config.get(section, "host")
self.CP['port'] = config.get(section, "port")
self.CP['username'] = config.get(section, "username")
self.CP['password'] = config.get(section, "password")
self.CP['apikey'] = config.get(section, "apikey")
self.CP['delay'] = config.get(section, "delay")
self.CP['method'] = config.get(section, "method")
self.CP['web_root'] = config.get(section, "web_root")
try:
self.CP['delay'] = float(self.CP['delay'])
except ValueError:
self.CP['delay'] = 60
try:
self.CP['delete_failed'] = config.getboolean(section, "delete_failed")
except (configparser.NoOptionError, ValueError):
self.CP['delete_failed'] = False
try:
if config.getboolean(section, 'ssl'):
self.CP['protocol'] = "https://"
else:
self.CP['protocol'] = "http://"
except (configparser.NoOptionError, ValueError):
self.CP['protocol'] = "http://"
#Read relevant uTorrent section information
section = "uTorrent"
self.uTorrent = {}
self.uTorrent['cp'] = config.get(section, "couchpotato-label").lower()
self.uTorrent['sb'] = config.get(section, "sickbeard-label").lower()
self.uTorrent['sr'] = config.get(section, "sickrage-label").lower()
self.uTorrent['sonarr'] = config.get(section, "sonarr-label").lower()
self.uTorrent['bypass'] = config.get(section, "bypass-label").lower()
try:
self.uTorrent['convert'] = config.getboolean(section, "convert")
except:
self.uTorrent['convert'] = False
self.uTorrentWebUI = config.getboolean(section, "webui")
self.uTorrentActionBefore = config.get(section, "action_before").lower()
self.uTorrentActionAfter = config.get(section, "action_after").lower()
self.uTorrentHost = config.get(section, "host").lower()
self.uTorrentUsername = config.get(section, "username")
self.uTorrentPassword = config.get(section, "password")
#Read relevant Deluge section information
section = "Deluge"
self.deluge = {}
self.deluge['cp'] = config.get(section, "couchpotato-label").lower()
self.deluge['sb'] = config.get(section, "sickbeard-label").lower()
self.deluge['sr'] = config.get(section, "sickrage-label").lower()
self.deluge['sonarr'] = config.get(section, "sonarr-label").lower()
self.deluge['bypass'] = config.get(section, "bypass-label").lower()
try:
self.deluge['convert'] = config.getboolean(section, "convert")
except:
self.deluge['convert'] = False
self.deluge['host'] = config.get(section, "host").lower()
self.deluge['port'] = config.get(section, "port")
self.deluge['user'] = config.get(section, "username")
self.deluge['pass'] = config.get(section, "password")
#Read relevant Sonarr section information
section = "Sonarr"
self.Sonarr = {}
self.Sonarr['host'] = config.get(section, "host")
self.Sonarr['port'] = config.get(section, "port")
self.Sonarr['apikey'] = config.get(section, "apikey")
self.Sonarr['ssl'] = config.get(section, "ssl")
self.Sonarr['web_root'] = config.get(section, "web_root")
#Read Sickbeard section information
section = "SickBeard"
self.Sickbeard = {}
self.Sickbeard['host'] = config.get(section, "host") # Server Address
self.Sickbeard['port'] = config.get(section, "port") # Server Port
self.Sickbeard['api_key'] = config.get(section, "api_key") # Sickbeard API key
self.Sickbeard['web_root'] = config.get(section, "web_root") # Sickbeard webroot
self.Sickbeard['ssl'] = config.getboolean(section, "ssl") # SSL
self.Sickbeard['user'] = config.get(section, "username")
self.Sickbeard['pass'] = config.get(section, "password")
#Read Sickrage section information
section = "Sickrage"
self.Sickrage = {}
self.Sickrage['host'] = config.get(section, "host") # Server Address
self.Sickrage['port'] = config.get(section, "port") # Server Port
self.Sickrage['api_key'] = config.get(section, "api_key") # Sickbeard API key
self.Sickrage['web_root'] = config.get(section, "web_root") # Sickbeard webroot
self.Sickrage['ssl'] = config.getboolean(section, "ssl") # SSL
self.Sickrage['user'] = config.get(section, "username")
self.Sickrage['pass'] = config.get(section, "password")
#Read SAB section information
section = "SABNZBD"
self.SAB = {}
try:
self.SAB['convert'] = config.getboolean(section, "convert") # Convert
except:
self.SAB['convert'] = False
self.SAB['cp'] = config.get(section, "Couchpotato-category").lower()
self.SAB['sb'] = config.get(section, "Sickbeard-category").lower()
self.SAB['sr'] = config.get(section, "Sickrage-category").lower()
self.SAB['sonarr'] = config.get(section, "Sonarr-category").lower()
self.SAB['bypass'] = config.get(section, "Bypass-category").lower()
#Read Plex section information
section = "Plex"
self.Plex = {}
self.Plex['host'] = config.get(section, "host")
self.Plex['port'] = config.get(section, "port")
try:
self.Plex['refresh'] = config.getboolean(section, "refresh")
except:
self.Plex['refresh'] = False
self.Plex['token'] = config.get(section, "token")
if self.Plex['token'] == '':
self.Plex['token'] = None;
#Pass the values on
self.config = config
self.configFile = configFile
def __init__(self, directory, filename, logger=None):
# Setup logging
if logger:
log = logger
else:
log = logging.getLogger(__name__)
try:
from babelfish import Language
except:
log.debug("Trying to install SetupTools.")
try:
import setup
setup.ez_setup.main()
from babelfish import Language
except:
log.exception("Please install SetupTools.")
# Default settings for SickBeard
sb_defaults = {
"host": "localhost",
"port": "8081",
"ssl": "False",
"api_key": "",
"web_root": "",
"username": "",
"password": "",
}
# Default MP4 conversion settings
mp4_defaults = {
"ffmpeg": "ffmpeg.exe",
"ffprobe": "ffprobe.exe",
"output_directory": "",
"copy_to": "",
"move_to": "",
"output_extension": "mp4",
"output_format": "mp4",
"delete_original": "True",
"relocate_moov": "True",
"ios-audio": "True",
"ios-first-track-only": "False",
"max-audio-channels": "",
"audio-language": "",
"audio-default-language": "",
"audio-codec": "ac3",
"audio-channel-bitrate": "256",
"video-codec": "h264, x264",
"video-bitrate": "",
"video-max-width": "",
"subtitle-codec": "mov_text",
"subtitle-language": "",
"subtitle-default-language": "",
"convert-mp4": "False",
"fullpathguess": "True",
"tagfile": "True",
"tag-language": "en",
"download-artwork": "True",
"download-subs": "False",
"embed-subs": "True",
"sub-providers": "addic7ed, podnapisi, thesubdb, opensubtitles",
"permissions": "777",
}
# Default settings for CouchPotato
cp_defaults = {
"host": "localhost",
"port": "5050",
"username": "",
"password": "",
"apikey": "",
"delay": "65",
"method": "renamer",
"delete_failed": "False",
"ssl": "False",
"web_root": "",
}
# Default settings for Sonarr
sonarr_defaults = {"host": "localhost", "port": "8989", "apikey": "", "ssl": "False", "web_root": ""}
# Default uTorrent settings
utorrent_defaults = {
"couchpotato-label": "couchpotato",
"sickbeard-label": "sickbeard",
"sickrage-label": "sickrage",
"sonarr-label": "sonarr",
"bypass-label": "bypass",
"convert": "True",
"webui": "False",
"action_before": "stop",
"action_after": "removedata",
"host": "http://localhost:8080/",
"username": "",
"password": "",
}
# Default SAB settings
sab_defaults = {
"convert": "True",
"Sickbeard-category": "sickbeard",
"Sickrage-category": "sickrage",
"Couchpotato-category": "couchpotato",
"Sonarr-category": "sonarr",
"Bypass-category": "bypass",
}
# Default Sickrage Settings
sr_defaults = {
"host": "localhost",
"port": "8081",
"ssl": "False",
"api_key": "",
"web_root": "",
"username": "",
"password": "",
}
# Default deluge settings
deluge_defaults = {
"couchpotato-label": "couchpotato",
"sickbeard-label": "sickbeard",
"sickrage-label": "sickrage",
"sonarr-label": "sonarr",
"bypass-label": "bypass",
"convert": "True",
"host": "localhost",
"port": "58846",
"username": "",
"password": "",
}
# Default Plex Settings
plex_defaults = {"host": "localhost", "port": "32400", "refresh": "true"}
defaults = {
"SickBeard": sb_defaults,
"CouchPotato": cp_defaults,
"Sonarr": sonarr_defaults,
"MP4": mp4_defaults,
"uTorrent": utorrent_defaults,
"SABNZBD": sab_defaults,
"Sickrage": sr_defaults,
"Deluge": deluge_defaults,
"Plex": plex_defaults,
}
write = False # Will be changed to true if a value is missing from the config file and needs to be written
config = configparser.SafeConfigParser()
configFile = os.path.join(directory, filename)
if os.path.isfile(configFile):
config.read(configFile)
else:
log.error("Config file not found, creating %s." % configFile)
# config.filename = filename
write = True
# Make sure all sections and all keys for each section are present
for s in defaults:
if not config.has_section(s):
config.add_section(s)
write = True
for k in defaults[s]:
if not config.has_option(s, k):
config.set(s, k, defaults[s][k])
write = True
# If any keys are missing from the config file, write them
if write:
self.writeConfig(config, configFile)
# Read relevant MP4 section information
section = "MP4"
self.ffmpeg = os.path.normpath(self.raw(config.get(section, "ffmpeg"))) # Location of FFMPEG.exe
self.ffprobe = os.path.normpath(self.raw(config.get(section, "ffprobe"))) # Location of FFPROBE.exe
self.output_dir = config.get(section, "output_directory")
if self.output_dir == "":
self.output_dir = None
else:
self.output_dir = os.path.normpath(self.raw(self.output_dir)) # Output directory
self.copyto = config.get(section, "copy_to") # Directories to make copies of the final product
if self.copyto == "":
self.copyto = None
else:
self.copyto = self.copyto.split("|")
for i in xrange(len(self.copyto)):
self.copyto[i] = os.path.normpath(self.copyto[i])
if not os.path.isdir(self.copyto[i]):
try:
os.makedirs(self.copyto[i])
except:
log.exception("Error making directory %s." % (self.copyto[i]))
self.moveto = config.get(section, "move_to") # Directory to move final product to
if self.moveto == "":
self.moveto = None
else:
self.moveto = os.path.normpath(self.moveto)
if not os.path.isdir(self.moveto):
try:
os.makedirs(self.moveto)
except:
log.exception("Error making directory %s." % (self.moveto))
self.moveto = None
self.output_extension = config.get(section, "output_extension") # Output extension
self.output_format = config.get(section, "output_format") # Output format
if self.output_format not in valid_formats:
self.output_format = "mov"
self.delete = config.getboolean(section, "delete_original") # Delete original file
self.relocate_moov = config.getboolean(section, "relocate_moov") # Relocate MOOV atom to start of file
self.acodec = config.get(
section, "audio-codec"
).lower() # Gets the desired audio codec, if no valid codec selected, default to AC3
if self.acodec == "":
self.acodec == ["ac3"]
else:
self.acodec = self.acodec.lower().replace(" ", "").split(",")
self.abitrate = config.get(section, "audio-channel-bitrate")
try:
self.abitrate = int(self.abitrate)
except:
self.abitrate = 256
log.warning("Audio bitrate was invalid, defaulting to 256 per channel.")
if self.abitrate > 256:
log.warning("Audio bitrates >256 may create errors with common codecs.")
# !!! Leaving this disabled for now, users will be responsible for knowing whicn codecs do and don't work with mp4 files !!!
# if self.acodec not in valid_audio_codecs:
# self.acodec = 'aac'
# print "Audio codec not valid, defaulting to AAC"
self.iOS = config.get(
section, "ios-audio"
) # Creates a second audio channel if the standard output methods are different from this for iOS compatability
if self.iOS == "" or self.iOS.lower() in ["false", "no", "f", "0"]:
self.iOS = False
else:
if self.iOS.lower() in ["true", "yes", "t", "1"]:
self.iOS = "aac"
self.iOSFirst = config.getboolean(
section, "ios-first-track-only"
) # Enables the iOS audio option only for the first track
self.downloadsubs = config.getboolean(
section, "download-subs"
) # Enables downloading of subtitles from the internet sources using subliminal
if self.downloadsubs:
try:
import subliminal
except Exception as e:
self.downloadsubs = False
log.exception("Subliminal is not installed, automatically downloading of subs has been disabled.")
self.subproviders = config.get(section, "sub-providers").lower()
if self.subproviders == "":
self.downloadsubs = False
log.warning(
"You must specifiy at least one subtitle provider to downlaod subs automatically, subtitle downloading disabled."
)
else:
self.subproviders = self.subproviders.lower().replace(" ", "").split(",")
self.embedsubs = config.getboolean(section, "embed-subs")
self.permissions = config.get(section, "permissions")
try:
self.permissions = int(self.permissions, 8)
except:
self.log.exception("Invalid permissions, defaulting to 777.")
self.permissions = int("0777", 8)
# Setup variable for maximum audio channels
self.maxchannels = config.get(section, "max-audio-channels")
if self.maxchannels == "":
self.maxchannels = None
else:
try:
self.maxchannels = int(self.maxchannels)
except:
log.exception("Invalid number of audio channels specified.")
self.maxchannels = None
if self.maxchannels is not None and self.maxchannels < 1:
log.warning("Must have at least 1 audio channel.")
self.maxchannels = None
self.vcodec = config.get(section, "video-codec")
if self.vcodec == "":
self.vcodec == ["h264", "x264"]
else:
self.vcodec = self.vcodec.lower().replace(" ", "").split(",")
self.vbitrate = config.get(section, "video-bitrate")
if self.vbitrate == "":
self.vbitrate = None
else:
try:
self.vbitrate = int(self.vbitrate)
except:
log.exception("Invalid video bitrate, defaulting to no video bitrate cap.")
self.vbitrate = None
self.vwidth = config.get(section, "video-max-width")
if self.vwidth == "":
self.vwidth = None
else:
try:
self.vwidth = int(self.vwidth)
except:
log.exception("Invalid video width, defaulting to none.")
self.vwidth = None
self.awl = (
config.get(section, "audio-language").strip().lower()
) # List of acceptable languages for audio streams to be carried over from the original file, separated by a comma. Blank for all
if self.awl == "":
self.awl = None
else:
self.awl = self.awl.replace(" ", "").split(",")
self.scodec = config.get(section, "subtitle-codec").strip().lower()
if not self.scodec or self.scodec == "":
if self.embedsubs:
self.scodec = "mov_text"
else:
self.scodec = "srt"
log.warning("Invalid subtitle codec, defaulting to '%s'." % self.scodec)
if self.embedsubs and self.scodec not in valid_internal_subcodecs:
log.warning("Invalid interal subtitle codec %s, defaulting to 'mov_text'." % self.scodec)
self.scodec = "mov_text"
if not self.embedsubs and self.scodec not in valid_external_subcodecs:
log.warning("Invalid external subtitle codec %s, defaulting to 'srt'." % self.scodec)
self.scodec = "srt"
self.swl = (
config.get(section, "subtitle-language").strip().lower()
) # List of acceptable languages for subtitle streams to be carried over from the original file, separated by a comma. Blank for all
if self.swl == "":
self.swl = None
else:
self.swl = self.swl.replace(" ", "").split(",")
self.adl = (
config.get(section, "audio-default-language").strip().lower()
) # What language to default an undefinied audio language tag to. If blank, it will remain undefined. This is useful for single language releases which tend to leave things tagged as und
if self.adl == "" or len(self.adl) > 3:
self.adl = None
self.sdl = (
config.get(section, "subtitle-default-language").strip().lower()
) # What language to default an undefinied subtitle language tag to. If blank, it will remain undefined. This is useful for single language releases which tend to leave things tagged as und
if self.sdl == "" or len(self.sdl) > 3:
self.sdl = None
# Prevent incompatible combination of settings
if self.output_dir == "" and self.delete is False:
log.error("You must specify an alternate output directory if you aren't going to delete the original file.")
sys.exit()
# Create output directory if it does not exist
if self.output_dir is not None:
if not os.path.isdir(self.output_dir):
os.makedirs(self.output_dir)
self.processMP4 = config.getboolean(
section, "convert-mp4"
) # Determine whether or not to reprocess mp4 files or just tag them
self.fullpathguess = config.getboolean(section, "fullpathguess") # Guess using the full path or not
self.tagfile = config.getboolean(section, "tagfile") # Tag files with metadata
self.taglanguage = config.get(section, "tag-language").strip().lower() # Language to tag files
if len(self.taglanguage) > 2:
try:
babel = Language.fromalpha3(self.taglanguage)
self.taglanguage = babel.alpha2
except:
log.exception("Unable to set tag language, defaulting to English.")
self.taglanguage = "en"
elif len(self.taglanguage) < 2:
log.exception("Unable to set tag language, defaulting to English.")
self.taglanguage = "en"
self.artwork = config.getboolean(section, "download-artwork") # Download and embed artwork
# Read relevant CouchPotato section information
section = "CouchPotato"
self.CP = {}
self.CP["host"] = config.get(section, "host")
self.CP["port"] = config.get(section, "port")
self.CP["username"] = config.get(section, "username")
self.CP["password"] = config.get(section, "password")
self.CP["apikey"] = config.get(section, "apikey")
self.CP["delay"] = config.get(section, "delay")
self.CP["method"] = config.get(section, "method")
self.CP["web_root"] = config.get(section, "web_root")
try:
self.CP["delay"] = float(self.CP["delay"])
except ValueError:
self.CP["delay"] = 60
try:
self.CP["delete_failed"] = config.getboolean(section, "delete_failed")
except (configparser.NoOptionError, ValueError):
self.CP["delete_failed"] = False
try:
if config.getboolean(section, "ssl"):
self.CP["protocol"] = "https://"
else:
self.CP["protocol"] = "http://"
except (configparser.NoOptionError, ValueError):
self.CP["protocol"] = "http://"
# Read relevant uTorrent section information
section = "uTorrent"
self.uTorrent = {}
self.uTorrent["cp"] = config.get(section, "couchpotato-label").lower()
self.uTorrent["sb"] = config.get(section, "sickbeard-label").lower()
self.uTorrent["sr"] = config.get(section, "sickrage-label").lower()
self.uTorrent["sonarr"] = config.get(section, "sonarr-label").lower()
self.uTorrent["bypass"] = config.get(section, "bypass-label").lower()
try:
self.uTorrent["convert"] = config.getboolean(section, "convert")
except:
self.uTorrent["convert"] = False
self.uTorrentWebUI = config.getboolean(section, "webui")
self.uTorrentActionBefore = config.get(section, "action_before").lower()
self.uTorrentActionAfter = config.get(section, "action_after").lower()
self.uTorrentHost = config.get(section, "host").lower()
self.uTorrentUsername = config.get(section, "username")
self.uTorrentPassword = config.get(section, "password")
# Read relevant Deluge section information
section = "Deluge"
self.deluge = {}
self.deluge["cp"] = config.get(section, "couchpotato-label").lower()
self.deluge["sb"] = config.get(section, "sickbeard-label").lower()
self.deluge["sr"] = config.get(section, "sickrage-label").lower()
self.deluge["sonarr"] = config.get(section, "sonarr-label").lower()
self.deluge["bypass"] = config.get(section, "bypass-label").lower()
try:
self.deluge["convert"] = config.getboolean(section, "convert")
except:
self.deluge["convert"] = False
self.deluge["host"] = config.get(section, "host").lower()
self.deluge["port"] = config.get(section, "port")
self.deluge["user"] = config.get(section, "username")
self.deluge["pass"] = config.get(section, "password")
# Read relevant Sonarr section information
section = "Sonarr"
self.Sonarr = {}
self.Sonarr["host"] = config.get(section, "host")
self.Sonarr["port"] = config.get(section, "port")
self.Sonarr["apikey"] = config.get(section, "apikey")
self.Sonarr["ssl"] = config.get(section, "ssl")
self.Sonarr["web_root"] = config.get(section, "web_root")
# Read Sickbeard section information
section = "SickBeard"
self.Sickbeard = {}
self.Sickbeard["host"] = config.get(section, "host") # Server Address
self.Sickbeard["port"] = config.get(section, "port") # Server Port
self.Sickbeard["api_key"] = config.get(section, "api_key") # Sickbeard API key
self.Sickbeard["web_root"] = config.get(section, "web_root") # Sickbeard webroot
self.Sickbeard["ssl"] = config.getboolean(section, "ssl") # SSL
self.Sickbeard["user"] = config.get(section, "username")
self.Sickbeard["pass"] = config.get(section, "password")
# Read Sickrage section information
section = "Sickrage"
self.Sickrage = {}
self.Sickrage["host"] = config.get(section, "host") # Server Address
self.Sickrage["port"] = config.get(section, "port") # Server Port
self.Sickrage["api_key"] = config.get(section, "api_key") # Sickbeard API key
self.Sickrage["web_root"] = config.get(section, "web_root") # Sickbeard webroot
self.Sickrage["ssl"] = config.getboolean(section, "ssl") # SSL
self.Sickrage["user"] = config.get(section, "username")
self.Sickrage["pass"] = config.get(section, "password")
# Read SAB section information
section = "SABNZBD"
self.SAB = {}
try:
self.SAB["convert"] = config.getboolean(section, "convert") # Convert
except:
self.SAB["convert"] = False
self.SAB["cp"] = config.get(section, "Couchpotato-category").lower()
self.SAB["sb"] = config.get(section, "Sickbeard-category").lower()
self.SAB["sr"] = config.get(section, "Sickrage-category").lower()
self.SAB["sonarr"] = config.get(section, "Sonarr-category").lower()
self.SAB["bypass"] = config.get(section, "Bypass-category").lower()
# Read Plex section information
section = "Plex"
self.Plex = {}
self.Plex["host"] = config.get(section, "host")
self.Plex["port"] = config.get(section, "port")
try:
self.Plex["refresh"] = config.getboolean(section, "refresh")
except:
self.Plex["refresh"] = False
# Pass the values on
self.config = config
self.configFile = configFile
