Update update.py

pull/5/head
RandomNinjaAtk 5 years ago committed by GitHub
parent fe31a9a9d6
commit c9185eebb4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -6,35 +6,9 @@ import logging
import configparser import configparser
import xml.etree.ElementTree as ET 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") autoProcess = os.path.join(os.environ.get("SMA_PATH", "/usr/local/sma"), "config/autoProcess.ini")
def main(): 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 = configparser.ConfigParser()
safeConfigParser.read(autoProcess) safeConfigParser.read(autoProcess)
@ -177,19 +151,6 @@ def main():
if os.environ.get("PLEX_TOKEN"): if os.environ.get("PLEX_TOKEN"):
safeConfigParser.set("Plex", "token", 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") fp = open(autoProcess, "w")
safeConfigParser.write(fp) safeConfigParser.write(fp)
fp.close() fp.close()

Loading…
Cancel
Save