# coding: utf-8

from SWSCloudCore import models


class ControllerBilling:
    """
    Класс работы с деньгами
    """
    def get(self, user_id):
        """
        Выбираем состояние баланса выбранного пользователя
        """
        x = models.UsersBalance.select(
            models.UsersBalance.balance
        ).where(
            models.UsersBalance.user == user_id
        ).get()
        return x.balance

    def update(self, user_id):
        """
        Обновление баланса
        """
        return None

    def exists(self, user_id):
        """
        Проверяет наличие записи в таблице балансов
        """
        cnt = models.UsersBalance.select().where(
            models.UsersBalance.user == user_id
        ).count()
        if cnt == 0:
            return False
        return True

    def create(self, user_id, balance=0):
        """
        Создание записи в таблице
        """
        models.UsersBalance.create(user=user_id, balance=balance)
        return True

    def transactions_get(self, user_id):
        transactions = dict()
        transactions['total'] = models.UsersBalanceTransactions.select().where(
            models.UsersBalanceTransactions.user == user_id
        ).count()
        transactions['items'] = models.UsersBalanceTransactions.select().where(
            models.UsersBalanceTransactions.user == user_id
        )
        return transactions