| Dir : /lib/python3.6/site-packages/authselect/ |
| Server: Linux ngx353.inmotionhosting.com 4.18.0-553.22.1.lve.1.el8.x86_64 #1 SMP Tue Oct 8 15:52:54 UTC 2024 x86_64 IP: 209.182.202.254 |
| Dir : //lib/python3.6/site-packages/authselect/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