firewall/main.py

50 lines
1.2 KiB
Python
Raw Normal View History

2019-04-29 22:16:32 +00:00
#! /usr/bin/python3
import os
import logging
from logging.handlers import RotatingFileHandler
import click
import nat as _nat
import mac_ip as _mac_ip
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
LOG_LEVEL = logging.INFO
logger = logging.getLogger()
logger.setLevel(LOG_LEVEL)
formatter = logging.Formatter('%(levelname)s :: %(message)s')
file_handler = RotatingFileHandler('/var/log/firewall.log', 'a', 1000000, 1)
file_handler.setLevel(LOG_LEVEL)
file_handler.setFormatter(formatter)
logger.addHandler(file_handler)
stream_handler = logging.StreamHandler()
stream_handler.setFormatter(formatter)
stream_handler.setLevel(LOG_LEVEL)
logger.addHandler(stream_handler)
@click.group(invoke_without_command=True)
@click.pass_context
def cli(ctx):
logger.info("Starting Re2o firewall manager.")
if ctx.invoked_subcommand is None:
logger.info("Loading firewall.")
os.system('nft -I {install_dir} -f {firewall}'.format(
install_dir=BASE_DIR,
firewall=os.path.join(BASE_DIR, 'firewall.nft')
))
_mac_ip.update_macip()
_nat.main()
@cli.command()
def macip():
_mac_ip.update_macip()
@cli.command()
def nat():
_nat.main()
if __name__ == '__main__':
cli()