90 lines
3.0 KiB
Python
90 lines
3.0 KiB
Python
# Copyright (C) 2013 Yahoo! Inc. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
from entropy.backends import base
|
|
from entropy import utils
|
|
|
|
|
|
class FileBackend(base.Backend):
|
|
"""A directory based backend."""
|
|
def __init__(self, conf):
|
|
super(FileBackend, self).__init__(conf)
|
|
self._audit_cfg = conf['audit_cfg']
|
|
self._repair_cfg = conf['repair_cfg']
|
|
self.setup()
|
|
|
|
def setup(self):
|
|
utils.create_files([self._audit_cfg, self._repair_cfg])
|
|
|
|
def open(self):
|
|
pass
|
|
|
|
def close(self):
|
|
pass
|
|
|
|
def get_audits(self):
|
|
audits = utils.load_yaml(self._audit_cfg)
|
|
return audits
|
|
|
|
def get_repairs(self):
|
|
repairs = utils.load_yaml(self._repair_cfg)
|
|
return repairs
|
|
|
|
def audit_cfg_from_name(self, name):
|
|
audits = self.get_audits()
|
|
conf = audits[name]['cfg']
|
|
audit_cfg = dict(utils.load_yaml(conf))
|
|
return audit_cfg
|
|
|
|
def repair_cfg_from_name(self, name):
|
|
repairs = self.get_repairs()
|
|
conf = repairs[name]['cfg']
|
|
repair_cfg = dict(utils.load_yaml(conf))
|
|
return repair_cfg
|
|
|
|
def get_script_cfg(self, script_type):
|
|
"""Return the audit/repair cfg file."""
|
|
if script_type == self._audit:
|
|
return self._audit_cfg
|
|
elif script_type == self._repair:
|
|
return self._repair_cfg
|
|
raise TypeError('Script type must be one of: ', [self._audit_cfg,
|
|
self._repair_cfg])
|
|
|
|
def check_script_exists(self, script_type, script_name):
|
|
script_metadata = self.get_script_cfg(script_type)
|
|
scripts = utils.load_yaml(script_metadata)
|
|
return scripts and script_name in scripts
|
|
|
|
def add_script(self, script_type, data):
|
|
script_metadata = self.get_script_cfg(script_type)
|
|
utils.write_yaml(data, script_metadata)
|
|
|
|
@staticmethod
|
|
def remove_script_from_cfg(metadata, name):
|
|
scripts = utils.load_yaml(metadata)
|
|
try:
|
|
scripts.pop(name)
|
|
except KeyError:
|
|
raise
|
|
return scripts
|
|
|
|
def remove_script(self, script_type, script_name):
|
|
try:
|
|
script_metadata = self.get_script_cfg(script_type)
|
|
final_data = FileBackend.remove_script_from_cfg(script_metadata,
|
|
script_name)
|
|
utils.write_yaml(final_data, script_metadata, append=False)
|
|
except Exception:
|
|
raise
|