interaction/infector/__init__.py

38 lines
898 B
Python

__author__ = 'RemiZOffAlex'
__email__ = 'remizoffalex@mail.ru'
import yaml
import npyscreen
from pyroute2 import NDB
ndb = NDB()
def network_config(ip, mask, gateway):
with ndb.interfaces['br0'] as br0:
address = '{ip}/{mask}'.format(ip=ip, mask=mask)
br0.add_ip(address).commit()
ndb.routes.create(dst='0.0.0.0/0', gateway=gateway).commit()
def TestApp(*args):
NetworkView = npyscreen.ActionForm(name = "SaaS",)
ip = NetworkView.add(npyscreen.TitleText, name = "IP:",)
mask = NetworkView.add(npyscreen.TitleText, name = "Netmask:",)
gateway = NetworkView.add(npyscreen.TitleText, name = "Gateway:",)
NetworkView.edit()
data = {
'ip': ip.value,
'mask': mask.value,
'gateway': gateway.value
}
with open('/root/network.yaml', 'w') as fd:
yaml.dump(data, fd)
network_config(**data)
return data