#!/usr/bin/env python3 # encoding: utf-8 # # $Id$ # $URL$ """ userconfig.py Created by Marcus Stoegbauer on 2013-01-10. """ import sys import os import tempfile import re import getopt import time # import Userconfig.cfgfile as cfgfile from Userconfig.checks import Checks import Userconfig.Tools as Tools debug = Tools.Debug() classchecks = Checks() cfg = cfgfile.Conf() hostclasses = classchecks.__classes_for_host__() help_message = """ -h help -v verbose level (multiple v for higher level) -c userconfig.cfg config file """ class Usage(Exception): def __init__(self, msg): self.msg = msg def workconf(directory, depth=2): """walks through directory, collecting all filenames, returns list of all filenames""" dirs = os.listdir(directory) ret = [] debug.debug("Finding files in directory %s." % directory, 4) for d in dirs: name = directory+"/"+d if os.path.isdir(name): # fixme: create name if it does not exist workconf(name, depth+1) if name.endswith(".swp"): continue if d == ".svn": continue ret.append(name) debug.debug("Found file %s in directory %s" % (name, directory), 4) return ret def workdir(directory): """walks through all host classes, checking if the classes directory exists in directory then collect all filenames within this directory and return a dict of all files for this directory """ debug.debug("Working on directory %s" % directory, 3) # skip directory if no CONFIGFILE present if not os.path.isfile(directory+"/"+cfg.get("Main", "configfile")): debug.debug("No %s in %s, skipping." % (cfg.get("Main", "configfile"), directory), 1) return {},None # get config file for directory dir_config = Tools.get_config(directory + "/" + cfg.get("Main", "configfile")) if not dir_config: debug.debug("Cannot read %s in %s, skipping." % (cfg.get("Main", "configfile"), directory), 1) return {},None destdir = dir_config.get("Main","dest") # destfiles is a dict of all files that will be created from the classes config # key is the destination filename, values are all classes filenames that are used to # build the file destfiles = {} # walk through all know classes in directory and find filenames for h in hostclasses: # build classes directory if h[0] != "": classdir = directory+"/"+h[0]+"_"+h[1] else: classdir = directory+"/"+h[1] debug.debug("Looking for directory %s." % classdir, 4) # if class directory exists if os.path.isdir(classdir): debug.debug("Found directory %s, getting files." % classdir, 4) # get list of files within this class directory tempfiles = workconf(classdir) debug.debug("Got %d files: %s." % (len(tempfiles), str(tempfiles)), 4) # put files into dict for f in tempfiles: destname = destdir+os.path.basename(f) # destination filename if not destname in destfiles: destfiles[destname] = [] destfiles[destname].append(f) # append each file to dict debug.debug("Added file to %s, now %d files: %s" % (destname, len(destfiles[destname]), destfiles[destname]), 4) debug.debug("workdir: %s, Files: %s" % (directory, str(destfiles)), 3) return destfiles, dir_config def build_file(classfiles, destfile, commentstring): """open all classfiles, assemble them and write the contents into a tempfile returns the name of tempfile :param classfiles: :param destfile: :param commentstring: :return: str """ content = [] if commentstring != "": debug.debug("commentstring found, adding header.", 3) content.append(commentstring + " " + cfg.get("Main","stamp") + " " + time.strftime("%+") + "\n") for f in classfiles: debug.debug("Merging %s." % f, 4) fp = open(f, "r") filecontent = fp.read() fp.close() if commentstring == "": # look for stamp in content, replace with real stamp if re.search(re.escape(cfg.get("Main","stampreplace")), filecontent): debug.debug("commentstring empty, replacing stamp in file", 3) filecontent = re.sub(re.escape(cfg.get("Main","stampreplace")), cfg.get("Main","stamp"), filecontent) content.append(filecontent) (tempfd, tempfilename) = tempfile.mkstemp(prefix=os.path.basename(destfile), dir="/tmp") try: fp = os.fdopen(tempfd, "w") except: Tools.error("Cannot write to temporary file %s" % tempfilename) os.remove(tempfilename) return False debug.debug("Writing merged files into tempfile %s." % tempfilename, 3) for block in content: fp.write(block) fp.write("\n") fp.close() return tempfilename def process_all_files(destfiles, dir_config): """processes all files in destfiles, generate files from classes, compare and copy if necessary""" for df in destfiles.keys(): debug.debug("Processing source files for %s." % df, 2) # assemble file to tmp commentstring = "" if dir_config.check("Main", "commentstring"): commentstring = dir_config.get("Main", "commentstring") debug.debug("Found commentstring %s in %s" % (commentstring, df), 3) tempfilename = build_file(destfiles[df], df, commentstring) if not tempfilename: debug.debug("Error while creating temp file for %s, skipping." % df, 1) continue debug.debug("Merged files %s for %s into %s" % (str(destfiles[df]), df, tempfilename), 2) # diff assembled file and config file if Tools.diff(df, tempfilename, commentstring, debug): debug.debug("File %s has changed" % df, 0) if not Tools.user_config_generated(df, cfg): debug.debug("%s not generated by userconfig, backing up." % df, 2) # file not generated from userconfig -> back up Tools.backup_file(df, debug) # copy tmp file to real location debug.debug("Copy %s to %s." % (tempfilename, df), 0) Tools.copy_file(tempfilename, df, debug) # remove tmp debug.debug("Removing temporary file %s." % tempfilename, 2) os.remove(tempfilename) def main(): configfile = os.environ['HOME']+"/etc/userconfig.cfg" try: try: opts, args = getopt.getopt(sys.argv[1:], "hdc:v", ["help", "debug", "config="]) except getopt.GetoptError as msg: raise Usage(msg) for option, value in opts: if option == "-v": debug.addverbose() if option in ("-h", "--help"): raise Usage(help_message) if option in ("-d", "--debug.debug"): pass if option in ("-c", "--config"): configfile = value except Usage as err: Tools.error(sys.argv[0].split("/")[-1] + ": " + str(err.msg)) Tools.error("\t for help use --help") return 2 debug.debug("Using configfile %s." % configfile, 1) if not os.path.isfile(configfile): Tools.error("No config file specified.") return 2 cfg.setdebug(debug) cfg.setfilename(configfile) debug.debug("Current host is in classes %s" % hostclasses, 1) configdir = cfg.get("Main", "configdir") for d in os.listdir(configdir): destfiles = {} name = configdir+"/"+d debug.debug("Working in %s" % name, 1) if not os.path.isdir(name): debug.debug("%s is not a directory, skipping." % name, 3) continue elif d.startswith(".svn"): debug.debug("%s is .svn, skipping." % name, 3) continue elif os.path.isfile(name+"/.ignore"): debug.debug("%s contains file .ignore, skipping." % name, 3) continue else: debug.debug("Processing files in %s" % name, 2) (destfiles, dirConfig) = workdir(name) if isinstance(destfiles, dict): if len(destfiles.keys()) > 0: debug.debug("Building %d files: %s" % (len(destfiles.keys()), destfiles.keys()), 3) process_all_files(destfiles, dirConfig) else: debug.debug("No files found for %s, skipping." % name, 1) if __name__ == '__main__': main()