Added config

This commit is contained in:
Ruakij 2021-08-02 11:40:44 +02:00
parent bbeac78d26
commit 3df89dce43
2 changed files with 22 additions and 3 deletions

4
config.yml Normal file
View File

@ -0,0 +1,4 @@
socket:
address: 0.0.0.0
port: 53

View File

@ -1,6 +1,7 @@
#!/usr/bin/python3
import os, subprocess, socket, datetime, getopt, sys, re, atexit, logging as log
import os, subprocess, socket, datetime, getopt, sys, re, atexit, yaml, logging as log
from pathlib import Path
import dns.flags
import dns.message
@ -15,7 +16,9 @@ def main(args):
setupLogging(True)
log.debug("Logging started")
s = setupSocket('172.17.0.1', 53)
config = readConfig("config.yml")
s = setupSocket(config['socket']['address'], config['socket']['port'])
startListen(s)
@ -25,7 +28,7 @@ def setupSocket(address, port):
return s
def startListen(s):
log.debug('Now listening')
log.debug(f'Now listening')
while True:
(address, dmsg) = receiveFromWire(s)
@ -98,5 +101,17 @@ def setupLogging(verbose: bool):
log.basicConfig(stream=sys.stdout, format=format, level=level)
def readConfig(file: str):
log.debug(f"Reading config '{file}'..")
if not os.path.isfile(file):
raise OSError(2, file)
return readYamlFile(file)
def readYamlFile(file: str):
with open(file, "r") as f:
return yaml.load(f, Loader=yaml.FullLoader)
if __name__ == "__main__":
sys.exit(main(sys.argv))