RandomNinjaAtk
5 years ago
committed by
GitHub
parent
f87507b818
commit
fe31a9a9d6
@ -0,0 +1,199 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
import os
|
||||
import sys
|
||||
import logging
|
||||
import configparser
|
||||
import xml.etree.ElementTree as ET
|
||||
|
||||
xml = "/config/config.xml"
|
||||
autoProcess = os.path.join(os.environ.get("SMA_PATH", "/usr/local/sma"), "config/autoProcess.ini")
|
||||
|
||||
|
||||
def main():
|
||||
if not os.path.isfile(xml):
|
||||
logging.error("No Sonarr/Radarr config file found")
|
||||
sys.exit(1)
|
||||
|
||||
if not os.path.isfile(autoProcess):
|
||||
logging.error("autoProcess.ini does not exist")
|
||||
sys.exit(1)
|
||||
|
||||
tree = ET.parse(xml)
|
||||
root = tree.getroot()
|
||||
port = root.find("Port").text
|
||||
try:
|
||||
sslport = root.find("SslPort").text
|
||||
except:
|
||||
sslport = port
|
||||
webroot = root.find("UrlBase").text
|
||||
webroot = webroot if webroot else ""
|
||||
ssl = root.find("EnableSsl").text
|
||||
ssl = ssl.lower() in ["true", "yes", "t", "1", "y"] if ssl else False
|
||||
apikey = root.find("ApiKey").text
|
||||
section = os.environ.get("SMA_APP")
|
||||
if not section:
|
||||
logging.error("No Sonarr/Radarr specifying ENV variable")
|
||||
sys.exit(1)
|
||||
|
||||
safeConfigParser = configparser.ConfigParser()
|
||||
safeConfigParser.read(autoProcess)
|
||||
|
||||
# Set FFMPEG/FFProbe Paths
|
||||
safeConfigParser.set("Converter", "ffmpeg", "ffmpeg")
|
||||
safeConfigParser.set("Converter", "ffprobe", "ffprobe")
|
||||
if os.environ.get("CONVERTER_THREADS"):
|
||||
safeConfigParser.set("Converter", "threads", os.environ.get("CONVERTER_THREADS"))
|
||||
if os.environ.get("CONVERTER_OUTPUT_FORMAT"):
|
||||
safeConfigParser.set("Converter", "output-format", os.environ.get("CONVERTER_OUTPUT_FORMAT"))
|
||||
if os.environ.get("CONVERTER_OUTPUT_EXTENSION"):
|
||||
safeConfigParser.set("Converter", "output-extension", os.environ.get("CONVERTER_OUTPUT_EXTENSION"))
|
||||
if os.environ.get("CONVERTER_SORT_STREAMS"):
|
||||
safeConfigParser.set("Converter", "sort-streams", os.environ.get("CONVERTER_SORT_STREAMS"))
|
||||
if os.environ.get("CONVERTER_PROCESS_SAME_EXTENSIONS"):
|
||||
safeConfigParser.set("Converter", "process-same-extensions", os.environ.get("CONVERTER_PROCESS_SAME_EXTENSIONS"))
|
||||
if os.environ.get("CONVERTER_FORCE_CONVERT"):
|
||||
safeConfigParser.set("Converter", "force-convert", os.environ.get("CONVERTER_FORCE_CONVERT"))
|
||||
if os.environ.get("CONVERTER_PREOPTS"):
|
||||
safeConfigParser.set("Converter", "preopts", os.environ.get("CONVERTER_PREOPTS"))
|
||||
if os.environ.get("CONVERTER_POSTOPTS"):
|
||||
safeConfigParser.set("Converter", "postopts", os.environ.get("CONVERTER_POSTOPTS"))
|
||||
|
||||
# SET Permissions
|
||||
if os.environ.get("PERMISSIONS_CHMOD"):
|
||||
safeConfigParser.set("Permissions", "chmod", os.environ.get("PERMISSIONS_CHMOD"))
|
||||
|
||||
# Set Metadata Settings
|
||||
if os.environ.get("METADATA_RELOCATE_MOV"):
|
||||
safeConfigParser.set("Metadata", "relocate-moov", os.environ.get("METADATA_RELOCATE_MOV"))
|
||||
if os.environ.get("METADATA_TAG"):
|
||||
safeConfigParser.set("Metadata", "tag", os.environ.get("METADATA_TAG"))
|
||||
if os.environ.get("METADATA_TAG_LANGUAGE"):
|
||||
safeConfigParser.set("Metadata", "tag-language", os.environ.get("METADATA_TAG_LANGUAGE"))
|
||||
if os.environ.get("METADATA_DOWNLOAD_ARTWORK"):
|
||||
safeConfigParser.set("Metadata", "download-artwork", os.environ.get("METADATA_DOWNLOAD_ARTWORK"))
|
||||
if os.environ.get("METADATA_PRESERVE_SOURCE_DISPOSITION"):
|
||||
safeConfigParser.set("Metadata", "preserve-source-disposition", os.environ.get("METADATA_PRESERVE_SOURCE_DISPOSITION"))
|
||||
|
||||
# Set Video Settings
|
||||
if os.environ.get("VIDEO_CODEC"):
|
||||
safeConfigParser.set("Video", "codec", os.environ.get("VIDEO_CODEC"))
|
||||
if os.environ.get("VIDEO_BITRATE"):
|
||||
safeConfigParser.set("Video", "bitrate", os.environ.get("VIDEO_BITRATE"))
|
||||
if os.environ.get("VIDEO_CRF"):
|
||||
safeConfigParser.set("Video", "crf", os.environ.get("VIDEO_CRF"))
|
||||
if os.environ.get("VIDEO_CRF_PROFILES"):
|
||||
safeConfigParser.set("Video", "crf-profiles", os.environ.get("VIDEO_CRF_PROFILES"))
|
||||
if os.environ.get("VIDEO_MAX_WIDTH"):
|
||||
safeConfigParser.set("Video", "max-width", os.environ.get("VIDEO_MAX_WIDTH"))
|
||||
if os.environ.get("VIDEO_PROFILE"):
|
||||
safeConfigParser.set("Video", "profile", os.environ.get("VIDEO_PROFILE"))
|
||||
if os.environ.get("VIDEO_MAX_LEVEL"):
|
||||
safeConfigParser.set("Video", "max-level", os.environ.get("VIDEO_MAX_LEVEL"))
|
||||
if os.environ.get("VIDEO_PIX_FMT"):
|
||||
safeConfigParser.set("Video", "pix-fmt", os.environ.get("VIDEO_PIX_FMT"))
|
||||
|
||||
# Set Audio Settings
|
||||
if os.environ.get("AUDIO_CODEC"):
|
||||
safeConfigParser.set("Audio", "codec", os.environ.get("AUDIO_CODEC"))
|
||||
if os.environ.get("AUDIO_LANGUAGES"):
|
||||
safeConfigParser.set("Audio", "languages", os.environ.get("AUDIO_LANGUAGES"))
|
||||
if os.environ.get("AUDIO_DEFAULT_LANGUAGE"):
|
||||
safeConfigParser.set("Audio", "default-language", os.environ.get("AUDIO_DEFAULT_LANGUAGE"))
|
||||
if os.environ.get("AUDIO_FIRST_STREAM_OF_LANGUAGE"):
|
||||
safeConfigParser.set("Audio", "first-stream-of-language", os.environ.get("AUDIO_FIRST_STREAM_OF_LANGUAGE"))
|
||||
if os.environ.get("AUDIO_CHANNEL_BITRATE"):
|
||||
safeConfigParser.set("Audio", "channel-bitrate", os.environ.get("AUDIO_CHANNEL_BITRATE"))
|
||||
if os.environ.get("AUDIO_MAX_BITRATE"):
|
||||
safeConfigParser.set("Audio", "max-bitrate", os.environ.get("AUDIO_MAX_BITRATE"))
|
||||
if os.environ.get("AUDIO_MAX_CHANNELS"):
|
||||
safeConfigParser.set("Audio", "max-channels", os.environ.get("AUDIO_MAX_CHANNELS"))
|
||||
if os.environ.get("AUDIO_PREFER_MORE_CHANNELS"):
|
||||
safeConfigParser.set("Audio", "prefer-more-channels", os.environ.get("AUDIO_PREFER_MORE_CHANNELS"))
|
||||
if os.environ.get("AUDIO_DEFAULT_MORE_CHANNELS"):
|
||||
safeConfigParser.set("Audio", "default-more-channels", os.environ.get("AUDIO_DEFAULT_MORE_CHANNELS"))
|
||||
if os.environ.get("AUDIO_FILTER"):
|
||||
safeConfigParser.set("Audio", "filter", os.environ.get("AUDIO_FILTER"))
|
||||
if os.environ.get("AUDIO_SAMPLE_RATES"):
|
||||
safeConfigParser.set("Audio", "sample-rates", os.environ.get("AUDIO_SAMPLE_RATES"))
|
||||
if os.environ.get("AUDIO_COPY_ORIGINAL"):
|
||||
safeConfigParser.set("Audio", "copy-original", os.environ.get("AUDIO_COPY_ORIGINAL"))
|
||||
if os.environ.get("AUDIO_AAC_ADTSTOASC"):
|
||||
safeConfigParser.set("Audio", "aac-adtstoasc", os.environ.get("AUDIO_AAC_ADTSTOASC"))
|
||||
if os.environ.get("AUDIO_IGNORE_TREHD"):
|
||||
safeConfigParser.set("Audio", "ignore-truehd", os.environ.get("AUDIO_IGNORE_TREHD"))
|
||||
|
||||
# Set Universal Audio Settings
|
||||
if os.environ.get("UAUDIO_CODEC"):
|
||||
safeConfigParser.set("Universal Audio", "codec", os.environ.get("UAUDIO_CODEC"))
|
||||
if os.environ.get("UAUDIO_CHANNEL_BITRATE"):
|
||||
safeConfigParser.set("Universal Audio", "channel-bitrate", os.environ.get("UAUDIO_CHANNEL_BITRATE"))
|
||||
if os.environ.get("UAUDIO_FIRST_STREAM_ONLY"):
|
||||
safeConfigParser.set("Universal Audio", "first-stream-only", os.environ.get("UAUDIO_FIRST_STREAM_ONLY"))
|
||||
if os.environ.get("UAUDIO_MOVE_AFTER"):
|
||||
safeConfigParser.set("Universal Audio", "move-after", os.environ.get("UAUDIO_MOVE_AFTER"))
|
||||
if os.environ.get("UAUDIO_FILTER"):
|
||||
safeConfigParser.set("Universal Audio", "filter", os.environ.get("UAUDIO_FILTER"))
|
||||
|
||||
# Set Subtitle Settings
|
||||
if os.environ.get("SUBTITLE_CODEC"):
|
||||
safeConfigParser.set("Subtitle", "codec", os.environ.get("SUBTITLE_CODEC"))
|
||||
if os.environ.get("SUBTITLE_CODEC_IMAGE_BASED"):
|
||||
safeConfigParser.set("Subtitle", "codec-image-based", os.environ.get("SUBTITLE_CODEC_IMAGE_BASED"))
|
||||
if os.environ.get("SUBTITLE_LANGUAGES"):
|
||||
safeConfigParser.set("Subtitle", "languages", os.environ.get("SUBTITLE_LANGUAGES"))
|
||||
if os.environ.get("SUBTITLE_DEFAULT_LANGUAGE"):
|
||||
safeConfigParser.set("Subtitle", "default-language", os.environ.get("SUBTITLE_DEFAULT_LANGUAGE"))
|
||||
if os.environ.get("SUBTITLE_FIRST_STREAM_OF_LANGUAGE"):
|
||||
safeConfigParser.set("Subtitle", "first-stream-of-language", os.environ.get("SUBTITLE_FIRST_STREAM_OF_LANGUAGE"))
|
||||
if os.environ.get("SUBTITLE_ENCODING"):
|
||||
safeConfigParser.set("Subtitle", "encoding", os.environ.get("SUBTITLE_ENCODING"))
|
||||
if os.environ.get("SUBTITLE_BURN_SUBTITLES"):
|
||||
safeConfigParser.set("Subtitle", "burn-subtitles", os.environ.get("SUBTITLE_BURN_SUBTITLES"))
|
||||
if os.environ.get("SUBTITLE_DOWNLOAD_SUBS"):
|
||||
safeConfigParser.set("Subtitle", "download-subs", os.environ.get("SUBTITLE_DOWNLOAD_SUBS"))
|
||||
if os.environ.get("SUBTITLE_DOWNLOAD_HEARING_IMPAIRED_SUBS"):
|
||||
safeConfigParser.set("Subtitle", "download-hearing-impaired-subs", os.environ.get("SUBTITLE_DOWNLOAD_HEARING_IMPAIRED_SUBS"))
|
||||
if os.environ.get("SUBTITLE_DOWNLOAD_PROVIDERS"):
|
||||
safeConfigParser.set("Subtitle", "download-providers", os.environ.get("SUBTITLE_DOWNLOAD_PROVIDERS"))
|
||||
if os.environ.get("SUBTITLE_EMBED_SUBS"):
|
||||
safeConfigParser.set("Subtitle", "embed-subs", os.environ.get("SUBTITLE_EMBED_SUBS"))
|
||||
if os.environ.get("SUBTITLE_EMBED_IMAGE_SUBS"):
|
||||
safeConfigParser.set("Subtitle", "embed-image-subs", os.environ.get("SUBTITLE_EMBED_IMAGE_SUBS"))
|
||||
if os.environ.get("SUBTITLE_EMBED_ONLY_INTERNAL_SUBS"):
|
||||
safeConfigParser.set("Subtitle", "embed-only-internal-subs", os.environ.get("SUBTITLE_EMBED_ONLY_INTERNAL_SUBS"))
|
||||
if os.environ.get("SUBTITLE_IGNORE_EMBEDDED_SUBS"):
|
||||
safeConfigParser.set("Subtitle", "ignore-embedded-subs", os.environ.get("SUBTITLE_IGNORE_EMBEDDED_SUBS"))
|
||||
if os.environ.get("SUBTITLE_ATTACHMENT_CODEC"):
|
||||
safeConfigParser.set("Subtitle", "attachment-codec", os.environ.get("SUBTITLE_ATTACHMENT_CODEC"))
|
||||
|
||||
# Set Plex Settings
|
||||
if os.environ.get("PLEX_HOST"):
|
||||
safeConfigParser.set("Plex", "host", os.environ.get("PLEX_HOST"))
|
||||
if os.environ.get("PLEX_PORT"):
|
||||
safeConfigParser.set("Plex", "port", os.environ.get("PLEX_PORT"))
|
||||
if os.environ.get("PLEX_REFRESH"):
|
||||
safeConfigParser.set("Plex", "refresh", os.environ.get("PLEX_REFRESH"))
|
||||
if os.environ.get("PLEX_TOKEN"):
|
||||
safeConfigParser.set("Plex", "token", os.environ.get("PLEX_TOKEN"))
|
||||
|
||||
# Set values from config.xml
|
||||
safeConfigParser.set(section, "apikey", apikey)
|
||||
safeConfigParser.set(section, "ssl", str(ssl))
|
||||
safeConfigParser.set(section, "port", sslport if ssl else port)
|
||||
safeConfigParser.set(section, "webroot", webroot)
|
||||
|
||||
# Set IP from environment variable
|
||||
ip = os.environ.get("HOST")
|
||||
if ip:
|
||||
safeConfigParser.set(section, "host", ip)
|
||||
else:
|
||||
safeConfigParser.set(section, "host", "127.0.0.1")
|
||||
|
||||
fp = open(autoProcess, "w")
|
||||
safeConfigParser.write(fp)
|
||||
fp.close()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Loading…
Reference in new issue