policy2tosca patch set-6

Signed-off-by: shiva-charan <shiva-charan.m-s@hpe.com>

6 files changed, 407 insertions, 0 deletions

diff --git a/policy2tosca/policy2tosca/__init__.py b/policy2tosca/policy2tosca/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/policy2tosca/policy2tosca/__init__.py
diff --git a/policy2tosca/policy2tosca/add_definition.py b/policy2tosca/policy2tosca/add_definition.py
new file mode 100644
index 0000000..cb534f4
--- /dev/null
+++ b/policy2tosca/policy2tosca/add_definition.py
@@ -0,0 +1,157 @@
+# 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 os
+import re
+from cliff.command import Command
+from cliff.show import ShowOne
+import yaml
+import json
+
+
+class Add_definition(Command):
+    "Add policy definition in a tosca file"
+
+    log = logging.getLogger(__name__)
+
+    def get_parser(self, prog_name):
+        parser = super(Add_definition, self).get_parser(prog_name)
+        parser.add_argument('--policy_name', dest='policy_name', required=True,
+                            help='add policy name')
+        parser.add_argument('--policy_type', dest='policy_type', required=True,
+                            help='belongs to policy type')
+        parser.add_argument('--targets', dest='targets', nargs='*',
+                            help='specify N number of targets')
+        parser.add_argument('--metadata', dest='metadata', nargs='?',
+                            help='metadata')
+        parser.add_argument('--description', dest='description', nargs='?',
+                            help='description')
+        parser.add_argument('--properties', dest='properties', nargs='+',
+                            help='properties')
+        parser.add_argument('--triggers', dest='triggers', nargs='*',
+                            help='triggers')
+        parser.add_argument('--source', dest='source', nargs='?',
+                            help='source tosca yaml file')
+        return parser
+
+    def take_action(self, parsed_args):
+        inputfile = open(parsed_args.source, 'r').readlines()
+        policy_def = 0
+        policy_type = 0
+        policy_type = self.check_policy_type(parsed_args)
+        policy_def = self.check_definition(parsed_args)
+        if policy_def == 0 and policy_type == 0:
+            self.insert_policy_def(parsed_args)
+
+    def check_definition(self, parsed_args):
+        filename = parsed_args.source
+        filehandler = open(filename, "r")
+        lines = filehandler.readlines()
+        policy_def = 0
+        policy_name = "    - " + str(parsed_args.policy_name) + ":"
+        for line in lines:
+            if policy_name in line:
+                print "policy definition exists"
+                policy_def = 1
+        return policy_def
+
+    def insert_policy_def(self, parsed_args):
+        searchline = '  policies:\n'
+        f = open(parsed_args.source, "r")
+        lines = f.readlines()
+        num = lines.index(searchline)
+        string = lines[num + 1]
+        leading_spaces = len(string) - len(string.lstrip())
+        policy_name = "- " + str(parsed_args.policy_name) + ":"
+        properties = parsed_args.properties[0]
+        prop_str = self.make_list(properties,leading_spaces)
+        if (parsed_args.triggers and parsed_args.targets):
+            policy = "".join((leading_spaces * " ") + policy_name  + "\n"
+            + "".join((leading_spaces * " ")) + "  type: "
+            + str(parsed_args.policy_type) + "\n"  + "".join((leading_spaces * " "))
+            + "  description: " + str(parsed_args.description) + "\n"
+            + "".join((leading_spaces * " ")) + "  metadata: " + str(parsed_args.metadata) + "\n"
+            + "".join((leading_spaces * " ")) + "  properties: \n" + "".join((leading_spaces * 2 * " ")) + prop_str
+            + "targets: " + str(parsed_args.targets) + "\n"
+            + "".join((leading_spaces * " ")) + "  triggers: " + str(parsed_args.triggers) + "\n"
+            )
+        elif (parsed_args.triggers):
+            policy = "".join((leading_spaces * " ") + policy_name  + "\n"
+            + "".join((leading_spaces * " ")) + "  type: "
+            + str(parsed_args.policy_type) + "\n"  + "".join((leading_spaces * " "))
+            + "  description: " + str(parsed_args.description) + "\n"
+            + "".join((leading_spaces * " ")) + "  metadata: " + str(parsed_args.metadata) + "\n"
+            + "".join((leading_spaces * " ")) + "  properties: \n" + "".join((leading_spaces * 2 * " ")) + prop_str
+            + "".join((leading_spaces * " ")) + "  triggers: " + str(parsed_args.triggers) + "\n"
+            )
+        elif (parsed_args.targets):
+            policy = "".join((leading_spaces * " ") + policy_name  + "\n"
+            + "".join((leading_spaces * " ")) + "  type: "
+            + str(parsed_args.policy_type) + "\n"  + "".join((leading_spaces * " "))
+            + "  description: " + str(parsed_args.description) + "\n"
+            + "".join((leading_spaces * " ")) + "  metadata: " + str(parsed_args.metadata) + "\n"
+            + "".join((leading_spaces * " ")) + "  properties: \n" + "".join((leading_spaces * 2 * " ")) + prop_str
+            + "targets: " + str(parsed_args.targets) + "\n"
+            )
+
+        lines.insert(num+1,policy)
+
+        f2 = open(parsed_args.source, "w")
+        for line in lines:
+            f2.write(line)
+
+    def make_list(self, properties, leading_spaces):
+        props = properties.split(",")
+        lists = []
+        i = 0
+        for p in props:
+            lables = p.split(":")
+
+            lists.insert(i,lables)
+            i = i + 1
+        title_list=[]
+        i =0
+        for l in lists:
+            title_list.insert(i,l[0])
+            i = i + 1
+        prop_title = ""
+        len_list = len(lists)
+        title_list =  set(title_list)
+        final_list = []
+        i = 0
+
+        for t in title_list:
+            final_str = t + ":"
+            for l in lists:
+                if t == l[0]:
+                    final_str = final_str +  " " + l[1] + ","
+            final_str = final_str[:-1]
+            final_list.insert(i,final_str)
+            i =+ 1
+
+        prop_str = ""
+        for f in final_list:
+            prop_str = prop_str + f +"\n" + "".join((leading_spaces * 2 *" "))
+        return prop_str[:-1]
+
+    def check_policy_type(self, parsed_args):
+        policy_type = parsed_args.policy_type + ":$"
+        inputfile = open(parsed_args.source, 'r').readlines()
+        policy_exists = 0
+        for line in inputfile:
+            if re.search(policy_type, line):
+                print "policy type exists"
+                policy_exists = 1
+                break
+        return policy_exists
diff --git a/policy2tosca/policy2tosca/add_type.py b/policy2tosca/policy2tosca/add_type.py
new file mode 100644
index 0000000..ded3732
--- /dev/null
+++ b/policy2tosca/policy2tosca/add_type.py
@@ -0,0 +1,121 @@
+# 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 os
+from cliff.command import Command
+from cliff.show import ShowOne
+import re
+
+
+class Add_type(Command):
+
+    "Add policy type in a tosca file"
+    log = logging.getLogger(__name__)
+
+    def get_parser(self, prog_name):
+        parser = super(Add_type, self).get_parser(prog_name)
+        parser.add_argument('--policy_type', dest='policy_type', nargs='?',
+                            help='add policy type')
+        parser.add_argument('--derived_from', dest='derived_from', nargs='?',
+                            help='policy dervied from')
+        parser.add_argument('--version_number', dest='version_number',
+                            nargs='?', help='version number of the policy')
+        parser.add_argument('--targets', dest='targets', nargs='+',
+                            help='specify N number of targets')
+        parser.add_argument('--metadata', dest='metadata', nargs='?',
+                            help='metadata')
+        parser.add_argument('--description', dest='description', nargs='?',
+                            help='metadata')
+        parser.add_argument('--properties', dest='properties', nargs='+',
+                            help='properties')
+        parser.add_argument('--triggers', dest='triggers', nargs='+',
+                            help='triggers')
+        parser.add_argument('--source', dest='source', nargs='?',
+                            help='source tosca yaml file')
+        return parser
+
+    def take_action(self, parsed_args):
+        inputfile = open(parsed_args.source, 'r').readlines()
+        policy_block = 0
+        for line in inputfile:
+            if re.search('policy_types:', line):
+                print "policy block exists"
+                policy_block = 1
+        if policy_block == 0:
+            self.insert_policy_block(parsed_args)
+            policy_block = 1
+        policy_exists = self.check_policy(parsed_args)
+        if policy_exists == 0:
+            self.insert_policy(parsed_args)
+
+    def insert_policy_block(self, parsed_args):
+        inputfile = open(parsed_args.source, 'r').readlines()
+        filename = parsed_args.source + "_temp"
+        filehandle = open(filename, 'w')
+        for line in inputfile:
+            filehandle.write(line)
+            if re.search('topology_template:', line):
+                print "Inserting policy block"
+                filehandle.write("\tpolicy_types:\n")
+
+        filehandle.close()
+        filehandles = open(parsed_args.source, "w")
+        filename = parsed_args.source + "_temp"
+        filehandle = open(filename, 'r')
+        lines = filehandle.readlines()
+        for line in lines:
+            filehandles.write(line)
+        os.system('rm '+filename)
+
+    def check_policy(self, parsed_args):
+        policy_name = parsed_args.policy_type + ":$"
+        inputfile = open(parsed_args.source, 'r').readlines()
+        policy_exists = 0
+        for line in inputfile:
+            if re.search(policy_name, line):
+                print "policy type exists"
+                policy_exists = 1
+                break
+        return policy_exists
+
+    def insert_policy(self, parsed_args):
+        inputfile = open(parsed_args.source, 'r').readlines()
+        filename = parsed_args.source + "_temp"
+        filehandle = open(filename, 'w')
+        for line in inputfile:
+            filehandle.write(line)
+            if re.search('policy_types:', line):
+                print "Inserting policy type"
+                filehandle.write("\t  "+str(parsed_args.policy_type) + ":\n")
+                filehandle.write("\t    derived_from: "
+                                 + str(parsed_args.derived_from) + "\n")
+                filehandle.write("\t    version: " +
+                                 str(parsed_args.version_number) + "\n")
+                filehandle.write("\t    description: " +
+                                 str(parsed_args.description) + "\n")
+                filehandle.write("\t    metadata: \n\t        " +
+                                 str(parsed_args.metadata) + "\n")
+                filehandle.write("\t    properties: \n\t        " +
+                                 str(parsed_args.properties) + "\n")
+                filehandle.write("\t    targets:\n\t        " +
+                                 str(parsed_args.targets) + "\n")
+                filehandle.write("\t    triggers: \n\t        " +
+                                 str(parsed_args.triggers) + "\n")
+        filehandle.close()
+        filehandles = open(parsed_args.source, "w")
+        filehandle = open(filename, 'r')
+        lines = filehandle.readlines()
+        for line in lines:
+            filehandles.write(line)
+        os.system('rm ' + filename)
diff --git a/policy2tosca/policy2tosca/del_definition.py b/policy2tosca/policy2tosca/del_definition.py
new file mode 100644
index 0000000..d8345cd
--- /dev/null
+++ b/policy2tosca/policy2tosca/del_definition.py
@@ -0,0 +1,45 @@
+# 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 os
+from cliff.command import Command
+from cliff.show import ShowOne
+import yaml
+import json
+
+
+class Delete_definition(Command):
+    "Delete policy in a tosca file"
+
+    log = logging.getLogger(__name__)
+
+    def get_parser(self, prog_name):
+        parser = super(Delete_definition, self).get_parser(prog_name)
+        parser.add_argument('--policy_name', dest='policy_name', nargs='?',
+                            help='delete policy')
+        parser.add_argument('--source', dest='source', nargs='?',
+                            help='source tosca yaml file')
+        return parser
+
+    def take_action(self, parsed_args):
+        yml = open(parsed_args.source, "r")
+        data = yaml.load(yml)
+        json_dump = json.dumps(data)
+        tosca_json = json.loads(json_dump)
+        del tosca_json['topology_template']['policies'][parsed_args.policy_name]
+        tosca = json.dumps(tosca_json)
+        data2 = json.loads(tosca)
+        yml2 = yaml.safe_dump(data2, default_flow_style=False)
+        tosca_file = open(parsed_args.source, "w")
+        tosca_file.write(yml2)
diff --git a/policy2tosca/policy2tosca/del_type.py b/policy2tosca/policy2tosca/del_type.py
new file mode 100644
index 0000000..144deec
--- /dev/null
+++ b/policy2tosca/policy2tosca/del_type.py
@@ -0,0 +1,47 @@
+# 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 os
+from cliff.command import Command
+from cliff.show import ShowOne
+import yaml
+import json
+
+
+class Delete_type(Command):
+    "Delete policy type in a tosca file"
+
+    log = logging.getLogger(__name__)
+
+    def get_parser(self, prog_name):
+        parser = super(Delete_type, self).get_parser(prog_name)
+        parser.add_argument('--policy_type', dest='policy_type', nargs='?',
+                            help='delete policy type')
+        parser.add_argument('--source', dest='source', nargs='?',
+                            help='source tosca yaml file')
+        return parser
+
+    def take_action(self, parsed_args):
+        yml = open(parsed_args.source, "r")
+        #lines=filehandle.readlines()
+        data = yaml.load(yml)
+        json_dump = json.dumps(data)
+        #print json_dump
+        tosca_json = json.loads(json_dump)
+        del tosca_json['topology_template']['policy_types'][parsed_args.policy_type]
+        tosca = json.dumps(tosca_json)
+        data2 = json.loads(tosca)
+        yml2 = yaml.safe_dump(data2, default_flow_style=False)
+        tosca_file = open(parsed_args.source, "w")
+        tosca_file.write(yml2)
diff --git a/policy2tosca/policy2tosca/main.py b/policy2tosca/policy2tosca/main.py
new file mode 100644
index 0000000..39aabfa
--- /dev/null
+++ b/policy2tosca/policy2tosca/main.py
@@ -0,0 +1,37 @@
+import logging
+import sys
+
+from cliff.app import App
+from cliff.commandmanager import CommandManager
+
+
+class policy2tosca(App):
+
+    log = logging.getLogger(__name__)
+
+    def __init__(self):
+        super(policy2tosca, self).__init__(
+            description='Parser Policy2Tosca',
+            version='0.1',
+            command_manager=CommandManager('cliff.demo'),
+            )
+
+    def initialize_app(self, argv):
+        self.log.debug('initialize_app')
+
+    def prepare_to_run_command(self, cmd):
+        self.log.debug('prepare_to_run_command %s', cmd.__class__.__name__)
+
+    def clean_up(self, cmd, result, err):
+        self.log.debug('clean_up %s', cmd.__class__.__name__)
+        if err:
+            self.log.debug('got an error: %s', err)
+
+
+def main(argv=sys.argv[1:]):
+    myapp = policy2tosca()
+    return myapp.run(argv)
+
+
+if __name__ == '__main__':
+    sys.exit(main(sys.argv[1:]))