Change logging format

anchal00 · Sep 6, 2024 · 662a067 · 662a067
1 parent 80f9c68
commit 662a067
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/optimus/dns/resolver.py b/optimus/dns/resolver.py
@@ -2,13 +2,11 @@
 import random
 from typing import List, Union
 
-from optimus.dns.models.packet import (DNSHeader, DNSPacket, Question,
-                                       ResponseCode)
-from optimus.dns.models.records import (AAAA, NS, A, Record, RecordClass,
-                                        RecordType)
+from optimus.dns.models.packet import DNSHeader, DNSPacket, Question, ResponseCode
+from optimus.dns.models.records import AAAA, NS, A, Record, RecordClass, RecordType
 from optimus.dns.parser.parse import DNSParser
 from optimus.networking.udp import query_server_over_udp
-from optimus.optimus_server.context import get_root_servers
+from optimus.server.context import get_root_servers
 
 
 # TODO: Improve logging
@@ -17,6 +15,7 @@ def resolve(qpacket: DNSPacket) -> DNSPacket:
     server_addr: str = random.choice(get_root_servers())
     while True:
         _bytes: bytes = query_server_over_udp(qpacket.to_bin(), server_addr)
+        # TODO: Implement retries
         if not _bytes:
             return DNSPacket(
                 DNSHeader(

diff --git a/optimus/logging/logger.py b/optimus/logging/logger.py
@@ -1,6 +1,6 @@
 import logging
 
-logging.basicConfig(level=logging.INFO, format="%(asctime)s %(levelname)s %(message)s")
+logging.basicConfig(level=logging.INFO, format="ts=%(asctime)s level=%(levelname)s message=%(message)s")
 
 
 # TODO: Improve logging