def test_getlist_params_noexit(self): argv = [0,1,'epg',2] actual = params.getlist_params(argv) assert actual.data_type == 'epg'
def test_main(self): arg = ['dummy',1,'epg','download'] param = params.getlist_params(arg) actual = getlist.main(param)
def test_getlist_params_exit2(self): argv = [0,1,2,3] params.getlist_params(argv)
# -*- coding: utf-8 -*- import os import sys from tools import aws from tools import log from tools import util from tools import params def main(param): try: # get result from rds with aws.rdsconn(config) as rdsconn: sql = "select to_char(data_date, 'YYYY-MM-DD'), msg from data_ctrl_mst where maker_id = %s and data_type = '%s' and status= '%s' order by 1,2;" % (param.maker_id, param.data_type, param.status) res = aws.fetch_sql(rdsconn, sql) for i in res: datadate = i[0] filename = i[1].split('/')[-1] log.print_log(logger, '{0}, {1}'.format(datadate, filename)) except Exception as e: log.except_log(logger, e) 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.getlist_params(sys.argv) main(param)