read config file name from command line

This commit is contained in:
root
2021-04-10 19:44:59 +02:00
parent fde7d60dc3
commit 83534572c2

View File

@@ -5,6 +5,7 @@ import requests
import tomlkit import tomlkit
import urllib3 import urllib3
import datetime import datetime
import sys
urllib3.disable_warnings() urllib3.disable_warnings()
class StatusPage: class StatusPage:
@@ -115,7 +116,10 @@ class StatusPage:
'username': 'root', 'username': 'root',
'password': 'foobar' '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__": if __name__ == "__main__":
page = StatusPage() page = StatusPage()