From 905b0231e93ce2409a45dd6c4f5f983689fdb790 Mon Sep 17 00:00:00 2001 From: Harry Huang Date: Wed, 1 Nov 2017 11:56:50 +0800 Subject: Add compass-deck RESTful API and DB Handlers for Compass Change-Id: I1ce411f279943764c286ea48dca9185d453cf254 Signed-off-by: Harry Huang --- compass-deck/bin/chef/addcookbooks.py | 54 ++++++++++++++++++++++ compass-deck/bin/chef/adddatabags.py | 71 +++++++++++++++++++++++++++++ compass-deck/bin/chef/addroles.py | 62 +++++++++++++++++++++++++ compass-deck/bin/chef/clean_clients.sh | 6 +++ compass-deck/bin/chef/clean_environments.sh | 13 ++++++ compass-deck/bin/chef/clean_nodes.sh | 6 +++ 6 files changed, 212 insertions(+) create mode 100755 compass-deck/bin/chef/addcookbooks.py create mode 100755 compass-deck/bin/chef/adddatabags.py create mode 100755 compass-deck/bin/chef/addroles.py create mode 100755 compass-deck/bin/chef/clean_clients.sh create mode 100755 compass-deck/bin/chef/clean_environments.sh create mode 100755 compass-deck/bin/chef/clean_nodes.sh (limited to 'compass-deck/bin/chef') diff --git a/compass-deck/bin/chef/addcookbooks.py b/compass-deck/bin/chef/addcookbooks.py new file mode 100755 index 0000000..f23dac4 --- /dev/null +++ b/compass-deck/bin/chef/addcookbooks.py @@ -0,0 +1,54 @@ +#!/usr/bin/env python +# +# Copyright 2014 Huawei Technologies Co. Ltd +# +# 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 cookbooks to chef server.""" +import logging +import os +import os.path +import sys + + +current_dir = os.path.dirname(os.path.realpath(__file__)) +sys.path.append(current_dir) + + +import switch_virtualenv + + +from compass.utils import flags +from compass.utils import logsetting + + +flags.add('cookbooks_dir', + help='chef cookbooks directory', + default='/var/chef/cookbooks') + + +def main(): + """main entry.""" + flags.init() + logsetting.init() + cookbooks_dir = flags.OPTIONS.cookbooks_dir + logging.info('add cookbooks %s', cookbooks_dir) + cmd = "knife cookbook upload --all --cookbook-path %s" % cookbooks_dir + status = os.system(cmd) + logging.info('run cmd %s returns %s', cmd, status) + if status: + sys.exit(1) + + +if __name__ == '__main__': + main() diff --git a/compass-deck/bin/chef/adddatabags.py b/compass-deck/bin/chef/adddatabags.py new file mode 100755 index 0000000..ba2d08c --- /dev/null +++ b/compass-deck/bin/chef/adddatabags.py @@ -0,0 +1,71 @@ +#!/usr/bin/env python +# +# Copyright 2014 Huawei Technologies Co. Ltd +# +# 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 databags to chef server.""" +import logging +import os +import os.path +import sys + + +current_dir = os.path.dirname(os.path.realpath(__file__)) +sys.path.append(current_dir) + + +import switch_virtualenv + +from compass.utils import flags +from compass.utils import logsetting + + +flags.add('databags_dir', + help='chef databags directory', + default='/var/chef/databags') + + +def main(): + """main entry.""" + flags.init() + logsetting.init() + databags = [] + databags_dir = flags.OPTIONS.databags_dir + for item in os.listdir(databags_dir): + databags.append(item) + + for databag in databags: + logging.info('add databag %s', databag) + cmd = "knife data bag create %s" % databag + os.system(cmd) + databag_items = [] + databagitem_dir = os.path.join(databags_dir, databag) + for item in os.listdir(databagitem_dir): + if item.endswith('.json'): + databag_items.append(os.path.join(databagitem_dir, item)) + else: + logging.info('ignore %s in %s', item, databagitem_dir) + + for databag_item in databag_items: + logging.info('add databag item %s to databag %s', + databag_item, databag) + cmd = 'knife data bag from file %s %s' % (databag, databag_item) + status = os.system(cmd) + logging.info('run cmd %s returns %s', cmd, status) + if status: + sys.exit(1) + + +if __name__ == '__main__': + main() diff --git a/compass-deck/bin/chef/addroles.py b/compass-deck/bin/chef/addroles.py new file mode 100755 index 0000000..2745506 --- /dev/null +++ b/compass-deck/bin/chef/addroles.py @@ -0,0 +1,62 @@ +#!/usr/bin/env python +# +# Copyright 2014 Huawei Technologies Co. Ltd +# +# 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. + +"""script to import roles to chef server.""" +import logging +import os +import os.path +import sys + + +current_dir = os.path.dirname(os.path.realpath(__file__)) +sys.path.append(current_dir) + + +import switch_virtualenv + +from compass.utils import flags +from compass.utils import logsetting + + +flags.add('roles_dir', + help='chef roles directory', + default='/var/chef/roles') + + +def main(): + """main entry.""" + flags.init() + logsetting.init() + rolelist = [] + roles_dir = flags.OPTIONS.roles_dir + + for item in os.listdir(roles_dir): + if item.endswith('.rb') or item.endswith('.json'): + rolelist.append(os.path.join(roles_dir, item)) + else: + logging.info('ignore %s in %s', item, roles_dir) + + for role in rolelist: + logging.info('add role %s', role) + cmd = "knife role from file %s" % role + status = os.system(cmd) + logging.info('run cmd %s returns %s', cmd, status) + if status: + sys.exit(1) + + +if __name__ == '__main__': + main() diff --git a/compass-deck/bin/chef/clean_clients.sh b/compass-deck/bin/chef/clean_clients.sh new file mode 100755 index 0000000..7a26bea --- /dev/null +++ b/compass-deck/bin/chef/clean_clients.sh @@ -0,0 +1,6 @@ +#!/bin/bash +echo "clean chef clients" +yes | knife client bulk delete '^(?!chef-).*' +if [[ "$?" != "0" ]]; then + echo "failed to clean all clients" +fi diff --git a/compass-deck/bin/chef/clean_environments.sh b/compass-deck/bin/chef/clean_environments.sh new file mode 100755 index 0000000..f9b5052 --- /dev/null +++ b/compass-deck/bin/chef/clean_environments.sh @@ -0,0 +1,13 @@ +#!/bin/bash +echo "clean chef environments" +environments=$(knife environment list) +for environment in $environments; do + if [[ "$environment" != "_default" ]]; then + yes | knife environment delete $environment + if [[ "$?" != "0" ]]; then + echo "failed to delete environment $environment" + else + echo "environment $environment is deleted" + fi + fi +done diff --git a/compass-deck/bin/chef/clean_nodes.sh b/compass-deck/bin/chef/clean_nodes.sh new file mode 100755 index 0000000..8224b82 --- /dev/null +++ b/compass-deck/bin/chef/clean_nodes.sh @@ -0,0 +1,6 @@ +#!/bin/bash +echo "clean chef nodes" +yes | knife node bulk delete '.*' +if [[ "$?" != "0" ]]; then + echo "failed to clean all nodes" +fi -- cgit 1.2.3-korg