86 lines
2.2 KiB
Python
86 lines
2.2 KiB
Python
|
|
||
|
|
|
||
|
|
import asyncio
|
||
|
|
from orm.base import local_session
|
||
|
|
from storages.shoutauthor import ShoutAuthorStorage
|
||
|
|
from orm.topic import ShoutTopic, TopicFollower
|
||
|
|
|
||
|
|
|
||
|
|
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('[storage.topicstat] authors sorted')
|
||
|
|
print('[storage.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('[storage.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) -> dict:
|
||
|
|
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("[storage.topicstat] updated")
|
||
|
|
except Exception as err:
|
||
|
|
print("[storage.topicstat] errror: %s" % (err))
|
||
|
|
await asyncio.sleep(self.period)
|
||
|
|
|