def set(self, value):
logger.info("set env: %s=%s", self.__name, value)
if sys.platform == 'win32':
ret = run(
"wmic ENVIRONMENT where \"name='%s'\" get UserName,VariableValue"
% self.__name)
if "No Instance(s) Available" in ret[1]:
runex(
'wmic ENVIRONMENT create Name="%s",UserName="******",VariableValue="%s"'
% (self.__name, value))
else:
runex(
"wmic ENVIRONMENT where \"name='%s' and username='******'\" set VariableValue=\"%s\""
% (self.__name, value))
runex("set %s=%s" % (self.__name, value))
elif sys.platform.startswith('linux'):
runex("export %s=%s" % (self.__name, value))
txtedit = TextEditor("/etc/profile")
txtedit.set_param(self.__name, value)
txtedit.delete("export %s" % self.__name)
txtedit.insert("export %s" % self.__name)
txtedit.save()
else:
pass
os.environ[self.__name] = value
def umount(target, fstab=True):
logger.debug("umount: target=%s, fstab=%s", target, fstab)
(retcode, mount_info) = oscommon.run("mount")
if retcode != 0:
raise Exception("can't run 'mount' command")
if target in mount_info:
oscommon.runex("umount %s" % target)
else:
logger.debug("target %s is not mounted", target)
if fstab:
txt = TextEditor("/etc/fstab")
txt.delete(target)
txt.save()
def mount(source, target, fstype, fstab=False):
logger.debug("mount: source=%s, target=%s, fstype=%s, fstab=%s", source, target, fstype, fstab)
(retcode, mount_info) = oscommon.run("mount")
if retcode != 0:
raise Exception("can't run 'mount' command")
if source in mount_info:
logger.debug("%s is already mounted" % source)
else:
oscommon.runex("mount -t %s %s %s" % (fstype, source, target))
if fstab:
txt = TextEditor("/etc/fstab")
txt.delete(target)
txt.insert("%s %s %s defaults 0 0" % (source, target, fstype))
txt.save()
def unset(self):
logger.info("unset env: %s", self.__name)
if self.__name in os.environ:
del os.environ[self.__name]
if sys.platform == 'win32':
ret = run(
"wmic ENVIRONMENT where \"name='%s'\" get UserName,VariableValue"
% self.__name)
if "No Instance(s) Available" not in ret[1]:
runex("wmic ENVIRONMENT where \"name='%s'\" delete" %
self.__name)
elif sys.platform.startswith('linux'):
runex("unset -v %s" % self.__name)
txtedit = TextEditor("/etc/profile")
txtedit.delete(r'^%s=' % self.__name)
txtedit.delete(r'^export %s' % self.__name)
txtedit.save()
else:
pass
def delete(name, filename=None):
filename = filename or Hosts.locate()
txt = TextEditor(filename)
txt.delete(name)
txt.save()
def set(name, addr, filename=None):
filename = filename or Hosts.locate()
txt = TextEditor(filename)
txt.set_param(name, addr, "%(value)s %(name)s")
txt.save()
def disable_selinux():
txt = TextEditor("/etc/sysconfig/selinux")
txt.set_param("SELINUX", "disabled")
txt.save()
oscommon.run("setenforce 0")
def sethostname(name):
oscommon.runex("hostname %s" % name)
txt = TextEditor("/etc/sysconfig/network")
txt.set_param("HOSTNAME", name)
txt.save()
def __init__(self, cls):
if not issubclass(cls, Daemon):
raise TypeError
self.svc_name = cls._svc_name_
self.svc_loct = "/etc/init.d/%s" % self.svc_name
self._txt = TextEditor()
