28 lines
558 B
Python
28 lines
558 B
Python
|
import logging
|
||
|
from util import run_op
|
||
|
|
||
|
class UfwInterface:
|
||
|
|
||
|
def __init__(self):
|
||
|
pass
|
||
|
|
||
|
def enable(self):
|
||
|
op = "ufw enable"
|
||
|
logging.info(f"Enabling ufw: {op}")
|
||
|
run_op(op)
|
||
|
|
||
|
def allow_app(self, app_name: str):
|
||
|
op = f"ufw allow {app_name}"
|
||
|
logging.info(f"Allowing ufw app: {op}")
|
||
|
run_op(op)
|
||
|
|
||
|
class Firewall:
|
||
|
|
||
|
def __init__(self):
|
||
|
self.ufw = UfwInterface()
|
||
|
|
||
|
def allow_app(self, app_name: str):
|
||
|
self.ufw.allow_app(app_name)
|
||
|
|
||
|
def enable(self):
|
||
|
self.ufw.enable()
|