diff options
Diffstat (limited to 'app/policy_create.py')
| -rw-r--r-- | app/policy_create.py | 72 |
1 files changed, 72 insertions, 0 deletions
diff --git a/app/policy_create.py b/app/policy_create.py new file mode 100644 index 0000000..9e89157 --- /dev/null +++ b/app/policy_create.py @@ -0,0 +1,72 @@ +#Author: Shu Shi +#emaiL: shushi@research.att.com + +from toscalib.tosca_workbook import ToscaWorkBook +from toscalib.tosca_builder import ToscaBuilder + +import getopt, sys, json, os, logging + +def usage(): + print('OPTIONS:') + print('\t-h|--help: print this help message') + print('\t-i|--input: The PATH to spec file') + print('\t-o|--output: the output file name') + print('\t-n|--name: the name of the service') + + +def main(): + try: + opts, args = getopt.getopt(sys.argv[1:], "hi:o:n:", ["help", "input=", "output=", "name="]) + except getopt.GetoptError as err: + logging.error( str(err)) # will print something like "option -a not recognized" + usage() + sys.exit(2) + + spec_file = None + output_file = None + name = None + + for o, a in opts: + if o in ("-h", "--help"): + usage() + sys.exit() + elif o in ("-i", "--input"): + spec_file = a + elif o in ("-o", "--output"): + output_file = a + elif o in ("-n", "--name"): + name = a + else: + logging.error( 'Unrecognized option: ' + o) + usage() + sys.exit(2) + + if spec_file is None or output_file is None: + logging.error( 'Incorrect arguments!') + usage() + sys.exit(2) + + dirname = os.path.dirname(output_file) + + if dirname is not None and len(dirname) > 0: + try: + os.stat(dirname) + except: + os.mkdir(dirname) + + meta_model = './data/meta_model/meta_policy_schema.yaml' + + builder = ToscaBuilder() + + builder.import_schema(meta_model) + builder.import_spec(spec_file) + if name is None: + name = builder.spec_import.name + if builder._using_policy() is False: + logging.warning( 'NO policy is defined in the spec') + return + builder.create_policy() + builder.export_policy(output_file) + +if __name__ == "__main__": + main()
\ No newline at end of file |