43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
import os
|
|
|
|
|
|
class Network(object):
|
|
def __init__(self, network_xml):
|
|
self.xml = network_xml
|
|
self.hostname = self.xml.attrib['hostname'].strip()
|
|
self.provider = self.xml.attrib.get('provider', 'networkd').strip()
|
|
handler = None
|
|
if self.provider == 'netctl':
|
|
import aif.network.netctl as handler
|
|
elif self.provider == 'nm':
|
|
import aif.network.networkmanager as handler
|
|
elif self.provider == 'networkd':
|
|
import aif.network.networkd as handler
|
|
self.provider = handler
|
|
if not self.provider:
|
|
raise RuntimeError('Could not determine handler')
|
|
self.connections = []
|
|
self._initConns()
|
|
|
|
def _initConns(self):
|
|
for e in self.xml.xpath('ethernet|wireless'):
|
|
conn = None
|
|
if e.tag == 'ethernet':
|
|
conn = self.provider.Ethernet(e)
|
|
elif e.tag == 'wireless':
|
|
conn = self.provider.Wireless(e)
|
|
self.connections.append(conn)
|
|
|
|
def apply(self, chroot_base):
|
|
cfg = os.path.join(chroot_base, 'etc', 'hostname')
|
|
with open(cfg, 'w') as fh:
|
|
fh.write('{0}\n'.format(self.hostname))
|
|
os.chown(cfg, 0, 0)
|
|
os.chmod(cfg, 0o0644)
|
|
for iface in self.connections:
|
|
for src, dest in iface.services.items():
|
|
realdest = os.path.join(chroot_base, dest)
|
|
os.symlink(src, realdest)
|
|
iface.writeConf(chroot_base)
|
|
return()
|