core/services/following.py

import asyncio


class FollowingResult:
    def __init__(self, event, kind, payload):
        self.event = event
        self.kind = kind
        self.payload = payload


class Following:
    queue = asyncio.Queue()

    def __init__(self, kind, uid):
        self.kind = kind  # author topic shout community
        self.uid = uid


class FollowingManager:
    lock = asyncio.Lock()
    data = {"author": [], "topic": [], "shout": [], "community": []}

    @staticmethod
    async def register(kind, uid):
        async with FollowingManager.lock:
            FollowingManager[kind].append(uid)

    @staticmethod
    async def remove(kind, uid):
        async with FollowingManager.lock:
            FollowingManager[kind].remove(uid)

    @staticmethod
    async def push(kind, payload):
        try:
            async with FollowingManager.lock:
                for entity in FollowingManager[kind]:
                    if payload.shout["created_by"] == entity.uid:
                        entity.queue.put_nowait(payload)
        except Exception as e:
            print(Exception(e))