diff options
author | shiva-charan.m-s <shiva-charan.m-s@hpe.com> | 2016-07-26 22:34:23 -0500 |
---|---|---|
committer | shiva-charan <shiva-charan.m-s@hpe.com> | 2016-08-22 12:32:45 -0500 |
commit | d93490753ba93e4c3aa00cd487855e3d2970ae9a (patch) | |
tree | 9129d35d1df3c36f206d37b657a9100a39a94952 /policy2tosca/policy2tosca | |
parent | 1a9b3ea86e026115938b1872ac1fc3bce30055ed (diff) |
policy2tosca patch set-6
Signed-off-by: shiva-charan <shiva-charan.m-s@hpe.com>
Diffstat (limited to 'policy2tosca/policy2tosca')
-rw-r--r-- | policy2tosca/policy2tosca/__init__.py | 0 | ||||
-rw-r--r-- | policy2tosca/policy2tosca/add_definition.py | 157 | ||||
-rw-r--r-- | policy2tosca/policy2tosca/add_type.py | 121 | ||||
-rw-r--r-- | policy2tosca/policy2tosca/del_definition.py | 45 | ||||
-rw-r--r-- | policy2tosca/policy2tosca/del_type.py | 47 | ||||
-rw-r--r-- | policy2tosca/policy2tosca/main.py | 37 |
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:])) |