karma.py

import database

import logging
import peewee
import telebot


def get_user_karma(user: telebot.types.User):
    """func to get user's karma

    Args:
        user (telebot.types.User): user.

    Returns:
        int | None: int if there's no errors, none if exception is raised.
    """
    check_user_in_database(user)
    try:
        _user = database.User.select().where(database.User.id == str(user.id)).get()
        return _user.karma
    except Exception as exception:
        logging.error(exception)
        return None
    
    
def get_increase_times(user: telebot.types.User):
    """func to get user's increasion karma times.

    Args:
        user (telebot.types.User): user.

    Returns:
        int | None: int if there's no errors, none if exception is raised.
    """
    check_user_in_database(user)
    try:
        _user = database.User.select().where(database.User.id == str(user.id)).get()
        return _user.increase_karma
    except Exception as exception:
        logging.exception(exception)
        return None


def get_decrease_times(user: telebot.types.User):
    """func to get user's decreasion karma times.

    Args:
        user (telebot.types.User): user.

    Returns:
        int | None: int if there's no errors, none if exception is raised.
    """
    check_user_in_database(user)
    try:
        _user = database.User.select().where(database.User.id == str(user.id)).get()
        return _user.decrease_karma
    except Exception as exception:
        logging.exception(exception)
        return None
    
    
def check_user_in_database(user: telebot.types.User):
    """check if user in database. if not - add him.

    Args:
        user (telebot.types.User): user.
    """
    try:
        database.User.select().where(database.User.id == str(user.id)).get()
    except peewee.DoesNotExist:
        add_new_user_to_database(user)


def add_new_user_to_database(user: telebot.types.User):
    """func to add new user to database.

    args:
        user (telebot.types.User): the user. that's all.
    """
    for _user in database.User.select():
        if _user.id == str(user.id):
            logging.warning(f"User with id={_user.id} already exists!")
            return
        
    database.User.create(id=str(user.id),
                         name=str(user.first_name),
                         karma=0,
                         warns=0,
                         state=0,
                         increase_karma=0,
                         decrease_karma=0,
                        )
    logging.info(f"Added new user: {user.full_name} with id={user.id}).")


def change_user_karma(target_user: telebot.types.User, from_user: telebot.types.User,  amount: int = 1):
    """func to change user's karma.

    args:
        user (telebot.types.User): user.
        amount (int, optional): amount of karma to change. defaults to 1.
    """
    _target_user = database.User.select().where(database.User.id == target_user.id).get()
    _target_user.karma += amount
    
    _from_user = database.User.select().where(database.User.id == from_user.id).get()
    if amount > 0:
        _from_user.increase_karma += 1
    elif amount < 0:
        _from_user.decrease_karma += 1
    try:
        _target_user.save()
        _from_user.save()
        logging.info(f"Changed karma for user: {target_user.full_name}, now karma = {_target_user.karma}")
    except Exception as exception:
        logging.error(exception)


def get_top_users(sort: str = 'asc'):
    """func to get top of users.
    
    args:
        sort (str): sorting method (asc, desc).
    """
    if sort == 'asc':
        top_users = database.User.select().where(database.User.karma < 0).order_by(database.User.karma.asc()).limit(15)
    elif sort == 'desc':
        top_users = database.User.select().where(database.User.karma > 0).order_by(database.User.karma.desc()).limit(15)
        
    return top_users