@@ -520,8 +520,8 @@ def write(self, symbol, data, initial_image=None):
520
520
start = data [0 ]['index' ]
521
521
end = data [- 1 ]['index' ]
522
522
elif isinstance (data , pd .DataFrame ):
523
- start = data .index [0 ].to_datetime ()
524
- end = data .index [- 1 ].to_datetime ()
523
+ start = data .index [0 ].to_pydatetime ()
524
+ end = data .index [- 1 ].to_pydatetime ()
525
525
pandas = True
526
526
else :
527
527
raise UnhandledDtypeException ("Can't persist type %s to tickstore" % type (data ))
@@ -609,18 +609,18 @@ def _pandas_compute_final_image(df, image, end):
609
609
@staticmethod
610
610
def _pandas_to_bucket (df , symbol , initial_image ):
611
611
rtn = {SYMBOL : symbol , VERSION : CHUNK_VERSION_NUMBER , COLUMNS : {}, COUNT : len (df )}
612
- end = to_dt (df .index [- 1 ].to_datetime ())
612
+ end = to_dt (df .index [- 1 ].to_pydatetime ())
613
613
if initial_image :
614
614
if 'index' in initial_image :
615
- start = min (to_dt (df .index [0 ].to_datetime ()), initial_image ['index' ])
615
+ start = min (to_dt (df .index [0 ].to_pydatetime ()), initial_image ['index' ])
616
616
else :
617
- start = to_dt (df .index [0 ].to_datetime ())
617
+ start = to_dt (df .index [0 ].to_pydatetime ())
618
618
image_start = initial_image .get ('index' , start )
619
619
image = {k : v for k , v in initial_image .items () if k != 'index' }
620
620
rtn [IMAGE_DOC ] = {IMAGE_TIME : image_start , IMAGE : initial_image }
621
621
final_image = TickStore ._pandas_compute_final_image (df , initial_image , end )
622
622
else :
623
- start = to_dt (df .index [0 ].to_datetime ())
623
+ start = to_dt (df .index [0 ].to_pydatetime ())
624
624
final_image = {}
625
625
rtn [END ] = end
626
626
rtn [START ] = start
0 commit comments