def apply(request):
# do generate
configfile = open(settings.AFS_CONFIG, "r")
options = yaml.load(configfile)
configfile.close()
network = {}
for host in Host.objects.filter(enabled=True):
network[str(host.entry)] = {"ip": host.ip, "mac": str(host.mac), "host": str(host.host), "radius": host.wifi}
output = open(options["network"], "w")
yaml.dump(network, output, default_flow_style=False)
output.close()
args = Object(object)
args.config = settings.AFS_CONFIG
args.noreload = False
parse = parse_config(args)
config = Config(parse["options"], parse["network"], parse["doreload"])
try:
config.validate()
generate(config)
messages.success(request, "Success")
except Error as e:
messages.error(request, e.msg)
return redirect("/admin/network/host/")
def apply(request):
#do generate
configfile = open(settings.AFS_CONFIG, 'r')
options = yaml.load(configfile)
configfile.close()
network = {}
for host in Host.objects.filter(enabled=True):
network[str(host.entry)] = { 'ip': host.ip, 'mac': str(host.mac), 'host': str(host.host), 'radius': host.wifi }
output = open(options['network'],'w')
yaml.dump(network, output, default_flow_style=False)
output.close()
args = Object(object)
args.config = settings.AFS_CONFIG
args.noreload = False
parse = parse_config(args)
config = Config(parse['options'], parse['network'], parse['doreload'])
try:
config.validate()
generate(config)
messages.success(request, "Success")
except Error as e:
messages.error(request, e.msg)
return redirect("/admin/network/host/")
def main():
parser = argparse.ArgumentParser(description='Automated Firewall System')
parser.add_argument('command', choices=[ 'generate', 'check' ], help='what operation to perform (choices: %(choices)s)', metavar="command")
parser.add_argument('-c', '--config', default='/etc/afs/config.yml', help='use alternate config file (default: %(default)s)')
parser.add_argument('-n', '--noreload', action="store_const", const=True, help="don't reload services")
parser.add_argument('-v', '--verbose', action="store_const", const=True, help="verbose mode")
args = parser.parse_args()
try:
if args.verbose == True:
print "Parsing configuration"
parse = parse_config(args)
except Error as e:
sys.exit(e.msg)
config = Config(parse['options'], parse['network'], parse['doreload'])
try:
if args.verbose == True:
print "Validating configuration"
config.validate()
except Error as e:
sys.exit(e.msg)
if args.command == 'check':
print "Syntax OK"
sys.exit(0)
if args.command == 'generate':
try:
if args.verbose == True:
print "Starint config generation"
generate(config, args.verbose)
if args.verbose == True:
print "Configuration has been generated"
except Error as e:
sys.exit(e.msg)
