diff --git a/.idea/misc.xml b/.idea/misc.xml index 3e48b91..8c7b51b 100644 --- a/.idea/misc.xml +++ b/.idea/misc.xml @@ -3,5 +3,5 @@ - + \ No newline at end of file diff --git a/.idea/pyuserconfig.iml b/.idea/pyuserconfig.iml index 85c7612..82ffee1 100644 --- a/.idea/pyuserconfig.iml +++ b/.idea/pyuserconfig.iml @@ -4,7 +4,7 @@ - + diff --git a/cli/__init__.py b/cli/__init__.py index 8939e3c..d2ae0d0 100755 --- a/cli/__init__.py +++ b/cli/__init__.py @@ -1,11 +1,65 @@ import os import argparse from userconfig.cfgfile import Conf -from userconfig.tools import Debug from userconfig import Userconfig import sys import subprocess + +class Debug: + _verbose = 0 + + _COLOR_RED = '\033[91m' + _COLOR_BLUE = '\33[34m' + _COLOR_GREEN = '\33[32m' + _COLOR_YELLOW = '\033[93m' + _COLOR_END = '\33[0m' + + _FORMAT = {'STANDARD': '', + 'ERROR': f'{_COLOR_RED}ERROR:{_COLOR_END} ', + 'NOTICE': f'{_COLOR_BLUE}NOTICE:{_COLOR_END} ', + 'WARNING': f'{_COLOR_YELLOW}WARNING:{_COLOR_END} ', + 'SUCCESS': f'{_COLOR_GREEN}SUCCESS:{_COLOR_END} ' + } + + def __init__(self, verbose=0): + self.set_verbose(verbose) + + def set_verbose(self, verbose): + self._verbose = verbose + + def add_verbose(self): + self._verbose += 1 + + def get_verbose(self): + return self._verbose + + def stdout(self, out, verbose_level=0, category='STANDARD'): + if self._verbose >= verbose_level: + if category in self._FORMAT: + print(f'{self._FORMAT[category]}{out}') + else: + print(f'[category {category} unknown]:{out}') + + def stderr(self, out, verbose_level=0, category='STANDARD'): + if self._verbose >= verbose_level: + if category in self._FORMAT: + print(f'{self._FORMAT[category]}{out}', file=sys.stderr) + else: + print(f'[category {category} unknown]:{out}', file=sys.stderr) + + def red(self, out): + return f'{self._COLOR_RED}{out}{self._COLOR_END}' + + def green(self, out): + return f'{self._COLOR_GREEN}{out}{self._COLOR_END}' + + def blue(self, out): + return f'{self._COLOR_BLUE}{out}{self._COLOR_END}' + + def yellow(self, out): + return f'{self._COLOR_YELLOW}{out}{self._COLOR_END}' + def main(): parser = argparse.ArgumentParser(prog='userconfig', description='Manages configuration files, usually in the user home directory') @@ -20,13 +74,12 @@ def main(): debug = Debug() # debug.set_verbose(cmdline.verbose) # cfg = Conf(filename=cmdline.file, debug=debug) - debug.set_verbose(4) - cfg = Conf(filename='/Users/lysis/userconfig2-test.conf', debug=debug) - # cfg = Conf(filename='/Users/lysis/etc/userconfig2.conf', debug=debug) + debug.set_verbose(1) + cfg = Conf(filename='/home/lysis/userconfig2-test.conf', debug=debug) uc = Userconfig(cfg) - cfg.debug.stdout(f"Verbose level is {cfg.debug.get_verbose()}", 1, 'STANDARD') - configdir = cfg.get("configdir") + cfg.debug.stdout(f'Verbose level: {cfg.debug.get_verbose()}', 1, 'STANDARD') + configdir = cfg.get('configdir') # configdir is the root of the userconfig files # Directory structure: # configdir/ @@ -49,7 +102,7 @@ def main(): # [Number]_[category]_[value] # Number: can have leading zeros for better sorting in directory structure, will be used for sorting and priority # category: which kind of match we are looking for. currently: Arch for `uname -s`, Host for `hostname -s` - # value: optional, if category requires input, for example: Arch_Linux + # value: optional, if category requires an input value, for example: Arch_Linux # can be empty for example if category is all (no match needed, we want this always to be applied) cfg.debug.stdout(f'configdir: {configdir}', 1) @@ -62,7 +115,8 @@ def main(): if os.path.isfile(f'{package.path}/.ignore'): cfg.debug.stdout(f'{package.path} contains .ignore, skipping', 2, 'WARNING') continue - cfg.debug.stdout(f'============ start {package.path} ============', 2) + cfg.debug.stdout(f'Package: {package.path}', 1, 'NOTICE') + cfg.debug.stdout(f'============ start {cfg.debug.green(package.path)} ============', 2) (category_dirs, dir_config) = uc.process_package_dir(package.path) cfg.debug.stdout(f'Got categories: {category_dirs}', 2) host_category_dirs = uc.filter_categories(category_dirs) @@ -84,8 +138,9 @@ def main(): # Make sure all directories for destination file exist uc.create_destination_directories(dir_config.get(section="Main", option="dest")) temp_filename = uc.build_file(file_list[file], dest, comment_string) - uc.copy_file(temp_filename, dest, comment_string) - cfg.debug.stdout(f'============ end {package.path} ============\n\n', 2) + uc.diff_and_copy_file(temp_filename, dest, comment_string) + cfg.debug.stdout(f'============ end {cfg.debug.green(package.path)} ============\n\n', 2) + if __name__ == '__main__': main() diff --git a/userconfig/__init__.py b/userconfig/__init__.py index c0a219b..4bb229e 100644 --- a/userconfig/__init__.py +++ b/userconfig/__init__.py @@ -1,10 +1,12 @@ -from userconfig.tools import get_config from userconfig.checks import check_class +from userconfig.cfgfile import Conf import os -from userconfig.tools import diff, user_config_generated, backup_file, copy_file import time import tempfile from pathlib import Path +import shutil +import re +import sys class Userconfig: @@ -13,7 +15,6 @@ class Userconfig: def __init__(self, cfg): self._cfg = cfg - """ This is NEW """ def process_package_dir(self, package_dir): """ Walk through package_dir, collect all directories inside, parse them according to our specs and return @@ -28,7 +29,7 @@ class Userconfig: self._cfg.debug.stdout(f'No config file {self._cfg.get("configfile")} in {package_dir}, skipping', 1, 'ERROR') return None - dir_config = get_config(package_config_file, self._cfg) + dir_config = self.get_config(package_config_file) if not dir_config: self._cfg.debug.stdout(f'Cannot read config file {package_config_file}, skipping', 1, 'ERROR') return None @@ -95,52 +96,52 @@ class Userconfig: file_list[file.name].append(file.path) return file_list - def build_file(self, files, destfile, commentstring): + def build_file(self, files, dest_file, comment_string): """ - merge all files in files[] and write contents into a tempfile + merge all files in files[] and write contents into a temporary file :param files: - :param destfile: - :param commentstring: - :return: tempfile name + :param dest_file: + :param comment_string: + :return: temporary filename """ content = [] - self._cfg.debug.stdout(f'*** building file for {destfile}', 3) - if commentstring: - content.append(f'{commentstring} {self._cfg.get("stamp")} {time.strftime("%+")}\n') + self._cfg.debug.stdout(f'*** building file for {dest_file}', 3) + if comment_string: + content.append(f'{comment_string} {self._cfg.get("stamp")} {time.strftime("%+")}\n') for file in files: self._cfg.debug.stdout(f'Merging {file}', 3) fp = open(file, "r") - filecontent = fp.read() + file_content = fp.read() fp.close() - # if commentstring exists, add comment with category dir and filename - if commentstring: - content.append(f'{commentstring} {"/".join(file.split("/")[-2:])}') - content.append(filecontent) + # if comment_string exists, add comment with category dir and filename + if comment_string: + content.append(f'{comment_string} {"/".join(file.split("/")[-2:])}') + content.append(file_content) - (tempfd, tempfilename) = tempfile.mkstemp(prefix=os.path.basename(destfile), dir="/tmp") + (temp_fd, temp_filename) = tempfile.mkstemp(prefix=os.path.basename(dest_file), dir="/tmp") try: - fp = os.fdopen(tempfd, "w") + fp = os.fdopen(temp_fd, "w") except Exception as e: - self._cfg.debug.stderr(f"Cannot write to temporary file {tempfilename}: {e}", 0, 'ERROR') - os.remove(tempfilename) + self._cfg.debug.stderr(f"Cannot write to temporary file {temp_filename}: {e}", 0, 'ERROR') + os.remove(temp_filename) return False - self._cfg.debug.stdout(f'Writing merged files into tempfile {tempfilename}', 3) + self._cfg.debug.stdout(f'Writing merged files into temporary file {temp_filename}', 3) for block in content: fp.write(block) fp.write("\n") fp.close() - return tempfilename + return temp_filename - def copy_file(self, temp_filename, dest_filename, commentstring): - if diff(dest_filename, temp_filename, commentstring, self._cfg): + def diff_and_copy_file(self, temp_filename, dest_filename, comment_string): + if self.diff(dest_filename, temp_filename, comment_string): self._cfg.debug.stdout(f"File {dest_filename} has changed", 0, 'NOTICE') - if not user_config_generated(dest_filename, self._cfg): + if not self.user_config_generated(dest_filename): self._cfg.debug.stdout(f"{dest_filename} not generated by userconfig, backing up.", 3) - backup_file(dest_filename, self._cfg) + self.backup_file(dest_filename) self._cfg.debug.stdout(f"Copy {temp_filename} to {dest_filename}", 0, 'NOTICE') - copy_file(temp_filename, dest_filename, self._cfg) + self.copy_file(temp_filename, dest_filename) self._cfg.debug.stdout(f"Removing temporary file {temp_filename}", 3) os.remove(temp_filename) @@ -150,160 +151,103 @@ class Userconfig: if not path.exists(): path.mkdir(parents=True) - # def build_file_old(self, 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 = [] - # self._cfg.debug.stdout(f" ================ build_file {destfile} ===============", 1) - # if commentstring != "": - # self._cfg.debug.stdout(" +++ commentstring found, adding header.", 3) - # content.append(commentstring + " " + self._cfg.get("stamp") + " " + time.strftime("%+") + "\n") - # - # for f in classfiles: - # self._cfg.debug.stdout(" +++ 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(self._cfg.get("stampreplace")), filecontent): - # self._cfg.debug.stdout(" +++ commentstring empty, replacing stamp in file", 3) - # filecontent = re.sub(re.escape(self._cfg.get("stampreplace")), self._cfg.get("stamp"), - # filecontent) - # content.append(filecontent) - # - # (tempfd, tempfilename) = tempfile.mkstemp(prefix=os.path.basename(destfile), dir="/tmp") - # - # try: - # fp = os.fdopen(tempfd, "w") - # except Exception as e: - # self._cfg.debug.stderr(f"Cannot write to temporary file {tempfilename}: {e}") - # os.remove(tempfilename) - # return False - # self._cfg.debug.stdout(" +++ Writing merged files into tempfile %s." % tempfilename, 3) - # for block in content: - # fp.write(block) - # fp.write("\n") - # fp.close() - # self._cfg.debug.stdout(f" ================ done: build_file {destfile} ===============", 1) - # return tempfilename + def get_config(self, filename): + """reads filename as config, checks for DEST parameter and returns cfgfile object""" + try: + ret = Conf(filename=filename, debug=self._cfg.debug, force_filename=True) + except ValueError: + self._cfg.debug.stderr("Error reading config file %s" % filename) + return False + # check for DEST parameter + if not ret.check(section="Main", option="dest"): + self._cfg.debug.stderr("No dest in config file %s" % filename) + return False + # make sure DEST ends with / + if not ret.get(section="Main", option="dest").endswith("/"): + ret.set(section="Main", option="dest", value=ret.get(section="Main", option="dest")+"/") + return ret - # def process_all_files(self, destfiles, dir_config): - # """processes all files in destfiles, generate files from classes, compare and copy if necessary""" - # self._cfg.debug.stdout(" ================ process_all_files ===============", 1) - # for df in destfiles.keys(): - # self._cfg.debug.stdout(" ??? Processing source files for %s." % df, 2) - # if not os.path.exists(os.path.dirname(df)): - # self._cfg.debug.stdout(" +++ Directory %s does not exist, creating" % os.path.dirname(df), 1) - # os.mkdir(os.path.dirname(df)) - # if not os.path.isdir(os.path.dirname(df)): - # self._cfg.debug.stdout(f" --- Destination directory {os.path.dirname(df)} does not exist, skipping.", - # 1) - # return False - # # assemble file to tmp - # commentstring = "" - # if dir_config.check(section="Main", option="commentstring"): - # commentstring = dir_config.get(section="Main", option="commentstring") - # self._cfg.debug.stdout(" +++ Found commentstring %s in %s" % (commentstring, df), 3) - # - # tempfilename = self.build_file(destfiles[df], df, commentstring) - # if not tempfilename: - # self._cfg.debug.stdout(" --- Error while creating temp file for %s, skipping." % df, 1) - # continue - # self._cfg.debug.stdout(" +++ Merged files %s for %s into %s" % (str(destfiles[df]), df, tempfilename), 2) - # - # # diff assembled file and config file - # if diff(df, tempfilename, commentstring, self._cfg): - # self._cfg.debug.stdout("File %s has changed" % df, 0) - # if not user_config_generated(df, self._cfg): - # self._cfg.debug.stdout(" +++ %s not generated by userconfig, backing up." % df, 2) - # # file not generated from userconfig -> back up - # backup_file(df, self._cfg) - # # copy tmp file to real location - # self._cfg.debug.stdout("Copy %s to %s." % (tempfilename, df), 0) - # copy_file(tempfilename, df, self._cfg) - # # remove tmp - # self._cfg.debug.stdout(" +++ Removing temporary file %s." % tempfilename, 2) - # os.remove(tempfilename) - # self._cfg.debug.stdout(" ================ process_all_files ===============", 1) + @staticmethod + def read_skip_comment(fp, comment_string): + """Read line from filehandle fp and skip all empty (whitespace) lines and lines starting with comment_string + """ + for line in fp: + line = line[:-1] + if ((comment_string != "" and not re.match("^"+re.escape(comment_string), line)) and line != "" and + not re.match(r"^\s+$", line)): + yield line - # def workconf(self, directory, depth=2): - # """walks through directory, collecting all filenames, returns list of all filenames""" - # dirs = os.listdir(directory) - # ret = [] - # self._cfg.debug.stdout(f" ================ workconf {directory} ===============", 1) - # for d in dirs: - # name = directory + "/" + d - # if os.path.isdir(name): - # self.workconf(name, depth + 1) - # if name.endswith(".swp"): - # continue - # if d == ".svn": - # continue - # ret.append(name) - # self._cfg.debug.stdout(" +++ Found file %s in directory %s" % (name, directory), 4) - # self._cfg.debug.stdout(f" ================ done: workconf {directory} ===============", 1) - # return ret + def diff(self, dest_file, temp_file, comment_string): + """diff dest_file and temp_file, returns True if files differ, False if they are the same""" + self._cfg.debug.stdout("Diffing %s and %s, comment: %s" % (dest_file, temp_file, comment_string), 3) + if not os.path.isfile(dest_file): + self._cfg.debug.stdout("Destination file %s does not exist, returning True." % dest_file, 3) + # dest_file does not exist -> copy temp_file over + return True + if not os.path.isfile(temp_file): + # temp_file does not exist, this should never happen + self._cfg.debug.stderr("Temporary file %s does not exist, this should not happen." % temp_file) + sys.exit(1) - # def workdir(self, 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 - # """ - # self._cfg.debug.stdout(f" ================ workdir {directory} ===============", 1) - # # skip directory if no CONFIGFILE present - # if not os.path.isfile(directory+"/"+self._cfg.get("configfile")): - # self._cfg.debug.stdout(f' --- No file {self._cfg.get("configfile")} in {directory}, skipping.', 1) - # return {}, None - # - # # get config file for directory - # dir_config = get_config(directory + "/" + self._cfg.get("configfile"), self._cfg) - # if not dir_config: - # self._cfg.debug.stdout(f' --- Cannot read config file {self._cfg.get("configfile")} in {directory}, ' - # f'skipping.', 1) - # return {}, None - # - # destdir = dir_config.get(section="Main", option="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 = {} - # try: - # reverse_sort = dir_config.get(section="Main", option="reverse", boolean=True) - # except ValueError: - # reverse_sort = False - # self._cfg.debug.stdout(f' +++ reverse_sort is {reverse_sort}', 3) - # if os.access(directory + "/install.sh", os.X_OK): - # subprocess.call([directory + "/install.sh"]) - # - # # walk through all know classes in directory and find filenames - # for h in classes_for_host(reverse_sort): - # # build classes directory - # if h[0] != "": - # classdir = directory+"/"+h[0]+"_"+h[1] - # else: - # classdir = directory+"/"+h[1] - # self._cfg.debug.stdout(" ??? Looking for directory %s." % classdir, 4) - # # if class directory exists - # if os.path.isdir(classdir): - # self._cfg.debug.stdout(" +++ Found directory %s, getting files." % classdir, 4) - # # get list of files within this class directory - # tempfiles = self.workconf(classdir) - # self._cfg.debug.stdout(" +++ 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 destname not in destfiles: - # destfiles[destname] = [] - # destfiles[destname].append(f) # append each file to dict - # self._cfg.debug.stdout(f" +++ Added file to {destname}, now {len(destfiles[destname])} files: " - # f"{destfiles[destname]}", 4) - # - # self._cfg.debug.stdout(" === workdir: %s, Files: %s" % (directory, str(destfiles)), 3) - # self._cfg.debug.stdout(f" ================ done: workdir {directory} ===============", 1) - # return destfiles, dir_config + fp1 = open(temp_file) + fp2 = open(dest_file) + + for line1, line2 in zip(self.read_skip_comment(fp1, comment_string), self.read_skip_comment(fp2, comment_string)): + if line1 != line2: + fp1.close() + fp2.close() + self._cfg.debug.stdout("%s differs, return true" % dest_file, 3) + return True + fp1.close() + fp2.close() + self._cfg.debug.stdout("%s is the same, return false" % dest_file, 3) + return False + + def user_config_generated(self, filename): + """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 self._cfg.check("stamp"): + # no STAMP in userconfig.cfg, so no way to check if file was generated by userconfig + return False + + fp = open(filename, "r") + + for line in fp: + if re.search(re.escape(self._cfg.get("stamp")), line): + return True + return False + + def backup_file(self, filename): + """make backup of filename, returns True if backup is successful, False else""" + if os.path.isfile(filename): + self._cfg.debug.stdout("%s exists, finding backup name." % filename, 3) + backup_name = filename+".userconfig."+time.strftime("%F") + test_backup_name = backup_name + counter = 0 + while os.path.isfile(test_backup_name): + counter += 1 + test_backup_name = backup_name+"."+str(counter) + self._cfg.debug.stdout("Renaming %s to %s" % (filename, test_backup_name), 1) + os.rename(filename, test_backup_name) + return True + else: + self._cfg.debug.stdout("%s does not exist, do not need backup." % filename, 3) + return False + + def copy_file(self, sourcefile, dest_file): + """copy sourcefile to dest_file, returns True if successful, False else""" + + if os.path.isfile(sourcefile): + # sourcefile exists + self._cfg.debug.stdout("Source file %s exists, proceeding with copy." % sourcefile, 3) + if not os.path.isfile(dest_file) or os.access(dest_file, os.W_OK): + self._cfg.debug.stdout("Copying %s to %s" % (sourcefile, dest_file), 1) + shutil.copy(sourcefile, dest_file) + return True + else: + self._cfg.debug.stdout("Destination file %s does not exist or is not writable." % dest_file, 3) + return False diff --git a/userconfig/tools.py b/userconfig/tools.py deleted file mode 100644 index 20b7603..0000000 --- a/userconfig/tools.py +++ /dev/null @@ -1,160 +0,0 @@ -import sys -import os -from userconfig.cfgfile import Conf -import re -import time -import shutil - - -class Debug: - _verbose = 0 - - _COLOR_RED = '\033[91m' - _COLOR_BLUE = '\33[34m' - _COLOR_GREEN = '\33[32m' - _COLOR_YELLOW = '\033[93m' - _COLOR_END = '\33[0m' - - _FORMAT = { 'STANDARD': '', - 'ERROR': f'{_COLOR_RED}ERROR:{_COLOR_END} ', - 'NOTICE': f'{_COLOR_BLUE}NOTICE:{_COLOR_END} ', - 'WARNING': f'{_COLOR_YELLOW}WARNING:{_COLOR_END} ', - 'SUCCESS': f'{_COLOR_GREEN}SUCCESS:{_COLOR_END} ' - } - - def __init__(self, verbose=0): - self.set_verbose(verbose) - - def set_verbose(self, verbose): - self._verbose = verbose - - def add_verbose(self): - self._verbose += 1 - - def get_verbose(self): - return self._verbose - - def stdout(self, out, verbose_level=0, category='STANDARD'): - if self._verbose >= verbose_level: - if category in self._FORMAT: - print(f'{self._FORMAT[category]}{out}') - else: - print(f'[category {category} unknown]:{out}') - - def stderr(self, out, verbose_level=0, category='STANDARD'): - if self._verbose >= verbose_level: - if category in self._FORMAT: - print(f'{self._FORMAT[category]}{out}', file=sys.stderr) - else: - print(f'[category {category} unknown]:{out}', file=sys.stderr) - - -def get_config(filename, cfg): - """reads filename as config, checks for DEST parameter and returns cfgfile object""" - try: - ret = Conf(filename=filename, debug=cfg.debug, force_filename=True) - except ValueError: - cfg.debug.stderr("Error reading config file %s" % filename) - return False - # check for DEST parameter - if not ret.check(section="Main", option="dest"): - cfg.debug.stderr("No dest in config file %s" % filename) - return False - # make sure DEST ends with / - if not ret.get(section="Main", option="dest").endswith("/"): - ret.set(section="Main", option="dest", value=ret.get(section="Main", option="dest")+"/") - return ret - - -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(r"^\s+$", line)): - yield line - - -def diff(destfile, tempfile, commentstring, cfg): - """diff destfile and tempfile, returns True if files differ, False if they are the same""" - cfg.debug.stdout("Diffing %s and %s, comment: %s" % (destfile, tempfile, commentstring), 3) - if not os.path.isfile(destfile): - cfg.debug.stdout("Destfile %s does not exist, returning True." % destfile, 3) - # 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 - cfg.debug.stderr("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 zip(read_skip_comment(fp1, commentstring), read_skip_comment(fp2, commentstring)): - if line1 != line2: - fp1.close() - fp2.close() - cfg.debug.stdout("%s differs, return true" % destfile, 3) - return True - # if differ - # for line - fp1.close() - fp2.close() - cfg.debug.stdout("%s is the same, return false" % destfile, 3) - return False - - -def user_config_generated(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 cfg.check("stamp"): - # no STAMP in userconfig.cfg, so no way to check if file was generated by userconfig - return False - - fp = open(filename, "r") - - for line in fp: - if re.search(re.escape(cfg.get("stamp")), line): - return True - return False - - -def backup_file(filename, cfg): - """make backup of filename, returns True if backup is successful, False else""" - if os.path.isfile(filename): - cfg.debug.stdout("%s exists, finding backup name." % filename, 3) - backupname = filename+".userconfig."+time.strftime("%F") - testbackupname = backupname - counter = 0 - while os.path.isfile(testbackupname): - counter += 1 - testbackupname = backupname+"."+str(counter) - cfg.debug.stdout("Renaming %s to %s" % (filename, testbackupname), 1) - os.rename(filename, testbackupname) - return True - else: - cfg.debug.stdout("%s does not exist, do not need backup." % filename, 3) - return False - - -def copy_file(sourcefile, destfile, cfg): - """copy sourcefile to destfile, returns True if successful, False else""" - - if os.path.isfile(sourcefile): - # sourcefile exists - cfg.debug.stdout("Source file %s exists, proceeding with copy." % sourcefile, 3) - if not os.path.isfile(destfile) or os.access(destfile, os.W_OK): - cfg.debug.stdout("Copying %s to %s" % (sourcefile, destfile), 1) - shutil.copy(sourcefile, destfile) - return True - # destfile is writable - else: - cfg.debug.stdout("Destination file %s does not exist or is not writable." % destfile, 3) - return False