def get_user_by_id(user_id, check=True):
user = session.query(User).filter(User.id == user_id).one_or_none()
if user == None:
if check:
raise RunException('user not register')
return user
def password_right(username=None, password=None):
count = session.query(User).filter(User.username==username).filter(User.password==password).count()
if count == 1:
return True
elif count == 0:
return False
else:
raise RunException('count the same username and password more than one')
def get_type_by_id(type_id=0, user_id=0, check=True):
result = session.query(Type).filter(Type.id == type_id).filter(
Type.user_id == user_id).one_or_none()
if result == None:
if check:
raise RunException('result not exist')
return result
def get_all():
res_list = []
for user in session.query(module.User).all():
d = {}
d.setdefault('id', user.id)
d.setdefault('username', user.username)
d.setdefault('password', user.password)
res_list.append(d)
session.close()
return jsonify(res_list)
def get_user_by_name_psword(username=None, password=None):
users = []
for user in session.query(User).filter(User.username == username).filter(User.password == password).all():
users.append(user)
session.close()
if len(users) == 1:
return users[0]
elif len(users) > 1:
raise RunException('count user by username and password more than one')
else:
raise RunException('password is wrong')
def get_blobs_by_user_id(user_id=None, type_id=None):
sql = session.query(Blob).outerjoin(
Sort, Blob.id == Sort.blob_id).filter(Blob.user_id == user_id)
if not type_id == None:
sql = sql.filter(Sort.type_id == type_id)
result = []
for blob in sql.group_by(Blob.id).all():
result.append(blob)
session.close()
return result
def get_person_by_user_id(user_id, check=True):
persons = []
for person in session.query(Person).filter(
Person.user_id == user_id).all():
persons.append(person)
if len(persons) == 0:
if check:
raise RunException(msg='person not found by user_id is [%d]' %
(user_id))
else:
return None
elif len(persons) > 1:
raise RunException(msg='person more than one by user_id is [%d]' %
(user_id))
else:
return persons[0]
def update(user):
session.query(User).filter(User.id == user.id).update({'username':user.username, 'password':user.password})
session.commit()
session.close()
def get_person_info_by_user_id(user_id):
persons = []
for user, person in session.query(User, Person).join(
Person,
Person.user_id == User.id).filter(User.id == user_id).all():
print user.id, person.id
def update(user_id, person):
session.query(Person).filter(Person.user_id == user_id).update(person)
session.commit()
session.close()
def get_blob_detail(blob_id):
result = []
for blob, type in session.query(Blob, Type).filter(
Blob.id == Sort.blob_id).filter(Type.id == Sort.type_id).all():
result.append((blob, type))
return result
