mirror of
https://github.com/sunrin-ana/2025-SSF-dotory-manage.git
synced 2026-03-09 18:10:02 +00:00
add gitignore
This commit is contained in:
commit
bbb2cf8d8d
4 changed files with 17 additions and 18 deletions
|
|
@ -3,25 +3,25 @@ from flask import current_app
|
|||
from db import db
|
||||
class DotoriService:
|
||||
@staticmethod
|
||||
def get_user_dotori(user_id: str):
|
||||
user_dotori = UserDotori.query.filter_by(user_id=user_id).first()
|
||||
def get_user_dotori(user_id: int):
|
||||
user_dotori = UserDotori.query.filter_by(id=user_id).first()
|
||||
if user_dotori:
|
||||
return user_dotori.dotori_count
|
||||
return None
|
||||
|
||||
@staticmethod
|
||||
def initialize_user_dotori(user_id: str):
|
||||
user_dotori = UserDotori.query.filter_by(user_id=user_id).first()
|
||||
def initialize_user_dotori(user_id: int):
|
||||
user_dotori = UserDotori.query.filter_by(id=user_id).first()
|
||||
if not user_dotori:
|
||||
user_dotori = UserDotori(user_id=user_id, dotori_count=1000000)
|
||||
user_dotori = UserDotori(id=user_id, dotori_count=1000000)
|
||||
print(f"초기화 완료, {user_id}님의 도토리: {user_dotori.dotori_count}")
|
||||
db.session.add(user_dotori)
|
||||
db.session.commit()
|
||||
return user_dotori.dotori_count
|
||||
|
||||
@staticmethod
|
||||
def buy_product(user_id: str, product_price: int):
|
||||
user_dotori = UserDotori.query.filter_by(user_id=user_id).first()
|
||||
def buy_product(user_id: int, product_price: int):
|
||||
user_dotori = UserDotori.query.filter_by(id=user_id).first()
|
||||
if not user_dotori:
|
||||
return False
|
||||
|
||||
|
|
@ -33,10 +33,10 @@ class DotoriService:
|
|||
return True
|
||||
|
||||
@staticmethod
|
||||
def add_dotori(user_id: str, amount: int):
|
||||
user_dotori = UserDotori.query.filter_by(user_id=user_id).first()
|
||||
def add_dotori(user_id: int, amount: int):
|
||||
user_dotori = UserDotori.query.filter_by(id=user_id).first()
|
||||
if not user_dotori:
|
||||
user_dotori = UserDotori(user_id=user_id)
|
||||
user_dotori = UserDotori(id=user_id)
|
||||
db.session.add(user_dotori)
|
||||
print(f"도토리 추가, {user_id}님의 도토리: {user_dotori.dotori_count} -> {user_dotori.dotori_count + amount}")
|
||||
user_dotori.increment(amount)
|
||||
|
|
@ -44,6 +44,6 @@ class DotoriService:
|
|||
return user_dotori.dotori_count
|
||||
|
||||
@staticmethod
|
||||
def get_all_users_dotori():
|
||||
def get_all_users_dotori() -> list:
|
||||
users = UserDotori.query.all()
|
||||
return [user.to_dict() for user in users]
|
||||
return [user.to_response() for user in users]
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue