shell bypass 403
import os.path
from copy import deepcopy
from ..config import Configuration
from .base import Source
__all__ = (
'HomeDirectory',
'ConfigDirectory',
'FileBasedSource',
)
class DirectoryModifier(object):
def __init__(self, target_file):
self.target_file = target_file
def __call__(self):
raise NotImplementedError()
class HomeDirectory(DirectoryModifier):
def __call__(self):
return os.path.expanduser(
os.path.join(
'~',
self.target_file,
)
)
class ConfigDirectory(DirectoryModifier):
def __call__(self):
config_dir = os.getenv('XDG_CONFIG_HOME') \
or os.path.expanduser(
os.path.join(
'~',
'.config',
)
)
return os.path.join(config_dir, self.target_file)
class FileBasedSource(Source):
def __init__(self, files, base_path=None, combine=False):
super(FileBasedSource, self).__init__()
if isinstance(files, (str, DirectoryModifier)):
files = [files]
elif not isinstance(files, (tuple, list)):
raise TypeError('files must be a string or list of strings')
self.files = []
for target in files:
if isinstance(target, str):
self.files.append(target)
elif isinstance(target, DirectoryModifier):
self.files.append(target())
else:
raise TypeError('files must be a string or list of strings')
self.base_path = base_path or os.getcwd()
self.combine = combine
def get_config(self, settings, manager=None, parent=None):
parsed_settings = []
for file_source in self.files:
if os.path.isabs(file_source):
file_path = file_source
else:
file_path = os.path.join(self.base_path, file_source)
if os.path.exists(file_path):
file_settings = self.get_settings_from_file(
file_path,
deepcopy(settings),
manager=manager,
)
if file_settings:
parsed_settings.append(file_settings)
if not self.combine:
# No need to gather any more, we only want one.
break
if parsed_settings:
config = parent
for parsed_setting in reversed(parsed_settings):
config = Configuration(
settings=parsed_setting,
parent=config,
)
else:
config = Configuration(settings=settings, parent=parent)
return config
def get_settings_from_file(self, file_path, settings, manager=None):
raise NotImplementedError()