Example #1
0
 def test_main(self):
   t = timedelta(days = 5)
   d = datetime.today() - t
   dp = datetime.strftime(d,'%Y-%m-%d')
   arg = ['dummy',1,'epg',dp,'test']
   param = params.setstatus_params(arg)
   actual = setstatus.main(param)
Example #2
0
 def test_setstatus_params_noexit(self):
   argv = [0,1,'epg','2015-08-15',4]
   actual = params.setstatus_params(argv)
   assert actual.data_type == 'epg'
Example #3
0
 def test_setstatus_params_exit3(self):
   argv = [0,1,'epg','20150815',4]
   params.setstatus_params(argv)
Example #4
0
 def test_setstatus_params_exit2(self):
   argv = [0,1,'epga',3,4]
   params.setstatus_params(argv)
Example #5
0
 def test_setstatus_params_exit1(self):
   argv = [0,1,2,3]
   params.setstatus_params(argv)
Example #6
0
import os
import sys
from tools import aws
from tools import log
from tools import util
from tools import params
from tools import ctrl


ERROR_STATUS = "{0}_failed"


def main(param):
    try:
        c = ctrl.data_ctrl_mst(param.maker_id, param.data_type, param.data_date.replace("-", ""), param.status)
    except Exception as e:
        log.except_log(logger, e)
        c.status = ERROR_STATUS.format(c.status)
    finally:
        # log to RDS
        with aws.rdsconn(config) as rdsconn:
            aws.commit_ctrl_update(rdsconn, c)


if __name__ == "__main__":
    logger = log.init_logger(False, os.path.basename(__file__))
    confpath = os.path.join(os.path.dirname(os.path.realpath(__file__)), "conf", "app.config")
    config = util.get_config(confpath)
    param = params.setstatus_params(sys.argv)
    main(param)