Unterverzeichnisse angelegt

userconfig/Tools.py

@@ -0,0 +1,170 @@
+#!/usr/bin/env python
+# encoding: utf-8
+# $Id$
+# $URL$
+
+"""
+Tools.py
+
+Created by Marcus Stoegbauer on 2013-01-12.
+Copyright (c) 2013 __MyCompanyName__. All rights reserved.
+"""
+
+import sys
+import os
+import cfgfile
+import re
+import time
+import shutil
+import itertools
+
+
+class Debug(object):
+  verbose = 0
+  
+  def __init__(self, verbose=0):
+    self.setverbose(verbose)
+  # def __init__
+  
+  def setverbose(self, verbose):
+    """docstring for setverbose"""
+    self.verbose = verbose
+  # def setverbose
+  
+  def debug(self, out, level=0):
+    """docstring for debug"""
+    if self.verbose:
+      if level > 0:
+        print "*"*level,out
+      else:
+        print out
+    # if verbose
+  # def debug
+# class Debug
+
+def error(out):
+  """Print error on stderr"""
+  print >> sys.stderr, str(out)+"\n"
+# def error
+
+def getConfig(filename):
+  """reads filename as config, checks for DEST parameter and returns cfgfile object"""
+  try:
+    ret = cfgfile.Conf(filename)
+  except:
+    error("Error reading config file %s" % filename)
+    return False
+  # try
+  
+  # check for DEST parameter
+  if not ret.check("Main", "DEST"):
+    error("No DEST in config file %s" % filename)
+    return False
+  # if no DEST
+
+  # replace $HOME with real home directory
+  if ret.get("Main", "DEST") == "$HOME":
+    ret.set("Main", "DEST", os.environ['HOME'])
+  # if $HOME
+  
+  # make sure DEST ends with /
+  if not ret.get("Main", "DEST").endswith("/"):
+    ret.set("Main", "DEST", ret.get("Main", "DEST")+"/")
+  # if not /
+  
+  return ret
+# def getConfig
+
+def read_skip_comment(fp, commentstring):
+  """Read line from filehandle fp and skip all empty (whitespace) lines and lines starting with commentstring
+  """
+  for line in fp:
+    line = line[:-1]
+    if (commentstring != "" and not re.match("^"+re.escape(commentstring), line)) and line !="" and not re.match("^\s+$", line):
+      yield line
+    # if not match
+  # for line
+# def read_skip_comment
+
+def diff(destfile, tempfile, commentstring, debug):
+  """diff destfile and tempfile, returns True if files differ, False if they are the same"""
+  debug.debug("Diffing %s and %s" % (destfile, tempfile))
+  if not os.path.isfile(destfile):
+    debug.debug("Destfile %s does not exist, returning True." % destfile)
+    # destfile does not exist -> copy tempfile over
+    return True
+  # if not destfile
+  if not os.path.isfile(tempfile):
+    # tempfile does not exist, this should never happen
+    error("Temporary file %s does not exist, this should not happen." % tempfile)
+    sys.exit(1)
+  # if not tempfile
+  
+  fp1 = open(tempfile)
+  fp2 = open(destfile)
+  
+  for line1, line2 in itertools.izip(read_skip_comment(fp1, commentstring), read_skip_comment(fp2, commentstring)):
+    if line1 != line2:
+      fp1.close()
+      fp2.close()
+      return True
+    # if differ
+  # for line
+  fp1.close()
+  fp2.close()
+  return False
+# def diff
+
+def userConfigGenerated(filename, cfg):
+  """returns True if filename has been generated by userconfig, False else"""
+
+  if not os.path.isfile(filename):
+    # filename does not exist, so it was not generated by userconfig
+    return False
+  # if not filename
+  
+  if not cfg.check("Main","STAMP"):
+    # no STAMP in userconfig.cfg, so no way to check if file was generated by userconfig
+    return False
+  # no STAMP
+
+  fp = open(filename, "r")
+  
+  for line in fp:
+    if re.search(re.escape(cfg.get("Main","STAMP")), line):
+      return True
+    # if search
+  # for line
+  return False
+# def userConfigGenerated
+
+def backupFile(filename, debug):
+  """make backup of filename, returns True if backup is successful, False else"""
+  if os.path.isfile(filename):
+    backupname = filename+".userconfig."+time.strftime("%F")
+    testbackupname = backupname
+    counter = 0
+    while os.path.isfile(testbackupname):
+      counter+=1
+      testbackupname=backupname+"."+str(counter)
+    # while
+    debug.debug("Renaming %s to %s" % (filename, testbackupname))
+    os.rename(filename, testbackupname)
+    return True
+  # if filename
+  return False
+# def backupFile
+
+def copyFile(sourcefile, destfile, debug):
+  """copy sourcefile to destfile, returns True if successful, False else"""
+  
+  if os.path.isfile(sourcefile):
+    # sourcefile exists
+    if not os.path.isfile(destfile) or os.access(destfile, os.W_OK):
+      debug.debug("Copying %s to %s" % (sourcefile, destfile))
+      shutil.copy(sourcefile, destfile)
+      return True
+      # destfile is writable
+    # if write ok
+  return False
+# def copyFile

userconfig/__init__.py

@@ -0,0 +1,2 @@
+# $Id$
+# $URL$

userconfig/cfgfile.py

@@ -0,0 +1,64 @@
+#!/usr/bin/env python
+# encoding: utf-8
+#
+# $Id$
+# $URL$
+
+"""
+cfgfile.py
+
+Created by Marcus Stoegbauer on 2013-01-12.
+Copyright (c) 2013 __MyCompanyName__. All rights reserved.
+"""
+from ConfigParser import ConfigParser
+
+class Conf(object):
+  confobj = ConfigParser()
+  cfgfile = ''
+
+  def __init__(self, filename = None):
+    """if filename is set, open config file and initialize the ConfigParser
+    """
+    if filename:
+      self.setfilename(filename)
+    # if filename
+  # def __init__
+
+  def setfilename(self, filename):
+    """initialize the ConfigParser
+    """
+    if len(self.confobj.read(filename)) == 0 or self.confobj.read(filename)[0] != filename:
+      raise Exception('Cannot read config file ' + filename)
+    # if cannot read
+    self.cfgfile = filename
+  # def setfilename
+
+  def get(self, section, option):
+    """returns the value of option in section
+    """
+    if not self.cfgfile:
+      raise Exception('No config file set')
+    # if not cfgfile
+    return self.confobj.get(section, option)
+  # def get
+
+  def set(self, section, option, value):
+    """docstring for update"""
+    self.confobj.set(section, option, value)
+  # def set
+
+
+  def getitems(self, section):
+    """returns all items in section
+    """
+    if not self.cfgfile:
+      raise Exception('No config file set')
+    # if not cfgfile
+    return self.confobj.items(section)
+  # def getitems
+
+  def check(self, section, option):
+    """checks for option in section"""
+    return self.confobj.has_option(section, option)
+  # def check    
+# class Conf

userconfig/checks.py

@@ -0,0 +1,62 @@
+#!/usr/bin/env python
+# encoding: utf-8
+#
+# $Id$
+# $URL$
+
+"""
+checks.py
+
+Created by Marcus Stoegbauer on 2013-01-10.
+Copyright (c) 2013 __MyCompanyName__. All rights reserved.
+"""
+
+import sys
+import os
+import platform
+
+class checks(object):
+  def __init__(self):
+    pass
+  # def __init__
+  
+  def __classesForHost__(self):
+    """docstring for __classesForHost"""
+    classes = []
+    for c in dir(self):
+      if c.startswith("__"): 
+        continue
+      # if __
+      ret = getattr(self, c)()
+      classes.append(ret)
+    # for c
+    return map(lambda k: (k[1],k[2]), sorted(classes, key=lambda k: k[0]))
+  # def __classesForHost__
+  
+  def header(self):
+    """docstring for header"""
+    return (0, "", "header")
+  # def header
+  
+  def footer(self):
+    """docstring for footer"""
+    return (1000, "", "footer")
+  
+  def all(self):
+    """docstring for all"""
+    return (998, "", "all")
+  # def all
+  
+  def arch(self):
+    """docstring for arch"""
+    return(800, "Arch", platform.system())
+  # def arch
+  
+  def hostname(self):
+    """docstring for hostname"""
+    hostname = platform.node()
+    if hostname.count("."):
+      hostname = hostname.split(".")[0]
+    return(10, "Host", hostname)
+  # def hostname
+# def checks