def add_data(self, model_class, data_obj):
"""add new data to database, `data_obj` must be a json object
"""
if type(data_obj) is not dict:
raise Exception('data_obj is not dict!')
session = getattr(model_class, '__session__')
db_type = getattr(model_class, '__db_type__', 'sqlorm')
if db_type == 'sqlorm':
try:
instance = dict_to_inst(model_class, data_obj)
session.begin()
session.add(instance)
session.commit()
except sqlalchemy.exc.SQLAlchemyError as e:
_LOGGER.error("Save Data Error, %s", e)
elif db_type == 'mongo':
try:
inst = session()
inst.update(data_obj)
inst.save()
except mongokit.ValidationError as e:
_LOGGER.error("Save Data Error, %s", e)
else:
pass
def add_data(self, model_class, data_obj):
"""add new data to database, `data_obj` must be a json object
"""
if type(data_obj) is not dict:
raise Exception('data_obj is not dict!')
session = getattr(model_class, '__session__')
db_type = getattr(model_class, '__db_type__', 'sqlorm')
if db_type == 'sqlorm':
instance = dict_to_inst(model_class, data_obj)
session.begin()
session.add(instance)
session.commit()
else:
pass
