discours.io/core
5
0
Fork 0
Code Pull Requests Actions Projects Activity

add rateShout

Browse Source
This commit is contained in:
knst-kotov
2021-09-25 14:40:37 +03:00
parent 9a5ec80b12
commit f23d375b4d
4 changed files with 35 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
orm/shout.py
View File

@@ -35,7 +35,7 @@ class ShoutRating(Base):
id = None
rater_id = Column(ForeignKey('user.id'), primary_key = True)
shout_id = Column(ForeignKey('shout.id'), primary_key = True)
ts: str = Column(DateTime, nullable=False, default = datetime.now, comment="Timestamp")
ts = Column(DateTime, nullable=False, default = datetime.now, comment="Timestamp")
value = Column(Integer)
class ShoutRatingStorage:
@@ -48,8 +48,8 @@ class ShoutRatingStorage:
return reduce((lambda x, y: x + y.value), shout_ratings, 0)
def update_rating(self, new_rating):
rating = next(x for x in self.ratings \
if x.rater_id == new_rating.rater_id and x.shout_id == new_rating.shout_id)
rating = next((x for x in self.ratings \
if x.rater_id == new_rating.rater_id and x.shout_id == new_rating.shout_id), None)
if rating:
rating.value = new_rating.value
rating.ts = new_rating.ts