96 lines
2.8 KiB
Python
96 lines
2.8 KiB
Python
|
from __future__ import absolute_import
|
||
|
|
from __future__ import division
|
||
|
|
from __future__ import print_function
|
||
|
|
from __future__ import unicode_literals
|
||
|
|
|
||
|
|
from builtins import *
|
||
|
|
from collections import namedtuple
|
||
|
|
|
||
|
|
import pandas as pd
|
||
|
|
|
||
|
|
|
||
|
|
def _to_date(row):
|
||
|
|
date = int(row['DATE'])
|
||
|
|
return pd.datetime(year=date // 10000, month=date // 100 % 100, day=date % 100)
|
||
|
|
|
||
|
|
|
||
|
|
def _to_datetime(row):
|
||
|
|
date = int(row['DATE'])
|
||
|
|
time = int(row['TIME']) // 1000
|
||
|
|
return pd.datetime(year=date // 10000, month=date // 100 % 100, day=date % 100,
|
||
|
|
hour=time // 10000, minute=time // 100 % 100, second=time % 100)
|
||
|
|
|
||
|
|
|
||
|
|
def _to_dataframe(cloumset, index_func=None, index_column=None):
|
||
|
|
df = pd.DataFrame(cloumset)
|
||
|
|
if index_func:
|
||
|
|
df.index = df.apply(index_func, axis=1)
|
||
|
|
elif index_column:
|
||
|
|
df.index = df[index_column]
|
||
|
|
del df.index.name
|
||
|
|
|
||
|
|
return df
|
||
|
|
|
||
|
|
|
||
|
|
def _error_to_str(error):
|
||
|
|
if error:
|
||
|
|
if 'message' in error:
|
||
|
|
return str(error['error']) + "," + error['message']
|
||
|
|
else:
|
||
|
|
return str(error['error']) + ","
|
||
|
|
else:
|
||
|
|
return ","
|
||
|
|
|
||
|
|
|
||
|
|
def to_obj(class_name, data):
|
||
|
|
try:
|
||
|
|
if isinstance(data, (list, tuple)):
|
||
|
|
result = []
|
||
|
|
for d in data:
|
||
|
|
result.append(namedtuple(class_name, list(d.keys()))(*list(d.values())))
|
||
|
|
return result
|
||
|
|
|
||
|
|
elif type(data) == dict:
|
||
|
|
result = namedtuple(class_name, list(data.keys()))(*list(data.values()))
|
||
|
|
return result
|
||
|
|
else:
|
||
|
|
return data
|
||
|
|
except Exception as e:
|
||
|
|
print(class_name, data, e)
|
||
|
|
return data
|
||
|
|
|
||
|
|
|
||
|
|
def extract_result(cr, format="", index_column=None, class_name=""):
|
||
|
|
"""
|
||
|
|
format supports pandas, obj.
|
||
|
|
"""
|
||
|
|
|
||
|
|
err = _error_to_str(cr['error']) if 'error' in cr else None
|
||
|
|
if 'result' in cr:
|
||
|
|
if format == "pandas":
|
||
|
|
if index_column:
|
||
|
|
return (_to_dataframe(cr['result'], None, index_column), err)
|
||
|
|
if 'TIME' in cr['result']:
|
||
|
|
return (_to_dataframe(cr['result'], _to_datetime), err)
|
||
|
|
elif 'DATE' in cr['result']:
|
||
|
|
return (_to_dataframe(cr['result'], _to_date), err)
|
||
|
|
else:
|
||
|
|
return (_to_dataframe(cr['result']), err)
|
||
|
|
|
||
|
|
elif format == "obj" and cr['result'] and class_name:
|
||
|
|
r = cr['result']
|
||
|
|
if isinstance(r, (list, tuple)):
|
||
|
|
result = []
|
||
|
|
for d in r:
|
||
|
|
result.append(namedtuple(class_name, list(d.keys()))(*list(d.values())))
|
||
|
|
elif isinstance(r, dict):
|
||
|
|
result = namedtuple(class_name, list(r.keys()))(*list(r.values()))
|
||
|
|
else:
|
||
|
|
result = r
|
||
|
|
|
||
|
|
return (result, err)
|
||
|
|
else:
|
||
|
|
return (cr['result'], err)
|
||
|
|
else:
|
||
|
|
return (None, err)
|