67 lines
2.3 KiB
Python
67 lines
2.3 KiB
Python
|
# Copyright 2016 Andreas Florath (andreas@florath.net)
|
||
|
|
#
|
||
|
|
# 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.
|
||
|
|
|
||
|
|
import logging
|
||
|
|
import sys
|
||
|
|
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
class LevelBase(object):
|
||
|
|
|
||
|
|
def __init__(self, lvl, config, default_config, result, sub_modules):
|
||
|
|
self.level = lvl
|
||
|
|
self.config = config
|
||
|
|
self.default_config = default_config
|
||
|
|
self.result = result
|
||
|
|
self.sub_modules = sub_modules
|
||
|
|
|
||
|
|
def call_sub_modules(self, callback):
|
||
|
|
"""Generic way calling submodules"""
|
||
|
|
result = {}
|
||
|
|
if self.result is not None:
|
||
|
|
result = self.result.copy()
|
||
|
|
for name, cfg in self.config:
|
||
|
|
if name in self.sub_modules:
|
||
|
|
logger.info("Calling sub module [%s]" % name)
|
||
|
|
sm = self.sub_modules[name](cfg, self.default_config,
|
||
|
|
self.result)
|
||
|
|
lres = callback(sm)
|
||
|
|
result.update(lres)
|
||
|
|
else:
|
||
|
|
logger.error("Unknown sub module [%s]" % name)
|
||
|
|
sys.exit(1)
|
||
|
|
return result
|
||
|
|
|
||
|
|
def create_cb(self, obj):
|
||
|
|
return obj.create()
|
||
|
|
|
||
|
|
def create(self):
|
||
|
|
"""Create the configured block devices"""
|
||
|
|
logger.info("Starting to create level [%d] block devices" % self.level)
|
||
|
|
result = self.call_sub_modules(self.create_cb)
|
||
|
|
logger.info("Finished creating level [%d] block devices" % self.level)
|
||
|
|
return result
|
||
|
|
|
||
|
|
def delete_cb(self, obj):
|
||
|
|
return obj.delete()
|
||
|
|
|
||
|
|
def delete(self):
|
||
|
|
"""Delete the configured block devices"""
|
||
|
|
logger.info("Starting to delete level [%d] block devices" % self.level)
|
||
|
|
res = self.call_sub_modules(self.delete_cb)
|
||
|
|
logger.info("Finished deleting level [%d] block devices" % self.level)
|
||
|
|
return all(p for p in res.values())
|