aboutsummaryrefslogtreecommitdiff
blob: f6341cabacc6ad6f4958a98d29983cc469807d21 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
import os
import shutil

from snakeoil.osutils import (ensure_dirs as snakeoil_ensure_dirs)


def ensure_dirs(path, gid=-1, uid=-1, mode=0o700, minimal=True, failback=None, fatal=False):
    '''Wrapper to snakeoil.osutil's ensure_dirs()
    This additionally allows for failures to run
    cleanup or other code and/or raise fatal errors.

    @param path: directory to ensure exists on disk
    @param gid: a valid GID to set any created directories to
    @param uid: a valid UID to set any created directories to
    @param mode: permissions to set any created directories to
    @param minimal: boolean controlling whether or not the specified mode
        must be enforced, or is the minimal permissions necessary.  For example,
        if mode=0755, minimal=True, and a directory exists with mode 0707,
        this will restore the missing group perms resulting in 757.
    @param failback: function to run in the event of a failed attemp
        to create the directory.
    @return: True if the directory could be created/ensured to have those
        permissions, False if not.
    '''
    succeeded = snakeoil_ensure_dirs(path, gid=-1, uid=-1, mode=mode, minimal=True)
    if not succeeded:
        if failback:
            failback()
        if fatal:
            raise IOError(
                "Failed to create directory: %s" % path)
    return succeeded


def updatefiles(config, logger, category=None, filename = None):
    if category and not filename:
        filename = config.get_key('seeds', category)
    elif not filename:
        logger.error("MAIN: updatefiles();  category or filename not supplied")
        return False
    old = filename + '.old'
    try:
        logger.info("Backing up existing file...")
        if os.path.exists(old):
            logger.debug(
                "MAIN: updatefiles(); Removing 'old' seed file: %s"
                % old)
            os.unlink(old)
        if os.path.exists(filename):
            logger.debug(
                "MAIN: updatefiles(); Renaming current seed file to: "
                "%s" % old)
            os.rename(filename, old)
        if os.path.exists(filename + '.new'):
            logger.debug(
                "MAIN: updatefiles(); Renaming '.new' seed file to: %s"
                % filename)
            os.rename(filename + '.new', filename)
        else:
            logger.error("MAIN: updatefiles(); Renaming "
                "'%s.new' seed DOES NOT EXIST!" % filename)
    except IOError:
        raise
        return False
    return True


def updateseeds(config, logger, src, dst):
    if not src or not dst:
        logger.error("MAIN: updateseeds();  source and/or destination not supplied")
        return False
    try:
        if os.path.exists(dst):
            logger.debug(
                "MAIN: updateseeds(); Removing 'old' seed file: %s"
                % dst)
            os.unlink(dst)
        if os.path.exists(src):
            logger.debug(
                "MAIN: updateseeds(); Moving new seed file to: "
                "%s" % dst)
            shutil.move(src, dst)
    except IOError:
        raise
        return False
    return True


def rm_files(config, logger, files, lock=None):
    pass