147 lines
2.9 KiB
Python
147 lines
2.9 KiB
Python
|
__author__ = 'vanzhiganov'
|
||
|
|
|
||
|
|
"""
|
||
|
|
import dnsmasq
|
||
|
|
dnsmasq.Dnsmasq().exists_name('test')
|
||
|
|
dnsmasq.Dnsmasq().add('test', '10.10.10.10')
|
||
|
|
dnsmasq.Dnsmasq().exists_name('test')
|
||
|
|
|
||
|
|
"""
|
||
|
|
|
||
|
|
|
||
|
|
import shutil
|
||
|
|
import subprocess
|
||
|
|
import os.path
|
||
|
|
|
||
|
|
|
||
|
|
class Error(Exception):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class NameAlreadyExists(Exception):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class NameNotExists(Exception):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class IPAlreadyExists(Exception):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class IPNotExists(Exception):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class Dnsmasq:
|
||
|
|
def __init__(self, dnsmasq_conf='/etc/lxc/dnsmasq.conf'):
|
||
|
|
"""
|
||
|
|
|
||
|
|
:param dnsmasq_conf:
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
if not self.__exists_conf(dnsmasq_conf):
|
||
|
|
raise Error("Config not exists")
|
||
|
|
|
||
|
|
self.dnsmasq_conf = dnsmasq_conf
|
||
|
|
self.list = self.__read_conf()
|
||
|
|
|
||
|
|
def exists_name(self, name):
|
||
|
|
"""
|
||
|
|
|
||
|
|
:param name:
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
# origin = self.__read_conf()
|
||
|
|
for i in self.list:
|
||
|
|
if name in i:
|
||
|
|
return True
|
||
|
|
return False
|
||
|
|
|
||
|
|
# @staticmethod
|
||
|
|
def exists_ip(self, ip):
|
||
|
|
"""
|
||
|
|
|
||
|
|
:param ip:
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
# origin = self.__read_conf()
|
||
|
|
for i in self.list:
|
||
|
|
if ip in i:
|
||
|
|
return True
|
||
|
|
return False
|
||
|
|
|
||
|
|
def add(self, name, ip):
|
||
|
|
"""
|
||
|
|
|
||
|
|
:param name:
|
||
|
|
:param ip:
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
if not self.__exists_conf(self.dnsmasq_conf):
|
||
|
|
return False
|
||
|
|
|
||
|
|
if self.exists_name(name):
|
||
|
|
raise NameAlreadyExists("Already exists: %s" % name)
|
||
|
|
if self.exists_ip(ip):
|
||
|
|
raise IPAlreadyExists("Already exists: %s" % ip)
|
||
|
|
self.list.append("dhcp-host=%s,%s" % (name, ip))
|
||
|
|
result_write = self.__write_conf()
|
||
|
|
if not result_write:
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
|
||
|
|
def delete(self, name):
|
||
|
|
if not self.__exists_conf(self.dnsmasq_conf):
|
||
|
|
return False
|
||
|
|
|
||
|
|
clean = []
|
||
|
|
for i in self.__read_conf():
|
||
|
|
if name not in i:
|
||
|
|
clean.append(i)
|
||
|
|
result_write = self.__write_conf()
|
||
|
|
if not result_write:
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
|
||
|
|
def __read_conf(self):
|
||
|
|
"""
|
||
|
|
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
origin = []
|
||
|
|
for i in open(self.dnsmasq_conf, "r"):
|
||
|
|
origin.append(i.rstrip())
|
||
|
|
return origin
|
||
|
|
|
||
|
|
def __write_conf(self):
|
||
|
|
"""
|
||
|
|
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
if not self.__exists_conf(self.dnsmasq_conf):
|
||
|
|
return False
|
||
|
|
|
||
|
|
f = open("/tmp/dnsmasq.conf", 'w')
|
||
|
|
# f.writelines(self.list)
|
||
|
|
for i in self.list:
|
||
|
|
f.write("%s\n" % i)
|
||
|
|
f.close()
|
||
|
|
shutil.copyfile("/tmp/dnsmasq.conf", "/etc/lxc/dnsmasq.conf")
|
||
|
|
return True
|
||
|
|
|
||
|
|
def __exists_conf(self, conf):
|
||
|
|
if os.path.isfile(conf):
|
||
|
|
return True
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
class Service:
|
||
|
|
def restart(self):
|
||
|
|
subprocess.call("service dnsmasq restart", shell=True)
|
||
|
|
return True
|
||
|
|
|
||
|
|
def pid(self):
|
||
|
|
return None
|