fixeria
/
pycrate
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

tools: handle exceptions gracefully in the asn1compile app

This commit is contained in:
p1-bmu 2020-09-16 15:14:12 +02:00
parent 861f869ad7
commit d50b0139a3
1 changed files with 22 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
tools/pycrate_asn1compile.py
View File

@ -109,8 +109,6 @@ def main():
# #
args = parser.parse_args() args = parser.parse_args()
# #
generator_class = PycrateGenerator
ckw = {} ckw = {}
if args.fautotags: if args.fautotags:
ckw['autotags'] = True ckw['autotags'] = True
@ -118,9 +116,12 @@ def main():
ckw['extimpl'] = True ckw['extimpl'] = True
if args.fverifwarn: if args.fverifwarn:
ckw['verifwarn'] = True ckw['verifwarn'] = True
#
generator_class = PycrateGenerator
if args.generator_path: if args.generator_path:
generator_class = import_generator_from_file(args.generator_path) generator_class, err = import_generator_from_file(args.generator_path)
if err:
return 0
# #
if args.spec: if args.spec:
if args.spec not in ASN_SPECS: if args.spec not in ASN_SPECS:
@ -190,7 +191,7 @@ def main():
else: else:
print('%s, args warning: invalid input %s' % (sys.argv[0], i)) print('%s, args warning: invalid input %s' % (sys.argv[0], i))
if not files: if not files:
print('%s, args error: no ASN.1 inputs found') print('%s, args error: no ASN.1 inputs found' % sys.argv[0])
return 0 return 0
else: else:
#print(files) #print(files)
@ -229,31 +230,39 @@ def main():
def import_generator_from_file(path): def import_generator_from_file(path):
if not os.path.isfile(path): if not os.path.isfile(path):
raise ValueError('{0} is not a file'.format(path)) print('%s, args error: generator must be a file, %s is not' % (sys.argv[0], path))
return None, 1
# set the directory of the path in the Python module path
module_dir = os.path.dirname(path) module_dir = os.path.dirname(path)
sys.path.append(module_dir) sys.path.append(module_dir)
module_name = path.split(os.path.sep)[-1][:-3:] # get the filename and strip its suffix to get the corresponding module name
# and load it
module_name = '.'.join(path.split(os.path.sep)[-1].split('.')[0:-1])
python_module = __import__(module_name) python_module = __import__(module_name)
# ensure we have an ASN.1 generator
generator_class = find_class_in_module(python_module, mother=_Generator) generator_class = find_class_in_module(python_module, mother=_Generator)
if not generator_class: if generator_class is None:
raise RuntimeError('Could not find an implementation of Abstract class _Generator in {0} module'.format(python_module)) print('%s, args error: generator file does not contain a _Generator class in module %s'\
return generator_class % (sys.argv[0], python_module))
return None, 1
return generator_class, 0
def find_class_in_module(python_module, mother=_Generator): def find_class_in_module(python_module, mother=_Generator):
# look for the 1st _Generator subclass in the module
found_class = None found_class = None
for name, obj in inspect.getmembers(python_module, inspect.isclass): for name, obj in inspect.getmembers(python_module, inspect.isclass):
if obj != mother and issubclass(obj, mother): if obj != mother and issubclass(obj, mother):
found_class = obj found_class = obj
break break
# potentially look for a subclass of the found class
if found_class: if found_class:
child = find_class_in_module(python_module, mother=found_class) child = find_class_in_module(python_module, mother=found_class)
if child: if child:
found_class = child found_class = child
return found_class return found_class
if __name__ == '__main__': if __name__ == '__main__':
sys.exit(main()) sys.exit(main())