working-on

services/stat/topicstat.py

@@ -0,0 +1,83 @@
+import asyncio
+from base.orm import local_session
+from services.zine.shoutauthor import ShoutAuthorStorage
+from orm.topic import ShoutTopic, TopicFollower
+from typing import Dict
+
+class TopicStat:
+	shouts_by_topic = {}
+	authors_by_topic = {}
+	followers_by_topic = {}
+	reactions_by_topic = {}
+	lock = asyncio.Lock()
+	period = 30*60 #sec
+
+	@staticmethod
+	async def load_stat(session):
+		self = TopicStat
+		self.shouts_by_topic = {}
+		self.authors_by_topic = {}
+		shout_topics = session.query(ShoutTopic)
+		for shout_topic in shout_topics:
+			topic = shout_topic.topic
+			shout = shout_topic.shout
+			if topic in self.shouts_by_topic:
+				self.shouts_by_topic[topic].append(shout)
+			else:
+				self.shouts_by_topic[topic] = [shout]
+
+			authors = await ShoutAuthorStorage.get_authors(shout)
+			if topic in self.authors_by_topic:
+				self.authors_by_topic[topic].update(authors)
+			else:
+				self.authors_by_topic[topic] = set(authors)
+
+		print('[service.topicstat] authors sorted')
+		print('[service.topicstat] shouts sorted')
+		
+		self.followers_by_topic = {}
+		followings = session.query(TopicFollower)
+		for flw in followings:
+			topic = flw.topic
+			user = flw.follower
+			if topic in self.followers_by_topic:
+				self.followers_by_topic[topic].append(user)
+			else:
+				self.followers_by_topic[topic] = [user]
+		print('[service.topicstat] followers sorted')
+
+	@staticmethod
+	async def get_shouts(topic):
+		self = TopicStat
+		async with self.lock:
+			return self.shouts_by_topic.get(topic, [])
+
+	@staticmethod
+	async def get_stat(topic):
+		self = TopicStat
+		async with self.lock:
+			shouts = self.shouts_by_topic.get(topic, [])
+			followers = self.followers_by_topic.get(topic, [])
+			authors = self.authors_by_topic.get(topic, [])
+			reactions = self.reactions_by_topic.get(topic, [])
+
+		return  { 
+			"shouts" : len(shouts),
+			"authors" : len(authors),
+			"followers" : len(followers),
+			"reactions" : len(reactions)
+		}
+
+	@staticmethod
+	async def worker():
+		self = TopicStat
+		while True:
+			try:
+				with local_session() as session:
+					async with self.lock:
+						await self.load_stat(session)
+						print("[service.topicstat] updated")
+			except Exception as err:
+				print("[service.topicstat] errror: %s" % (err))
+			await asyncio.sleep(self.period)
+