|
|
|
@ -235,6 +235,7 @@ class Generator (Learner):
|
|
|
|
|
if _args['year'] in ['',None,np.nan] :
|
|
|
|
|
return None
|
|
|
|
|
year = int(_args['year'])
|
|
|
|
|
|
|
|
|
|
offset = _args['offset'] if 'offset' in _args else 0
|
|
|
|
|
month = np.random.randint(1,13)
|
|
|
|
|
if month == 2:
|
|
|
|
@ -244,11 +245,11 @@ class Generator (Learner):
|
|
|
|
|
day = np.random.randint(1,_end)
|
|
|
|
|
|
|
|
|
|
#-- synthetic date
|
|
|
|
|
_date = datetime(year=year,month=month,day=day) #,minute=0,hour=0,second=0)
|
|
|
|
|
FORMAT = '%Y-%d-%m'
|
|
|
|
|
_date = datetime(year=year,month=month,day=day,minute=0,hour=0,second=0)
|
|
|
|
|
FORMAT = '%Y-%m-%d'
|
|
|
|
|
_name = _args['field'] if 'field' in _args else None
|
|
|
|
|
if 'format' in self.info and _name in self.info['format']:
|
|
|
|
|
_name = _args['field']
|
|
|
|
|
# _name = _args['field']
|
|
|
|
|
FORMAT = self.info['format'][_name]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@ -258,7 +259,7 @@ class Generator (Learner):
|
|
|
|
|
r = [_date.strftime(FORMAT)]
|
|
|
|
|
for _delta in offset :
|
|
|
|
|
_date = _date + timedelta(_delta)
|
|
|
|
|
r.append(_date.strftime(FORMAT))
|
|
|
|
|
r.append(_date.strptime(FORMAT))
|
|
|
|
|
return r
|
|
|
|
|
else:
|
|
|
|
|
return _date.strftime(FORMAT)
|
|
|
|
@ -270,16 +271,19 @@ class Generator (Learner):
|
|
|
|
|
name = _item['name']
|
|
|
|
|
|
|
|
|
|
if _item['type'].upper() in ['DATE','DATETIME','TIMESTAMP'] :
|
|
|
|
|
FORMAT = '%Y-%d-%m'
|
|
|
|
|
FORMAT = '%Y-%m-%d'
|
|
|
|
|
|
|
|
|
|
if 'format' in self.info and name in self.info['format'] :
|
|
|
|
|
FORMAT = self.info['format'][name]
|
|
|
|
|
else:
|
|
|
|
|
if _item['type'] == ['DATETIME','TIMESTAMP'] :
|
|
|
|
|
FORMAT = '%Y-%d-%m %H:%M:%S'
|
|
|
|
|
r[name] = FORMAT
|
|
|
|
|
|
|
|
|
|
elif _item['type'] in ['DATETIME','TIMESTAMP'] :
|
|
|
|
|
FORMAT = '%Y-%m-%d %H:%M:%S'
|
|
|
|
|
|
|
|
|
|
_df[name] = pd.to_datetime(_df[name], format=FORMAT).astype('datetime64[ns]')
|
|
|
|
|
r[name] = FORMAT
|
|
|
|
|
_df[name] = pd.to_datetime(_df[name], format=FORMAT) #.astype('datetime64[ns]')
|
|
|
|
|
if _item['type'] in ['DATETIME','TIMESTAMP']:
|
|
|
|
|
_df[name] = _df[name].astype('datetime64[ns]')
|
|
|
|
|
else:
|
|
|
|
|
_df[name] = _df[name].astype(str)
|
|
|
|
|
if r :
|
|
|
|
|
self.log(**{'action':'format','input':r})
|
|
|
|
|
return _df
|
|
|
|
@ -309,10 +313,12 @@ class Generator (Learner):
|
|
|
|
|
iname = self.info['make_date'][name]
|
|
|
|
|
|
|
|
|
|
years = _df[iname]
|
|
|
|
|
_dates = [self.make_date(year=year,field=name) for year in years]
|
|
|
|
|
_dates = [self.make_date(year=_year,field=name) for _year in years]
|
|
|
|
|
if _dates :
|
|
|
|
|
_df[name] = _dates
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
_schema = self.get_schema()
|
|
|
|
|
_schema = [{'name':_item.name,'type':_item.field_type} for _item in _schema]
|
|
|
|
|
_df = self.format(_df,_schema)
|
|
|
|
@ -341,7 +347,12 @@ class Shuffle(Generator):
|
|
|
|
|
|
|
|
|
|
_log = {'action':'io-data','input':{'candidates':1,'rows':int(self._df.shape[0])}}
|
|
|
|
|
self.log(**_log)
|
|
|
|
|
try:
|
|
|
|
|
self.post([self._df])
|
|
|
|
|
self.log(**{'action':'completed','input':{'candidates':1,'rows':int(self._df.shape[0])}})
|
|
|
|
|
except Exception as e :
|
|
|
|
|
# print (e)
|
|
|
|
|
self.log(**{'action':'failed','input':{'msg':e,'info':self.info}})
|
|
|
|
|
class factory :
|
|
|
|
|
_infocache = {}
|
|
|
|
|
@staticmethod
|
|
|
|
|