data_builder.py

# -*- coding: utf-8 -*-
from typing import List

from social_ethosa import BetterUser
from saya import Vk

from .data_service import BetterBotBaseDataService
import config


class DataBuilder:
    @staticmethod
    def build_programming_languages(
        user: BetterUser,
        data: BetterBotBaseDataService,
        default: str = "отсутствуют"
    ) -> str:
        """Builds the user's programming languages and returns its string representation.
        """
        languages = data.get_user_sorted_programming_languages(user)
        return ", ".join(languages) if len(languages) > 0 else default

    @staticmethod
    def build_github_profile(
        user: BetterUser,
        data: BetterBotBaseDataService,
        default: str = "",
        prefix: str = ""
    ) -> str:
        """Builds the user's github profile and returns its string representation.
        """
        profile = user["github_profile"]
        return f"{prefix}github.com/{profile}" if profile else default

    @staticmethod
    def build_karma(
        user: BetterUser,
        data: BetterBotBaseDataService
    ) -> str:
        """Builds the user's karma and returns its string representation.
        """
        plus_string = ""
        minus_string = ""
        karma = user["karma"]
        up_votes = len(user["supporters"])
        down_votes = len(user["opponents"])
        if up_votes > 0:
            plus_string = "+%.1f" % (up_votes / config.POSITIVE_VOTES_PER_KARMA)
        if down_votes > 0:
            minus_string = "-%.1f" % (down_votes / config.NEGATIVE_VOTES_PER_KARMA)
        if up_votes > 0 or down_votes > 0:
            return f"[{karma}][{plus_string}{minus_string}]"
        else:
            return f"[{karma}]"

    @staticmethod
    def get_users_sorted_by_karma(
        vk_instance: Vk,
        data: BetterBotBaseDataService,
        peer_id: int,
        reverse_sort: bool = True
    ) -> List[BetterUser]:
        members = vk_instance.get_members_ids(peer_id)
        users = data.get_users(
            other_keys=[
                "karma", "name", "programming_languages",
                "supporters", "opponents", "github_profile", "uid"],
            sort_key=lambda u: DataBuilder.calculate_real_karma(u, data),
            reverse_sort=reverse_sort)
        if members:
            users = [u for u in users if u["uid"] in members]
        return users

    @staticmethod
    def get_users_sorted_by_name(
        vk_instance,
        data: BetterBotBaseDataService,
        peer_id: int
    ) -> List[BetterUser]:
        members = vk_instance.get_members_ids(peer_id)
        users = data.get_users(
            other_keys=[
                "name", "programming_languages",
                "github_profile", "uid"
            ])
        if members:
            users = [u for u in users if u["uid"] in members]
        users.reverse()
        return users

    @staticmethod
    def get_user_top_position(
        user: BetterUser,
        vk_instance: Vk,
        data: BetterBotBaseDataService,
        peer_id: int
    ) -> int:
        """Gets user's position in the top ranking (1-based index).
        
        Returns:
        - Position number (1 for first place, 2 for second, etc.)
        - 0 if user is not found in ranking
        """
        users = DataBuilder.get_users_sorted_by_karma(vk_instance, data, peer_id)
        users = [u for u in users if
                 (u["karma"] != 0) or
                 ("programming_languages" in u and len(u["programming_languages"]) > 0)
                 ]
        
        user_uid = user["uid"]
        for index, ranked_user in enumerate(users):
            if ranked_user["uid"] == user_uid:
                return index + 1  # 1-based position
        return 0  # User not found in ranking

    @staticmethod
    def calculate_real_karma(
        user: BetterUser,
        data: BetterBotBaseDataService
    ) -> int:
        base_karma = user["karma"]
        up_votes = len(user["supporters"])/config.POSITIVE_VOTES_PER_KARMA
        down_votes = len(user["opponents"])/config.NEGATIVE_VOTES_PER_KARMA
        return base_karma + up_votes - down_votes