def modify_time(time_str):
'''
set time of kwargs
:param time_str: like '2017-02-16'
:return: dict
'''
kwargs = {'earliest_time':'',
'latest_time':'',
'search_mode':'normal',
'exec_mode':'blocking'}
append_hms = 'T00:00:00.000'
tmp_year = time_str.split('-')[0]
tmp_month = time_str.split('-')[1]
tmp_day = time_str.split('-')[2]
if '16' in tmp_day:
str_replace = time_str.replace('16', '01')
kwargs['earliest_time'] = str_replace + append_hms
kwargs['latest_time'] = time_str + append_hms
return kwargs
if '01' in tmp_day:
tmp_str_replace = tmp_month + '-' + tmp_day
if tmp_month == '01':
time_new_year = str(int(tmp_year)-1)
time_new_str = time_new_year+ '-' +'12' + '-' + '16'
str_replace = time_str.replace(time_str,time_new_str)
else:
time_new_month = str(int(tmp_month) - 1)
time_new_str = '0' + time_new_month + '-' + '16'
str_replace = time_str.replace(tmp_str_replace, time_new_str)
kwargs['earliest_time'] = str_replace + append_hms
kwargs['latest_time'] = time_str + append_hms
return kwargs
import unittest
from generatePeakUtil import modify_time
class Test(unittest.TestCase):
def setUp(self):
self.time_str = '2017-02-01'
def tearDown(self):
self.time_str = None
def test_split(self):
d = modify_time(self.time_str)['earliest_time']
self.assertEqual(d, '2017-01-16T00:00:00.000')
if __name__ == '__main__':
unittest.main()