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)
