mirror of
https://github.com/wlinator/luminara.git
synced 2024-10-03 02:23:13 +00:00
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
|
from db import database
|
||
|
|
||
|
|
||
|
class Currency:
|
||
|
@staticmethod
|
||
|
def get_cash_balance(user_id):
|
||
|
query = "SELECT cash_balance FROM currency WHERE user_id={}".format(user_id)
|
||
|
cash_balance = database.select_query(query)
|
||
|
|
||
|
if not cash_balance:
|
||
|
Currency.create_new_balance(user_id)
|
||
|
return 50
|
||
|
|
||
|
return cash_balance[0][0]
|
||
|
|
||
|
@staticmethod
|
||
|
def get_special_balance(user_id):
|
||
|
query = "SELECT special_balance FROM currency WHERE user_id={}".format(user_id)
|
||
|
special_balance = database.select_query(query)
|
||
|
|
||
|
if not special_balance:
|
||
|
Currency.create_new_balance(user_id)
|
||
|
return 3
|
||
|
|
||
|
return special_balance[0][0]
|
||
|
|
||
|
@staticmethod
|
||
|
def update_cash_balance(user_id, amount):
|
||
|
if amount < 0:
|
||
|
amount = 0
|
||
|
|
||
|
query = "UPDATE currency SET cash_balance = {} WHERE user_id = {}".format(round(amount, 2), user_id)
|
||
|
database.execute_query(query)
|
||
|
|
||
|
@staticmethod
|
||
|
def update_special_balance(user_id, amount):
|
||
|
if amount < 0:
|
||
|
amount = 0
|
||
|
|
||
|
query = "UPDATE currency SET special_balance = {} WHERE user_id = {}".format(amount, user_id)
|
||
|
database.execute_query(query)
|
||
|
|
||
|
@staticmethod
|
||
|
def create_new_balance(user_id):
|
||
|
query = "INSERT INTO currency(user_id, cash_balance, special_balance) VALUES ({}, 50, 3)".format(user_id)
|
||
|
database.execute_query(query)
|
||
|
print(f"BALANCE UPDATE --- USER with ID {user_id} created new balance")
|