diff --git a/backend/src/api/nftables_api.py b/backend/src/api/nftables_api.py index db5efe6..5f85970 100644 --- a/backend/src/api/nftables_api.py +++ b/backend/src/api/nftables_api.py @@ -18,7 +18,7 @@ import uuid from enum import Enum # pyroute2 nftables -from pyroute2 import nftables +from pyroute2.nftables import NFTables router = APIRouter() logger = logging.getLogger("nftables") @@ -89,7 +89,7 @@ class ReplaceResult(BaseModel): # ---------------------- nft wrapper ---------------------- class NFT: def __init__(self): - self.nft = nftables.main.NFTables() + self.nft = NFTables() def run(self, cmd: str) -> Dict[str, Any]: """