cli.py
168 lines
| 5.6 KiB
| text/x-python
|
PythonLexer
José Chávez
|
r934 | import click | ||
| import schainpy | ||||
|
José Chávez
|
r935 | import subprocess | ||
|
José Chávez
|
r936 | import os | ||
| import sys | ||||
| import glob | ||||
|
José Chávez
|
r944 | save_stdout = sys.stdout | ||
| sys.stdout = open('trash', 'w') | ||||
|
José Chávez
|
r943 | from multiprocessing import cpu_count | ||
|
José Chávez
|
r1047 | from schainpy.controller import Project | ||
|
José Chávez
|
r944 | from schainpy.model import Operation, ProcessingUnit | ||
|
José Chávez
|
r943 | from schainpy.utils import log | ||
|
José Chávez
|
r944 | from importlib import import_module | ||
| from pydoc import locate | ||||
| from fuzzywuzzy import process | ||||
|
José Chávez
|
r953 | from schainpy.utils import paramsFinder | ||
Juan C. Espinoza
|
r1113 | import templates | ||
|
José Chávez
|
r944 | sys.stdout = save_stdout | ||
|
José Chávez
|
r934 | |||
| def print_version(ctx, param, value): | ||||
| if not value or ctx.resilient_parsing: | ||||
| return | ||||
| click.echo(schainpy.__version__) | ||||
| ctx.exit() | ||||
|
José Chávez
|
r944 | |||
|
José Chávez
|
r943 | cliLogger = log.makelogger('schain cli') | ||
|
José Chávez
|
r944 | PREFIX = 'experiment' | ||
|
José Chávez
|
r934 | |||
| @click.command() | ||||
| @click.option('--version', '-v', is_flag=True, callback=print_version, help='SChain version', type=str) | ||||
| @click.argument('command', default='run', required=True) | ||||
|
José Chávez
|
r944 | @click.argument('nextcommand', default=None, required=False, type=str) | ||
|
José Chávez
|
r1081 | def main(command, nextcommand, version): | ||
|
José Chávez
|
r946 | """COMMAND LINE INTERFACE FOR SIGNAL CHAIN - JICAMARCA RADIO OBSERVATORY \n | ||
| Available commands.\n | ||||
| --xml: runs a schain XML generated file\n | ||||
| run: runs any python script starting 'experiment_'\n | ||||
| generate: generates a template schain script\n | ||||
| search: return avilable operations, procs or arguments of the give operation/proc\n""" | ||||
|
José Chávez
|
r1081 | if command == 'xml': | ||
| runFromXML(nextcommand) | ||||
|
José Chávez
|
r935 | elif command == 'generate': | ||
|
José Chávez
|
r934 | generate() | ||
| elif command == 'test': | ||||
| test() | ||||
|
José Chávez
|
r936 | elif command == 'run': | ||
|
José Chávez
|
r944 | runschain(nextcommand) | ||
| elif command == 'search': | ||||
| search(nextcommand) | ||||
| else: | ||||
| log.error('Command {} is not defined'.format(command)) | ||||
|
José Chávez
|
r1081 | |||
|
José Chávez
|
r944 | def check_module(possible, instance): | ||
| def check(x): | ||||
| try: | ||||
| instancia = locate('schainpy.model.{}'.format(x)) | ||||
| return isinstance(instancia(), instance) | ||||
| except Exception as e: | ||||
| return False | ||||
| clean = clean_modules(possible) | ||||
| return [x for x in clean if check(x)] | ||||
| 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 search(nextcommand): | ||||
| if nextcommand is None: | ||||
| log.error('There is no Operation/ProcessingUnit to search') | ||||
| elif nextcommand == 'procs': | ||||
|
José Chávez
|
r953 | procs = paramsFinder.getProcs() | ||
|
José Chávez
|
r1081 | log.success( | ||
| 'Current ProcessingUnits are:\n\033[1m{}\033[0m'.format('\n'.join(procs))) | ||||
|
José Chávez
|
r944 | |||
| elif nextcommand == 'operations': | ||||
|
José Chávez
|
r953 | operations = paramsFinder.getOperations() | ||
|
José Chávez
|
r1081 | log.success('Current Operations are:\n\033[1m{}\033[0m'.format( | ||
| '\n'.join(operations))) | ||||
|
José Chávez
|
r944 | else: | ||
| try: | ||||
|
José Chávez
|
r953 | args = paramsFinder.getArgs(nextcommand) | ||
|
José Chávez
|
r1081 | log.warning( | ||
| 'Use this feature with caution. It may not return all the allowed arguments') | ||||
|
José Chávez
|
r945 | if len(args) == 0: | ||
| log.success('{} has no arguments'.format(nextcommand)) | ||||
| else: | ||||
|
José Chávez
|
r1081 | log.success('Showing {} arguments:\n\033[1m{}\033[0m'.format( | ||
| nextcommand, '\n'.join(args))) | ||||
|
José Chávez
|
r944 | except Exception as e: | ||
| log.error('Module {} does not exists'.format(nextcommand)) | ||||
|
José Chávez
|
r953 | allModules = paramsFinder.getAll() | ||
| similar = process.extractOne(nextcommand, allModules)[0] | ||||
| log.success('Showing {} instead'.format(similar)) | ||||
|
José Chávez
|
r944 | search(similar) | ||
| def runschain(nextcommand): | ||||
| if nextcommand is None: | ||||
| currentfiles = glob.glob('./{}_*.py'.format(PREFIX)) | ||||
| numberfiles = len(currentfiles) | ||||
| if numberfiles > 1: | ||||
| log.error('There is more than one file to run') | ||||
| elif numberfiles == 1: | ||||
| subprocess.call(['python ' + currentfiles[0]], shell=True) | ||||
|
José Chávez
|
r936 | else: | ||
|
José Chávez
|
r944 | log.error('There is no file to run') | ||
|
José Chávez
|
r934 | else: | ||
|
José Chávez
|
r944 | try: | ||
| subprocess.call(['python ' + nextcommand], shell=True) | ||||
| except Exception as e: | ||||
| log.error("I cannot run the file. Does it exists?") | ||||
|
José Chávez
|
r934 | |||
|
José Chávez
|
r939 | |||
|
José Chávez
|
r935 | def basicInputs(): | ||
|
José Chávez
|
r934 | inputs = {} | ||
|
José Chávez
|
r1081 | inputs['desc'] = click.prompt( | ||
| 'Enter a description', default="A schain project", type=str) | ||||
| inputs['name'] = click.prompt( | ||||
| 'Name of the project', default="project", type=str) | ||||
| inputs['path'] = click.prompt('Data path', default=os.getcwd( | ||||
| ), type=click.Path(exists=True, resolve_path=True)) | ||||
| inputs['startDate'] = click.prompt( | ||||
| 'Start date', default='1970/01/01', type=str) | ||||
| inputs['endDate'] = click.prompt( | ||||
| 'End date', default='2017/12/31', type=str) | ||||
| inputs['startHour'] = click.prompt( | ||||
| 'Start hour', default='00:00:00', type=str) | ||||
|
José Chávez
|
r934 | inputs['endHour'] = click.prompt('End hour', default='23:59:59', type=str) | ||
|
José Chávez
|
r935 | inputs['figpath'] = inputs['path'] + '/figs' | ||
| return inputs | ||||
|
José Chávez
|
r939 | |||
|
José Chávez
|
r935 | def generate(): | ||
| inputs = basicInputs() | ||||
| inputs['multiprocess'] = click.confirm('Is this a multiprocess script?') | ||||
| if inputs['multiprocess']: | ||||
|
José Chávez
|
r1081 | inputs['nProcess'] = click.prompt( | ||
| 'How many process?', default=cpu_count(), type=int) | ||||
|
José Chávez
|
r935 | current = templates.multiprocess.format(**inputs) | ||
| else: | ||||
| current = templates.basic.format(**inputs) | ||||
|
José Chávez
|
r944 | scriptname = '{}_{}.py'.format(PREFIX, inputs['name']) | ||
|
José Chávez
|
r935 | script = open(scriptname, 'w') | ||
| try: | ||||
| script.write(current) | ||||
|
José Chávez
|
r944 | log.success('Script {} generated'.format(scriptname)) | ||
|
José Chávez
|
r935 | except Exception as e: | ||
|
José Chávez
|
r943 | log.error('I cannot create the file. Do you have writing permissions?') | ||
|
José Chávez
|
r935 | |||
|
José Chávez
|
r934 | |||
| def test(): | ||||
|
José Chávez
|
r943 | log.warning('testing') | ||
|
José Chávez
|
r939 | |||
| def runFromXML(filename): | ||||
|
José Chávez
|
r1047 | controller = Project() | ||
|
José Chávez
|
r939 | if not controller.readXml(filename): | ||
| return | ||||
| controller.start() | ||||
| return | ||||