Merge "Add compass-deck"

1 files changed, 54 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()