47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
import asyncio
|
|
from base.orm import local_session
|
|
from orm.shout import ShoutAuthor, Shout
|
|
|
|
|
|
class ShoutAuthorStorage:
|
|
authors_by_shout = {}
|
|
lock = asyncio.Lock()
|
|
period = 30 * 60 # sec
|
|
|
|
@staticmethod
|
|
async def load_captions(session):
|
|
self = ShoutAuthorStorage
|
|
sas = session.query(ShoutAuthor).join(Shout).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 indexed by authors" % 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_captions(session)
|
|
print("[zine.authors] index by authors was updated")
|
|
except Exception as err:
|
|
print("[zine.authors] error indexing by author: %s" % (err))
|
|
await asyncio.sleep(self.period)
|