unique-fix

This commit is contained in:
tonyrewin 2022-10-05 14:54:04 +03:00
parent c10f416567
commit 037c111b1d
2 changed files with 23 additions and 24 deletions

View File

@ -111,37 +111,34 @@ async def get_search_results(_, _info, searchtext, offset, limit):
@query.field("shoutsByAuthors")
async def shouts_by_authors(_, _info, slugs, offset, limit):
shouts = []
async with ShoutsCache.lock:
shouts = {}
for author in slugs:
shouts.extend(ShoutsCache.by_author.get(author, []))
shouts_prepared = []
for s in shouts:
if bool(s.publishedAt):
for a in s.authors:
a.caption = await ShoutAuthorStorage.get_author_caption(s.slug, a.slug)
if s not in shouts_prepared:
shouts_prepared.append(s)
for shouts_by_author in ShoutsCache.by_author.get(author, []):
for s in shouts_by_author:
for a in s.authors:
a.caption = await ShoutAuthorStorage.get_author_caption(s.slug, a.slug)
if bool(s.publishedAt):
shouts[s.slug] = s
shouts_prepared = shouts.values()
shouts_prepared.sort(key=lambda s: s.publishedAt, reverse=True)
return shouts_prepared[offset : offset + limit]
@query.field("shoutsByTopics")
async def shouts_by_topics(_, _info, slugs, offset, limit):
shouts = []
async def shouts_by_topics(_, _info, slugs, offset=0, limit=100):
async with ShoutsCache.lock:
shouts = {}
for topic in slugs:
shouts.extend(ShoutsCache.by_topic.get(topic, []))
shouts_prepared = []
for s in shouts:
if bool(s.publishedAt):
for a in s.authors:
a.caption = await ShoutAuthorStorage.get_author_caption(s.slug, a.slug)
if s not in shouts_prepared:
shouts_prepared.append(s)
shouts_prepared = list(set(shouts_prepared))
shouts_prepared.sort(key=lambda s: s.publishedAt, reverse=True)
return shouts_prepared[offset : offset + limit]
for shouts_by_topic in ShoutsCache.by_topic.get(topic, []):
for s in shouts_by_topic:
for a in s.authors:
a.caption = await ShoutAuthorStorage.get_author_caption(s.slug, a.slug)
if bool(s.publishedAt):
shouts[s.slug] = s
shouts_prepared = shouts.values()
shouts_prepared.sort(key=lambda s: s.publishedAt, reverse=True)
return shouts_prepared[offset : offset + limit]
@query.field("shoutsByCollection")

View File

@ -66,10 +66,12 @@ class ShoutsCache:
for s in shouts:
for a in s.authors:
ShoutsCache.by_author[a.slug] = ShoutsCache.by_author.get(a.slug, [])
ShoutsCache.by_author[a.slug].append(s)
if s not in ShoutsCache.by_author[a.slug]:
ShoutsCache.by_author[a.slug].append(s)
for t in s.topics:
ShoutsCache.by_topic[t.slug] = ShoutsCache.by_topic.get(t.slug, [])
ShoutsCache.by_topic[t.slug].append(s)
if s not in ShoutsCache.by_topic[t.slug]:
ShoutsCache.by_topic[t.slug].append(s)
print("[zine.cache] indexed by %d topics " % len(ShoutsCache.by_topic.keys()))
print("[zine.cache] indexed by %d authors " % len(ShoutsCache.by_author.keys()))
ShoutsCache.recent_published = shouts