def get(userId, session=None):
if not session:
session = db.Session()
ret = {}
result = session.query(User).filter_by(userId=userId).first()
if result:
obj = dict((key,value) for key, value in vars(result).items() if not key.startswith('_'))
ret = obj
return(ret)
def get_all(session=None):
if not session:
session = db.Session()
ret = []
our_results = session.query(User).filter_by()
for result in our_results:
obj = {}
obj.update(dict((key,value) for key, value in vars(result).items() if not key.startswith('_')))
ret.append(obj)
return(ret)
def add(userId, password, inobj, session=None):
if not session:
session = db.Session()
#our_result = session.query(User).filter_by(userId=userId, password=password).first()
our_result = session.query(User).filter_by(userId=userId).first()
if not our_result:
our_result = User(userId=userId, password=password)
if 'created_at' not in inobj:
inobj['created_at'] = int(time.time())
our_result.update(inobj)
session.add(our_result)
else:
our_result.update(inobj)
return (True)
def delete(userId, session=None):
if not session:
session = db.Session()
ret = False
result = session.query(User).filter_by(userId=userId).first()
if result:
session.delete(result)
ret = True
# try:
# session.commit()
# ret = True
# except Exception as err:
# raise err
# finally:
# session.rollback()
return (ret)
