|
|
@@ -1,6 +1,8 @@
|
|
|
import yaml
|
|
|
import os
|
|
|
|
|
|
+from glob import glob
|
|
|
+from slugify import slugify
|
|
|
from .util import pushd
|
|
|
|
|
|
|
|
|
@@ -9,39 +11,68 @@ class ConfigException(Exception):
|
|
|
|
|
|
|
|
|
class BaseConfig(object):
|
|
|
- def __init__(self, path):
|
|
|
- self.path = path
|
|
|
- if not os.path.exists(path):
|
|
|
- raise ConfigException("Base config file not found: {}".format(path))
|
|
|
-
|
|
|
- with open(path) as f:
|
|
|
- self._data = yaml.load(f, Loader=yaml.SafeLoader)
|
|
|
-
|
|
|
- self.sources = []
|
|
|
- with pushd(os.path.dirname(path)):
|
|
|
- for source in self["sources"]:
|
|
|
- self.sources.append(RepoConfig(self, source))
|
|
|
-
|
|
|
def __getitem__(self, name):
|
|
|
return self._data[name]
|
|
|
|
|
|
def __contains__(self, name):
|
|
|
return name in self._data
|
|
|
|
|
|
+ def __str__(self):
|
|
|
+ return yaml.dump(self._data, default_flow_style=False)
|
|
|
+
|
|
|
+
|
|
|
+class Config(BaseConfig):
|
|
|
+ def __init__(self, path):
|
|
|
+ self.path = os.path.join(path, 'pacman-repo.yml')
|
|
|
+ if not os.path.exists(self.path):
|
|
|
+ self.path = os.path.join(path, 'pacman-repo.yaml')
|
|
|
+ if not os.path.exists(self.path):
|
|
|
+ raise ConfigException("Base config file not found: {}".format(path))
|
|
|
+
|
|
|
+ with open(self.path) as f:
|
|
|
+ self._data = yaml.load(f, Loader=yaml.SafeLoader) or {}
|
|
|
+
|
|
|
+ self.repos = []
|
|
|
+ self.servers = []
|
|
|
+ with pushd(path):
|
|
|
+ for source in self["repos"]:
|
|
|
+ for path in glob('{}/*.yml'.format(source)) + \
|
|
|
+ glob('{}/*.yaml'.format(source)):
|
|
|
+ self.repos.append(RepoConfig(self, path))
|
|
|
+
|
|
|
+ for server in self["servers"]:
|
|
|
+ for path in glob('{}/*.yml'.format(server)) + \
|
|
|
+ glob('{}/*.yaml'.format(server)):
|
|
|
+ self.servers.append(ServerConfig(self, path))
|
|
|
+
|
|
|
|
|
|
-class RepoConfig(object):
|
|
|
+class RepoConfig(BaseConfig):
|
|
|
def __init__(self, config, path):
|
|
|
self.path = path
|
|
|
self.config = config
|
|
|
+ self.id = slugify(path, max_length=255)
|
|
|
+ self.name = os.path.splitext(os.path.basename(path))[0]
|
|
|
|
|
|
if not os.path.exists(path):
|
|
|
raise ConfigException("Repo config file not found: {}".format(path))
|
|
|
|
|
|
with open(path) as f:
|
|
|
- self._data = yaml.load(f, Loader=yaml.SafeLoader)
|
|
|
+ self._data = yaml.load(f, Loader=yaml.SafeLoader) or {}
|
|
|
|
|
|
- def __getitem__(self, name):
|
|
|
- return self._data[name]
|
|
|
+ self._data['name'] = self.name
|
|
|
|
|
|
- def __contains__(self, name):
|
|
|
- return name in self._data
|
|
|
+
|
|
|
+class ServerConfig(BaseConfig):
|
|
|
+ def __init__(self, config, path):
|
|
|
+ self.path = path
|
|
|
+ self.config = config
|
|
|
+ self.id = slugify(path, max_length=255)
|
|
|
+ self.name = os.path.splitext(os.path.basename(path))[0]
|
|
|
+
|
|
|
+ if not os.path.exists(path):
|
|
|
+ raise ConfigException("Repo config file not found: {}".format(path))
|
|
|
+
|
|
|
+ with open(path) as f:
|
|
|
+ self._data = yaml.load(f, Loader=yaml.SafeLoader) or {}
|
|
|
+
|
|
|
+ self._data['name'] = self.name
|
