diff --git a/service.py b/service.py index 56b521d..19aea1f 100755 --- a/service.py +++ b/service.py @@ -5,6 +5,7 @@ import requests import tomlkit import urllib3 import datetime +import sys urllib3.disable_warnings() class StatusPage: @@ -115,7 +116,10 @@ class StatusPage: 'username': 'root', 'password': 'foobar' } - self.config.update(tomlkit.loads(open('config.toml').read())) + config_file = "config.toml" + if len(sys.argv) > 1: + config_file = sys.argv[1] + self.config.update(tomlkit.loads(open(config_file).read())) if __name__ == "__main__": page = StatusPage()