Merge "Add compass-deck"

6 files changed, 212 insertions, 0 deletions

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