Skip to content

Commit

Permalink
Fix linter errors
Browse files Browse the repository at this point in the history
  • Loading branch information
anchal00 committed Sep 5, 2024
1 parent 90108fa commit 32ab5aa
Show file tree
Hide file tree
Showing 8 changed files with 21 additions and 26 deletions.
3 changes: 2 additions & 1 deletion optimus/dns/models/packet.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,8 @@
from enum import Enum
from typing import List, Optional, Union

from optimus.dns.models.records import AAAA, NS, A, SOA, Record, RecordClass, RecordType
from optimus.dns.models.records import (AAAA, NS, SOA, A, Record, RecordClass,
RecordType)


class ResponseCode(Enum): # 4 bits
Expand Down
18 changes: 5 additions & 13 deletions optimus/dns/parser/parse.py
Original file line number Diff line number Diff line change
@@ -1,20 +1,12 @@
from ipaddress import IPv4Address, IPv6Address
from typing import List, Union

from optimus.dns.models.packet import (DNSHeader, DNSPacket, Question,
ResponseCode)
from optimus.dns.models.records import (AAAA, CNAME, MX, NS, SOA, A,
OptPseudoRR, Record, RecordClass,
RecordType)
from optimus.dns.parser.iter import BytearrayIterator
from optimus.dns.models.packet import DNSHeader, DNSPacket, Question, ResponseCode
from optimus.dns.models.records import (
AAAA,
CNAME,
MX,
NS,
SOA,
A,
OptPseudoRR,
Record,
RecordClass,
RecordType,
)


class DNSParser:
Expand Down
6 changes: 4 additions & 2 deletions optimus/dns/resolver.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,11 @@
import random
from typing import List, Union

from optimus.dns.models.packet import DNSHeader, DNSPacket, Question, ResponseCode
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.dns.models.records import AAAA, NS, A, Record, RecordClass, RecordType
from optimus.networking.udp import query_server_over_udp
from optimus.optimus_server.context import get_root_servers

Expand Down
4 changes: 2 additions & 2 deletions optimus/networking/udp.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
from typing import Optional
from optimus.networking.cache import socket_cache
import socket
from typing import Optional

from optimus.logging.logger import log_error
from optimus.networking.cache import socket_cache


def query_server_over_udp(bin_data: bytearray, server_addr: str) -> bytes:
Expand Down
1 change: 0 additions & 1 deletion optimus/prometheus.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,6 @@

from optimus.logging.logger import log


inbound_rqc = Counter("inbound_requests", "Total Requests Received")
served_rqc = Counter("served_requests", "Total Requests Processed")
erred_rqc = Counter("erred_requests", "Total Requests Failed")
Expand Down
5 changes: 2 additions & 3 deletions optimus/server/context.py
Original file line number Diff line number Diff line change
@@ -1,10 +1,9 @@
import json
import os
import posixpath
from typing import List
import pathlib
import posixpath
import socket

from typing import List

__NAMESERVERS: List[str] = []

Expand Down
3 changes: 2 additions & 1 deletion optimus/server/router.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,9 @@
import socket

from optimus.dns.models.packet import DNSPacket, ResponseCode
from optimus.dns.parser.parse import DNSParser
from optimus.logging.logger import log
from optimus.dns.resolver import resolve
from optimus.logging.logger import log
from optimus.prometheus import record_metrics


Expand Down
7 changes: 4 additions & 3 deletions optimus/server/udp_listener.py
Original file line number Diff line number Diff line change
@@ -1,10 +1,11 @@
import socket
from concurrent import futures
from optimus.server.context import warmup_cache
from optimus.prometheus import with_prometheus_monitoring
from optimus.server.router import handle_request

from optimus.logging.logger import log
from optimus.networking.cache import socket_cache
from optimus.prometheus import with_prometheus_monitoring
from optimus.server.context import warmup_cache
from optimus.server.router import handle_request


@with_prometheus_monitoring
Expand Down

0 comments on commit 32ab5aa

Please sign in to comment.