console/app/cloud/controllers/billing/__init__.py

# coding: utf-8
from app import models


class ControllerBilling:
    def get(self, user_id):
        return models.UsersBalance.select(
            models.UsersBalance.balance
        ).where(
            models.UsersBalance.user == user_id
        )[0].balance

    def update(self, user_id):
        return None

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

    def create(self, user_id, balance=0):
        models.UsersBalance.create(user=user_id, balance=balance)
        return True