58 lines
2.4 KiB
Python
58 lines
2.4 KiB
Python
# -*- coding: utf-8 -*-
|
||
from application import app,manager
|
||
from flask_script import Command,Option
|
||
import argparse,sys,traceback
|
||
|
||
'''
|
||
python manage.py runjob -m Test ( jobs/tasks/Test.py )
|
||
python manage.py runjob -m test/Index ( jobs/tasks/test/Index.py )
|
||
* name or flags - 名称或选项字符串列表, e.g. foo or -f, --foo.
|
||
* action - 参数如果定义了选项，表示这是一个操作参数，至于调用时做哪种操作由用户输入或者default决定。
|
||
* nargs - 应该使用的命令行参数数。.
|
||
* const - 某些动作或参数个数的常数值。.
|
||
* default - 如果命令行没有对输入这个参数相应的值，则此参数用default给出的值.
|
||
* type -将用户输入的值转化为哪种类型.
|
||
* choices - 参数可输入值的范围或选择.
|
||
* required - 命令行输入的值是否可以被忽略（布尔量）.
|
||
* help - 参数的简要描述.
|
||
* metavar - useage中显示的参数的名称.
|
||
* dest - 要添加到解析参数返回的对象中的属性的名称.
|
||
'''
|
||
class runJob( Command ):
|
||
|
||
capture_all_args = True
|
||
def run(self,*args,**kwargs):
|
||
args = sys.argv[2:]
|
||
parser = argparse.ArgumentParser( add_help = True )
|
||
|
||
parser.add_argument("-m","--name",dest = "name",metavar = "name", help="指定job名",required=True)
|
||
parser.add_argument("-a","--act",dest = "act",metavar = "act", help="Job动作",required=False)
|
||
parser.add_argument("-p","--param",dest = "param",nargs = "*", metavar = "param",help="业务参数",default = '',required=False)
|
||
params = parser.parse_args( args )
|
||
params_dict = params.__dict__
|
||
ret_params = {}
|
||
for item in params_dict.keys():
|
||
ret_params[ item ] = params_dict[ item ]
|
||
|
||
if "name" not in ret_params or not ret_params['name']:
|
||
return self.tips()
|
||
|
||
module_name = ret_params['name'].replace( "/","." )
|
||
try:
|
||
import_string = "from jobs.tasks.%s import JobTask as job_target" % ( module_name )
|
||
exec( import_string , globals() )
|
||
target = job_target()
|
||
target.run( ret_params )
|
||
except Exception as e:
|
||
traceback.print_exc()
|
||
|
||
|
||
def tips(self):
|
||
tip_msg = '''
|
||
请正确调度Job
|
||
python manage runjob -m Test ( jobs/tasks/Test.py )
|
||
python manage runjob -m test/Index ( jobs/tasks/test/Index.py )
|
||
'''
|
||
app.logger.info( tip_msg )
|
||
return False
|