169 lines
6.7 KiB
Python
169 lines
6.7 KiB
Python
import json
|
|
|
|
from orm.topic import TopicFollower
|
|
from services.encoders import CustomJSONEncoder
|
|
from services.rediscache import redis
|
|
from services.db import local_session
|
|
|
|
DEFAULT_FOLLOWS = {
|
|
"topics": [],
|
|
"authors": [],
|
|
"communities": [{"id": 1, "name": "Дискурс", "slug": "discours", "pic": ""}],
|
|
}
|
|
|
|
|
|
async def cache_author(author: dict):
|
|
author_id = author.get("id")
|
|
payload = json.dumps(author, cls=CustomJSONEncoder)
|
|
await redis.execute("SET", f'user:{author.get("user")}', payload)
|
|
await redis.execute("SET", f"author:{author_id}", payload)
|
|
|
|
# update stat all field for followers' caches in <authors> list
|
|
followers_str = await redis.execute("GET", f"author:{author_id}:followers")
|
|
followers = []
|
|
if isinstance(followers_str, str):
|
|
followers = json.loads(followers_str)
|
|
if isinstance(followers, list):
|
|
for follower in followers:
|
|
follower_follows_authors = []
|
|
follower_follows_authors_str = await redis.execute(
|
|
"GET", f"author:{author_id}:follows-authors"
|
|
)
|
|
if isinstance(follower_follows_authors_str, str):
|
|
follower_follows_authors = json.loads(follower_follows_authors_str)
|
|
c = 0
|
|
for old_author in follower_follows_authors:
|
|
if int(old_author.get("id")) == int(author.get("id", 0)):
|
|
follower_follows_authors[c] = author
|
|
break # exit the loop since we found and updated the author
|
|
c += 1
|
|
else:
|
|
# author not found in the list, so add the new author with the updated stat field
|
|
follower_follows_authors.append(author)
|
|
|
|
# update stat field for all authors' caches in <followers> list
|
|
follows_str = await redis.execute("GET", f"author:{author_id}:follows-authors")
|
|
follows_authors = []
|
|
if isinstance(follows_str, str):
|
|
follows_authors = json.loads(follows_str)
|
|
if isinstance(follows_authors, list):
|
|
for followed_author in follows_authors:
|
|
followed_author_followers = []
|
|
followed_author_followers_str = await redis.execute(
|
|
"GET", f"author:{author_id}:followers"
|
|
)
|
|
if isinstance(followed_author_followers_str, str):
|
|
followed_author_followers = json.loads(followed_author_followers_str)
|
|
c = 0
|
|
for old_follower in followed_author_followers:
|
|
old_follower_id = int(old_follower.get("id"))
|
|
if old_follower_id == author_id:
|
|
followed_author_followers[c] = author
|
|
break # exit the loop since we found and updated the author
|
|
c += 1
|
|
# author not found in the list, so add the new author with the updated stat field
|
|
followed_author_followers.append(author)
|
|
await redis.execute(
|
|
"SET", f"author:{author_id}:followers", followed_author_followers
|
|
)
|
|
|
|
|
|
async def cache_follows(follower: dict, entity_type: str, entity: dict, is_insert=True):
|
|
# prepare
|
|
follows = []
|
|
follower_id = follower.get("id")
|
|
if follower_id:
|
|
redis_key = f"author:{follower_id}:follows-{entity_type}s"
|
|
follows_str = await redis.execute("GET", redis_key)
|
|
if isinstance(follows_str, str):
|
|
follows = json.loads(follows_str)
|
|
if is_insert:
|
|
follows.append(entity)
|
|
else:
|
|
entity_id = entity.get("id")
|
|
if not entity_id:
|
|
raise Exception("wrong entity")
|
|
# Remove the entity from follows
|
|
follows = [e for e in follows if e["id"] != entity_id]
|
|
|
|
# update follows cache
|
|
payload = json.dumps(follows, cls=CustomJSONEncoder)
|
|
await redis.execute("SET", redis_key, payload)
|
|
|
|
# update follower's stats everywhere
|
|
author_str = await redis.execute("GET", f"author:{follower_id}")
|
|
if isinstance(author_str, str):
|
|
author = json.loads(author_str)
|
|
author["stat"][f"{entity_type}s"] = len(follows)
|
|
await cache_author(author)
|
|
return follows
|
|
|
|
|
|
async def cache_follower(follower: dict, author: dict, is_insert=True):
|
|
author_id = author.get("id")
|
|
follower_id = follower.get("id")
|
|
followers = []
|
|
if author_id and follower_id:
|
|
redis_key = f"author:{author_id}:followers"
|
|
followers_str = await redis.execute("GET", redis_key)
|
|
followers = json.loads(followers_str) if isinstance(followers_str, str) else []
|
|
if is_insert and not any([int(f["id"]) == author_id for f in followers]):
|
|
followers.append(follower)
|
|
else:
|
|
followers = [e for e in followers if int(e["id"]) != author_id]
|
|
|
|
followers = [
|
|
dict(d)
|
|
for d in set(tuple(tuple(k, v) for k, v in d.items()) for d in followers)
|
|
]
|
|
|
|
author_str = await redis.execute("GET", f"author:{follower_id}")
|
|
if isinstance(author_str, str):
|
|
author = json.loads(author_str)
|
|
author["stat"]["followers"] = len(followers)
|
|
await cache_author(author)
|
|
payload = json.dumps(followers, cls=CustomJSONEncoder)
|
|
await redis.execute("SET", redis_key, payload)
|
|
return followers
|
|
|
|
|
|
async def cache_topic(topic_dict: dict):
|
|
# update stat all field for followers' caches in <topics> list
|
|
followers = (
|
|
local_session()
|
|
.query(TopicFollower)
|
|
.filter(TopicFollower.topic == topic_dict.get("id"))
|
|
.all()
|
|
)
|
|
for tf in followers:
|
|
follower_id = tf.follower
|
|
follower_follows_topics = []
|
|
follower_follows_topics_str = await redis.execute(
|
|
"GET", f"author:{follower_id}:follows-topics"
|
|
)
|
|
if isinstance(follower_follows_topics_str, str):
|
|
follower_follows_topics = json.loads(follower_follows_topics_str)
|
|
c = 0
|
|
for old_topic in follower_follows_topics:
|
|
if int(old_topic.get("id")) == int(topic_dict.get("id", 0)):
|
|
follower_follows_topics[c] = topic_dict
|
|
break # exit the loop since we found and updated the topic
|
|
c += 1
|
|
else:
|
|
# topic not found in the list, so add the new topic with the updated stat field
|
|
follower_follows_topics.append(topic_dict)
|
|
|
|
await redis.execute(
|
|
"SET",
|
|
"SET",
|
|
f"author:{follower_id}:follows-topics",
|
|
json.dumps(follower_follows_topics),
|
|
)
|
|
|
|
# update topic's stat
|
|
topic_dict["stat"]["followers"] = len(followers)
|
|
|
|
# save in cache
|
|
payload = json.dumps(topic_dict, cls=CustomJSONEncoder)
|
|
await redis.execute("SET", f'topic:{topic_dict.get("slug")}', payload)
|