def get_all(self):
try:
posts = session.query(PostsModel).all()
return posts
except:
raise SystemError
finally:
session.close()
def get_pool_by_user(self, user_id):
try:
pools = session.query(PoolModel).join(PoolJoinModel).filter(PoolJoinModel.userID == user_id)
return pools
except:
raise SystemError
finally:
session.close()
def get_pool_list(self, begin):
try:
pools = session.query(PoolModel).all()
return pools
except:
raise SystemError
finally:
session.close()
def get_all(self):
try:
user_info = session.query(UserInfoModel).all()
return user_info
except:
traceback.print_exc()
finally:
session.close()
def get_posts_by_user(self, user_id):
try:
posts = session.query(PostsModel).filter(
PostsModel.userID == user_id).all()
return posts
except:
raise SystemError
finally:
session.close()
def get_my(self, user_id):
try:
posts = session.query(RecruitModel).join(PostsModel). \
filter(PostsModel.userID == user_id).filter(PostsModel.id == RecruitModel.postsID).all()
return posts
except:
raise SystemError
finally:
session.close()
def get_activity_by_user(self, userID):
try:
activity = session.query(ActivityModel).join(
ActivityJoinModel).filter(ActivityJoinModel.userID == userID)
return activity
except:
raise SystemError
finally:
session.close()
def get_user_in_pool(self, pool_id):
try:
pools = session.query(UserInfoModel).join(PoolJoinModel, PoolJoinModel.userID == UserInfoModel.id)\
.filter(PoolJoinModel.poolID == pool_id).all()
return pools
except:
raise SystemError
finally:
session.close()
def get_your_love(self, user_id, pool_id):
try:
lovers = session.query(UserInfoModel).join(LoveRelationModel,LoveRelationModel.toID == UserInfoModel.id)\
.filter(LoveRelationModel.fromID == user_id)\
.filter(LoveRelationModel.poolID == pool_id).all()
return lovers
except:
raise SystemError
finally:
session.close()
def getActivityJoin(self, user_id, activity_id):
try:
join = session.query(ActivityJoinModel).filter(ActivityJoinModel.userID == user_id). \
filter(ActivityJoinModel.activityID == activity_id).first()
return join
except AlreadyExists:
raise AlreadyExists
except:
raise SystemError
finally:
session.close()
def get_pool(self, pool_id):
try:
pool = session.query(PoolModel).filter(PoolModel.id == pool_id).first()
if not pool:
raise NotFoundException
return pool
except NotFoundException:
raise NotFoundException
except:
raise SystemError
finally:
session.close()
def get_relation(self, from_id, to_id, pool_id):
try:
relation = session.query(LoveRelationModel).filter(LoveRelationModel.fromID == from_id).\
filter(LoveRelationModel.toID == to_id).filter(LoveRelationModel.poolID == pool_id).first()
if relation:
raise AlreadyExists
except AlreadyExists:
raise AlreadyExists
except:
raise SystemError
finally:
session.close()
def getPoolJoin(self, user_id, pool_id):
try:
join = session.query(PoolJoinModel).filter(PoolJoinModel.userID == user_id).\
filter(PoolJoinModel.poolID == pool_id).first()
print(join)
return join
except AlreadyExists:
raise AlreadyExists
except:
raise SystemError
finally:
session.close()
def get_recruit(self, user_id, posts_id):
try:
recruit = session.query(RecruitModel).filter(RecruitModel.userID == user_id).\
filter(RecruitModel.postsID == posts_id).first()
if recruit:
raise AlreadyExists
except AlreadyExists:
raise AlreadyExists
except:
raise SystemError
finally:
session.close()
def get_true_love(self, user_id, pool_id):
try:
alaised1 = aliased(LoveRelationModel)
alaised2 = aliased(LoveRelationModel)
lovers = (session.query(alaised1.toID).join(alaised2).filter(
alaised1.fromID == alaised2.toID).filter(
alaised2.toID == alaised1.fromID).filter(
alaised1.fromID == user_id).filter(
alaised1.poolID == pool_id).all())
return [i[0] for i in lovers]
except:
raise SystemError
finally:
session.close()
def get_activity(self, activity_id):
try:
activity = session.query(ActivityModel).filter(
ActivityModel.id == activity_id).first()
if not activity:
raise NotFoundException
return activity
except NotFoundException:
raise NotFoundException
except:
traceback.print_exc()
raise SystemError
finally:
session.close()
def get_user_info(self, user_id, is_stable):
try:
if is_stable:
user_info = session.query(UserInfoModel).filter(UserInfoModel.id == user_id).first()
else:
user_info = session.query(CheckingUserInfoModel).filter(CheckingUserInfoModel.id == user_id).first()
if not user_info:
raise NotFoundException
return user_info
except NotFoundException:
raise NotFoundException
except:
traceback.print_exc()
finally:
session.close()
def get_activity_list(self, begin, is_old):
try:
activity_list = []
if is_old:
activity_list = session.query(ActivityModel)\
.filter(ActivityModel.selectEndTime < datetime.datetime.now()).all()
else:
activity_list = session.query(ActivityModel)\
.filter(ActivityModel.selectEndTime <= datetime.datetime.now()).all()
return activity_list
except:
raise SystemError
finally:
session.close()
def update_user_info(self, userInfoModel: UserInfoModel):
try:
self.session = db.session
user_info: UserInfoModel = session.query(UserInfoModel).filter(
UserInfoModel.userID == userInfoModel.userID).first()
"""先设置成删除然后去增加的，后面再修改"""
if not user_info:
raise NotFoundException
print(user_info.__dict__)
print(userInfoModel.__dict__)
self.delete(user_info)
session.add(userInfoModel)
session.commit()
except NotFoundException:
raise NotFoundException
except IntegrityError:
raise AlreadyExists
finally:
session.close()
def check_user_info(self, user_id, result):
if result:
"""把一个数据库中的数据直接拿到另一个数据库中"""
user_info = session.query(CheckingUserInfoModel).filter(CheckingUserInfoModel.id == user_id).first()
if not user_info:
raise NotFoundException
current_info = session.query(UserInfoModel).filter(UserInfoModel.id == user_id).first()
if current_info:
self.delete(current_info)
self.insert(UserInfoConverter().mockToReal(user_info))
self.delete(user_info)
self.session.commit()
else:
session.query(CheckingUserInfoModel).filter(CheckingUserInfoModel.id == user_id) \
.update(dict(isReject=True))
# user_info.isRejected = True
self.session.commit()
user_info = [i[0] for i in session.query(CheckingUserInfoModel.id).all()]
session.close()
return user_info
def get_un_checking_list(self):
user_list = session.query(UserModel).join(CheckingUserInfoModel).all()
session.close()
return user_list