22

✏️ Editing: authcompat_EnvironmentFile.py

# -*- coding: utf-8 -*-
#
#    Authors:
#        Pavel Březina <pbrezina@redhat.com>
#
#    Copyright (C) 2018 Red Hat
#
#    This program is free software; you can redistribute it and/or modify
#    it under the terms of the GNU General Public License as published by
#    the Free Software Foundation; either version 3 of the License, or
#    (at your option) any later version.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
#

import errno
import os
import re

class EnvironmentFile:
    TEST = False

def __init__(self, filename,
                 delimiter='=', delimiter_re=None,
                 quotes=True):
        self.filename = filename
        self.delimiter = delimiter
        self.quotes = quotes
        self.environment = []

delimiter_re = delimiter_re if delimiter_re is not None else delimiter
        self.pattern = re.compile(r'^(\s*)(\S*)([^\n\S]*)('
                                  + delimiter_re
                                  + r')([^\n\S]*)(.*)$',
                                  re.MULTILINE)

self.read()

def read(self):
        try:
            with open(self.filename, "r") as f:
                lines = f.readlines()
        except FileNotFoundError:
            return

for line in lines:
            parsed = self.Line.Parse(line, self.pattern,
                                     self.delimiter, self.quotes)
            self.environment.append(parsed)

def write(self):
        output = ""
        for line in self.environment:
            output = output + line.getLine()

if self.TEST:
            print("========== BEGIN Content of [%s] ==========" % self.filename)
            print(output)
            print("========== END   Content of [%s] ==========\n" % self.filename)
            return

dirname = os.path.dirname(self.filename)
        if not os.path.exists(dirname):
            try:
                os.makedirs(dirname)
            except OSError as exception:
                if exception.errno == errno.EEXIST and os.path.isdir(dirname):
                    pass
                else:
                    raise

with open(self.filename, "w") as f:
            f.write(output)

def get(self, name, default=None):
        value = None
        for line in self.environment:
            if line.isVariable() and line.name == name:
                value = line.value

if value is None:
            return default

if value.lower() in [None, "no", "false", "f", "n"]:
            return False
        elif value.lower() in ["yes", "true", "t", "y"]:
            return True

return value

def getall(self):
        lines = []
        for line in self.environment:
            if line.isVariable():
                lines.append(line)

return lines

def set(self, name, value):
        if type(value) is bool:
            value = "yes" if value else "no"

for line in self.environment:
            if line.isVariable() and line.name == name:
                line.set(name, value)
                return

line = self.Line(self.delimiter, self.quotes)
        line.set(name, value)
        self.environment.append(line)

class Line:
        def __init__(self, delimiter, quotes,
                     name=None, value=None, original=None, fmt=None):
            self.delimiter = delimiter
            self.quotes = quotes
            self.name = name
            self.value = value
            self.original = original
            self.fmt = fmt

def isVariable(self):
            return self.fmt is not None

def isOriginal(self):
            return self.original is not None

def set(self, name, value):
            self.name = name
            self.value = value
            if self.fmt is None:
                self.fmt = "${name}%s${value}\n" % self.delimiter

def getLine(self):
            if self.isOriginal():
                return self.original

value = self.value if self.value is not None else ""
            replacement = {
                'name': self.name,
                'value': self.Escape(value, self.quotes)
            }

line = self.fmt
            for key, value in replacement.items():
                line = line.replace("${" + key + "}", str(value))

return line

@staticmethod
        def Parse(line, pattern, delimiter, quotes):
            match = pattern.match(line)
            if line.startswith('#') or not line.strip() or not match:
                return EnvironmentFile.Line(delimiter, quotes, original=line)

name = match.group(2)
            value = EnvironmentFile.Line.Unescape(match.group(6), quotes)
            fmt = "%s${name}%s%s%s${value}\n" % (match.group(1),
                                                 match.group(3),
                                                 match.group(4),
                                                 match.group(5))

return EnvironmentFile.Line(delimiter, quotes, name=name,
                                        value=value, fmt=fmt)

@staticmethod
        def Escape(value, quotes):
            if value is None:
                return ""

value = str(value)
            value = value.replace("\\", "\\\\")
            value = value.replace("\"", "\\\"")
            value = value.replace("'", "\\\'")
            value = value.replace("$", "\\$")
            value = value.replace("~", "\\~")
            value = value.replace("`", "\\`")

if quotes:
                if value.find(" ") > 0 or value.find("\t") > 0:
                    value = "\"" + value + "\""

return value

@staticmethod
        def Unescape(value, quotes):
            if not value:
                return value

value = str(value)

length = len(value)
            if quotes:
                if (value[0] == "\"" or value[0] == "'") and value[0] == value[length - 1]:
                    value = value[1:length - 1]

i = 0
            while True:
                i = value.find("\\", i)
                if i < 0:
                    break
                if i + 1 >= len(value):
                    value = value[0:i]
                    break

value = value[0:i] + value[i + 1:]
                i += 1

return value

MR SKB