console/app/cloud/controllers/users/manage.py

# coding: utf-8

from app import models
from . import ControllerUsersDetails


class ControllerManageUsers:
    def __init__(self):
        pass

    def exists_by_id(self, user_id):
        if models.Users.select().where(models.Users.id == user_id).count() == 0:
            return False
        return True

    def exists_by_email(self, email):
        if models.Users.select().where(models.Users.email == email).count() == 0:
            return False
        return True

    def items_get(self):
        """
        формируем массив со списком пользователей
        :return:
        """
        return {
            'total': models.Users.select().count(),
            'items': models.Users.select()
        }

    def item_get(self, user_id):
        return models.Users.select().where(models.Users.id == user_id)[0]


class ControllerManageUsersDetails:
    def exists(self, user_id):
        if models.UsersDetails.select().where(models.UsersDetails.user == user_id).count() == 0:
            return False
        return True

    def item_get(self, user_id):
        users_details = ControllerUsersDetails(user_id)
        if not users_details.details_exists():
            users_details.details_create()
        return models.UsersDetails.select().where(models.UsersDetails.user == user_id)[0]


class ControllerManageUsersBalance:
    def exists(self, user_id):
        if models.UsersBalance.select().where(models.UsersBalance.user == user_id).count() == 0:
            return False
        return True

    def item_get(self, user_id):
        return models.UsersBalance.select().where(models.UsersBalance.user == user_id)[0]