57 lines
1.3 KiB
Python
57 lines
1.3 KiB
Python
|
|
||
|
|
import asyncio
|
||
|
|
from orm.topic import Topic
|
||
|
|
|
||
|
|
|
||
|
|
class TopicStorage:
|
||
|
|
topics = {}
|
||
|
|
lock = asyncio.Lock()
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def init(session):
|
||
|
|
self = TopicStorage
|
||
|
|
topics = session.query(Topic)
|
||
|
|
self.topics = dict([(topic.slug, topic) for topic in topics])
|
||
|
|
for topic in self.topics.values():
|
||
|
|
self.load_parents(topic) # TODO: test
|
||
|
|
|
||
|
|
print('[storage.topics] %d ' % len(self.topics.keys()))
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def load_parents(topic):
|
||
|
|
self = TopicStorage
|
||
|
|
parents = []
|
||
|
|
for parent in self.topics.values():
|
||
|
|
if topic.slug in parent.children:
|
||
|
|
parents.append(parent.slug)
|
||
|
|
topic.parents = parents
|
||
|
|
return topic
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
async def get_topics_all():
|
||
|
|
self = TopicStorage
|
||
|
|
async with self.lock:
|
||
|
|
return self.topics.values()
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
async def get_topics_by_slugs(slugs):
|
||
|
|
self = TopicStorage
|
||
|
|
async with self.lock:
|
||
|
|
if not slugs:
|
||
|
|
return self.topics.values()
|
||
|
|
topics = filter(lambda topic: topic.slug in slugs, self.topics.values())
|
||
|
|
return list(topics)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
async def get_topics_by_community(community):
|
||
|
|
self = TopicStorage
|
||
|
|
async with self.lock:
|
||
|
|
topics = filter(lambda topic: topic.community == community, self.topics.values())
|
||
|
|
return list(topics)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
async def add_topic(topic):
|
||
|
|
self = TopicStorage
|
||
|
|
async with self.lock:
|
||
|
|
self.topics[topic.slug] = topic
|
||
|
|
self.load_parents(topic)
|