add viewShout

orm/shout.py

@@ -2,6 +2,7 @@ from typing import List
 from datetime import datetime
 from sqlalchemy import Table, Column, Integer, String, ForeignKey, DateTime, Boolean
 from sqlalchemy.orm import relationship
+from sqlalchemy.orm.attributes import flag_modified
 from orm import Permission, User, Topic
 from orm.comment import Comment
 from orm.base import Base
@@ -62,20 +63,44 @@ class ShoutViewByDay(Base):
 
 	id = None
 	shout_id = Column(ForeignKey('shout.id'), primary_key = True)
-	day: str = Column(DateTime, primary_key = True, default = datetime.now)
+	day = Column(DateTime, primary_key = True, default = datetime.now)
 	value = Column(Integer)
 
 class ShoutViewStorage:
 
 	def __init__(self, session):
 		self.views = session.query(ShoutViewByDay).all()
+		self.this_day_views = {}
+		for view in self.views:
+			shout_id = view.shout_id
+			if not shout_id in self.this_day_views:
+				self.this_day_views[shout_id] = view
+			this_day_view = self.this_day_views[shout_id]
+			if this_day_view.day < view.day:
+				self.this_day_views[shout_id] = view
 
 	def get_view(self, shout_id):
 		shout_views = list(filter(lambda x: x.shout_id == shout_id, self.views))
 		return reduce((lambda x, y: x + y.value), shout_views, 0)
 
-	def add_view(self, view):
-		self.views.append(view)
+	def inc_view(self, shout_id):
+		this_day_view = self.this_day_views.get(shout_id)
+		if not this_day_view:
+			this_day_view = ShoutViewByDay.create(shout_id = shout_id, value = 1)
+			self.this_day_views[shout_id] = this_day_view
+			self.views.append(this_day_view)
+		else:
+			this_day_view.value = this_day_view.value + 1
+			this_day_view.modified = True
+
+	def flush_changes(self, session):
+		for view in self.this_day_views.values():
+			if getattr(view, "modified", False):
+				session.add(view)
+				flag_modified(view, "value")
+				view.modified = False
+		session.commit()
+
 
 class Shout(Base):
 	__tablename__ = 'shout'