common.py 423 B

123456789101112131415161718
  1. import json
  2. import numpy as np
  3. from datetime import datetime
  4. class IEncoder (json.JSONEncoder):
  5. def default (self,object):
  6. if type(object) == np.integer :
  7. return int(object)
  8. elif type(object) == np.floating:
  9. return float(object)
  10. elif type(object) == np.ndarray :
  11. return object.tolist()
  12. elif type(object) == datetime :
  13. return object.isoformat()
  14. else:
  15. return super(IEncoder,self).default(object)