mirror of https://github.com/stijndcl/didier
44 lines
1.1 KiB
Python
44 lines
1.1 KiB
Python
|
from datetime import datetime
|
||
|
|
|
||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
|
|
||
|
|
from database.crud import users
|
||
|
|
from database.exceptions import currency as exceptions
|
||
|
|
from database.models import Bank
|
||
|
|
|
||
|
|
|
||
|
|
NIGHTLY_AMOUNT = 420
|
||
|
|
|
||
|
|
|
||
|
|
async def get_bank(session: AsyncSession, user_id: int) -> Bank:
|
||
|
|
"""Get a user's bank info"""
|
||
|
|
user = await users.get_or_add(session, user_id)
|
||
|
|
return user.bank
|
||
|
|
|
||
|
|
|
||
|
|
async def add_dinks(session: AsyncSession, user_id: int, amount: int):
|
||
|
|
"""Increase the Dinks counter for a user"""
|
||
|
|
bank = await get_bank(session, user_id)
|
||
|
|
bank.dinks += amount
|
||
|
|
session.add(bank)
|
||
|
|
await session.commit()
|
||
|
|
|
||
|
|
|
||
|
|
async def claim_nightly(session: AsyncSession, user_id: int):
|
||
|
|
"""Claim daily Dinks"""
|
||
|
|
user = await users.get_or_add(session, user_id)
|
||
|
|
nightly_data = user.nightly_data
|
||
|
|
|
||
|
|
now = datetime.now()
|
||
|
|
|
||
|
|
if nightly_data.last_nightly is not None and nightly_data.last_nightly.date() == now.date():
|
||
|
|
raise exceptions.DoubleNightly
|
||
|
|
|
||
|
|
bank = user.bank
|
||
|
|
bank.dinks += NIGHTLY_AMOUNT
|
||
|
|
nightly_data.last_nightly = now
|
||
|
|
|
||
|
|
session.add(bank)
|
||
|
|
session.add(nightly_data)
|
||
|
|
await session.commit()
|