paramsFinder.py
79 lines
| 2.0 KiB
| text/x-python
|
PythonLexer
José Chávez
|
r953 | import schainpy | |
| from schainpy.model import Operation, ProcessingUnit | |||
| from pydoc import locate | |||
| def clean_modules(module): | |||
| noEndsUnder = [x for x in module if not x.endswith('__')] | |||
| noStartUnder = [x for x in noEndsUnder if not x.startswith('__')] | |||
| noFullUpper = [x for x in noStartUnder if not x.isupper()] | |||
| return noFullUpper | |||
| def check_module(possible, instance): | |||
| def check(x): | |||
Juan C. Espinoza
|
r1097 | try: | |
|
José Chávez
|
r953 | instancia = locate('schainpy.model.{}'.format(x)) | |
| return isinstance(instancia(), instance) | |||
| except Exception as e: | |||
|
Juan C. Espinoza
|
r1097 | return False | |
|
José Chávez
|
r953 | clean = clean_modules(possible) | |
| return [x for x in clean if check(x)] | |||
| def getProcs(): | |||
|
Juan C. Espinoza
|
r1097 | module = dir(schainpy.model) | |
|
José Chávez
|
r953 | procs = check_module(module, ProcessingUnit) | |
| try: | |||
| procs.remove('ProcessingUnit') | |||
| except Exception as e: | |||
| pass | |||
| return procs | |||
| def getOperations(): | |||
|
Juan C. Espinoza
|
r1097 | module = dir(schainpy.model) | |
|
José Chávez
|
r953 | noProcs = [x for x in module if not x.endswith('Proc')] | |
| operations = check_module(noProcs, Operation) | |||
| try: | |||
| operations.remove('Operation') | |||
| except Exception as e: | |||
| pass | |||
| return operations | |||
| def getArgs(op): | |||
| module = locate('schainpy.model.{}'.format(op)) | |||
| args = module().getAllowedArgs() | |||
| try: | |||
| args.remove('self') | |||
| except Exception as e: | |||
| pass | |||
| try: | |||
| args.remove('dataOut') | |||
| except Exception as e: | |||
| pass | |||
| return args | |||
| def getAll(): | |||
|
Juan C. Espinoza
|
r1097 | allModules = dir(schainpy.model) | |
|
José Chávez
|
r953 | modules = check_module(allModules, Operation) | |
| modules.extend(check_module(allModules, ProcessingUnit)) | |||
| return modules | |||
| def formatArgs(op): | |||
| args = getArgs(op) | |||
| argsAsKey = ["\t'{}'".format(x) for x in args] | |||
| argsFormatted = ": 'string',\n".join(argsAsKey) | |||
| print op | |||
| print "parameters = { \n" + argsFormatted + ": 'string',\n }" | |||
| print '\n' | |||
| if __name__ == "__main__": | |||
| getAll() | |||
| [formatArgs(x) for x in getAll()] | |||
| ''' | |||
| parameters = { | |||
| 'id': , | |||
| 'wintitle': , | |||
| } | |||
| ''' |