core/services/zine/shoutauthor.py

import asyncio
from base.orm import local_session
from orm.shout import ShoutAuthor


class ShoutAuthorStorage:
	authors_by_shout = {}
	lock = asyncio.Lock()
	period = 30*60 #sec

	@staticmethod
	async def load(session):
		self = ShoutAuthorStorage
		sas = session.query(ShoutAuthor).all()
		for sa in sas:
			self.authors_by_shout[sa.shout] = self.authors_by_shout.get(sa.shout, [])
			self.authors_by_shout[sa.shout].append([sa.user, sa.caption])
		print('[zine.authors] %d shouts preprocessed' % len(self.authors_by_shout))

	@staticmethod
	async def get_authors(shout):
		self = ShoutAuthorStorage
		async with self.lock:
			return self.authors_by_shout.get(shout, [])

	@staticmethod
	async def get_author_caption(shout, author):
		self = ShoutAuthorStorage
		async with self.lock:
			for a in self.authors_by_shout.get(shout, []):
				if author in a:
					return a[1]
		return { "error": "author caption not found" }

	@staticmethod
	async def worker():
		self = ShoutAuthorStorage
		while True:
			try:
				with local_session() as session:
					async with self.lock:
						await self.load(session)
						print("[zine.authors] state updated")
			except Exception as err:
				print("[zine.authors] errror: %s" % (err))
			await asyncio.sleep(self.period)