fix: buy product

app.py

@@ -5,8 +5,8 @@ from db import db
 from flask_migrate import Migrate
 
 app = Flask(__name__)
-app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///dotori.db'
-app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
+app.config["SQLALCHEMY_DATABASE_URI"] = "sqlite:///dotori.db"
+app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
 
 db.init_app(app)
 migrate = Migrate(app, db)
@@ -14,7 +14,7 @@ migrate = Migrate(app, db)
 with app.app_context():
     db.create_all()
 
-api = Api(app, doc='/docs')
+api = Api(app, doc="/docs")
 add_namespaces(api)
-if __name__ == '__main__':
+if __name__ == "__main__":
     app.run(debug=True)

models/dotori.py

@@ -2,8 +2,9 @@ from flask_sqlalchemy import SQLAlchemy
 from datetime import datetime
 from db import db
 
+
 class UserDotori(db.Model):
-    __tablename__ = 'user_dotori'
+    __tablename__ = "user_dotori"
 
     id = db.Column(db.Integer, primary_key=True, autoincrement=True)
     user_id = db.Column(db.Integer, nullable=False)
@@ -13,17 +14,14 @@ class UserDotori(db.Model):
 
     def to_dict(self):
         return {
-            'user_id': self.user_id,
-            'dotori_count': self.dotori_count,
-            'created_at': self.created_at.isoformat(),
-            'updated_at': self.updated_at.isoformat()
+            "user_id": self.user_id,
+            "dotori_count": self.dotori_count,
+            "created_at": self.created_at.isoformat(),
+            "updated_at": self.updated_at.isoformat(),
         }
 
     def to_response(self):
-        return {
-            'userId': self.user_id,
-            'dotory': self.dotori_count
-        }
+        return {"user_id": self.user_id, "dotory": self.dotori_count}
 
     def increment(self, amount=1):
         self.dotori_count += amount
@@ -36,5 +34,6 @@ class UserDotori(db.Model):
             self.updated_at = datetime.now()
             return self.dotori_count
         return False
+
     def __repr__(self):
         return f"<UserDotori user_id={self.user_id} dotori_count={self.dotori_count}>"

routes/__init__.py

@@ -3,6 +3,7 @@ from flask_restx import Api
 from .dotori import dotori_ns
 from .product import product_ns
 
+
 def add_namespaces(api):
-    api.add_namespace(dotori_ns, path='/api/dotory')
-    api.add_namespace(product_ns, path='/api/buy')  
+    api.add_namespace(dotori_ns, path="/api/dotory")
+    api.add_namespace(product_ns, path="/api/buy")

routes/dotori.py

@@ -1,37 +1,47 @@
 from flask_restx import Namespace, Resource, fields
 from services.dotori_service import DotoriService
 
-dotori_ns = Namespace('dotory', description='도토리 관련 API')
+dotori_ns = Namespace("dotory", description="도토리 관련 API")
 
-user_model = dotori_ns.model('UserDotori', {
-    'user_id': fields.Integer,
-    'dotory': fields.Integer,
-})
+user_model = dotori_ns.model(
+    "UserDotori",
+    {
+        "user_id": fields.Integer,
+        "dotory": fields.Integer,
+    },
+)
 
-@dotori_ns.route('/<int:userId>')
+
+@dotori_ns.route("/<int:userId>")
 class UserDotori(Resource):
     @dotori_ns.marshal_with(user_model)
     def get(self, userId):
         user_dotori = DotoriService.get_user_dotori(userId)
         if not user_dotori:
-            dotori_ns.abort(404, 'User not found')
+            dotori_ns.abort(404, "User not found")
         return user_dotori.to_response()
 
-    @dotori_ns.expect(dotori_ns.model('AddDotori', {'num': fields.Integer(required=True)}))
+    @dotori_ns.expect(
+        dotori_ns.model("AddDotori", {"num": fields.Integer(required=True)})
+    )
     def put(self, userId):
         data = dotori_ns.payload
-        num = data.get('num')
+        num = data.get("num")
         return DotoriService.add_dotori(userId, num).to_response()
 
-@dotori_ns.route('')
+
+@dotori_ns.route("")
 class InitializeDotori(Resource):
-    @dotori_ns.expect(dotori_ns.model('Initialize', {'user_id': fields.Integer(required=True)}))
+    @dotori_ns.expect(
+        dotori_ns.model("Initialize", {"user_id": fields.Integer(required=True)})
+    )
     def post(self):
         data = dotori_ns.payload
-        user_id = data.get('user_id')
+        user_id = data.get("user_id")
         return DotoriService.initialize_user_dotori(user_id)
 
-@dotori_ns.route('/all')
+
+@dotori_ns.route("/all")
 class GetAllDotori(Resource):
     def get(self):
         users = DotoriService.get_all_users_dotori()

routes/product.py

@@ -1,37 +1,46 @@
 from flask_restx import Namespace, Resource, fields
 from services.dotori_service import DotoriService
 
-product_ns = Namespace('buy', description='상품 구매 API')
+product_ns = Namespace("buy", description="상품 구매 API")
 
-buy_request_model = product_ns.model('BuyRequest', {
-    'userId': fields.Integer(required=True),
-})
+buy_request_model = product_ns.model(
+    "BuyRequest",
+    {
+        "user_id": fields.Integer(required=True),
+    },
+)
 
-buy_response_model = product_ns.model('BuyResponse', {
-    'isSuccess': fields.Boolean,
-    'userId': fields.Integer,
-    'dotory': fields.Integer,
-})
+buy_response_model = product_ns.model(
+    "BuyResponse",
+    {
+        "isSuccess": fields.Boolean,
+        "user_id": fields.Integer,
+        "dotory": fields.Integer,
+    },
+)
 
-@product_ns.route('/<int:productId>')
+
+@product_ns.route("/<int:productId>")
 class BuyProduct(Resource):
     @product_ns.expect(buy_request_model)
-    @product_ns.response(200, 'Success', buy_response_model)
-    @product_ns.response(400, 'Failed')
+    @product_ns.response(200, "Success", buy_response_model)
+    @product_ns.response(400, "Failed")
     def post(self, productId):
         data = product_ns.payload
-        user_id = data.get('user_id')
+        user_id = data.get("user_id")
         if not user_id:
-            product_ns.abort(400, 'userId is required')
+            product_ns.abort(400, "userId is required")
 
-        product_price = 100 # 가격이 얼마에요?
+        product_price = 100  # 가격이 얼마에요?
         success = DotoriService.buy_product(user_id, product_price)
         if success:
             user_dotori = DotoriService.get_user_dotori(user_id)
+            if user_dotori is None:
+                product_ns.abort(400, "Failed to Buy Product")
             return {
-                'isSuccess': True,
-                'user_id': user_id,
-                'dotory': user_dotori if user_dotori else 0
+                "isSuccess": True,
+                "user_id": user_id,
+                "dotory": user_dotori.to_response(),
             }
         else:
-            product_ns.abort(400, 'Insufficient dotory or purchase failed')
+            product_ns.abort(400, "Insufficient dotory or purchase failed")

services/dotori_service.py

@@ -5,6 +5,8 @@ import sqlalchemy
 from models.dotori import UserDotori
 from flask import current_app
 from db import db
+
+
 class DotoriService:
     @staticmethod
     def get_user_dotori(user_id: int):
@@ -20,7 +22,17 @@ class DotoriService:
         if not user_dotori:
             user_dotori = UserDotori(id=user_id, user_id=user_id, dotori_count=1000000)
             print(f"초기화 완료, {user_id}님의 도토리: {user_dotori.dotori_count}")
-            db.session.execute(sqlalchemy.text("INSERT INTO user_dotori (user_id, dotori_count, created_at, updated_at) VALUES (:user_id, :dotori_count, :created_at, :updated_at)"), {"user_id":user_id, "dotori_count":1000000, "created_at":datetime.now(), "updated_at":datetime.now()})
+            db.session.execute(
+                sqlalchemy.text(
+                    "INSERT INTO user_dotori (user_id, dotori_count, created_at, updated_at) VALUES (:user_id, :dotori_count, :created_at, :updated_at)"
+                ),
+                {
+                    "user_id": user_id,
+                    "dotori_count": 1000000,
+                    "created_at": datetime.now(),
+                    "updated_at": datetime.now(),
+                },
+            )
             db.session.commit()
         return user_dotori.dotori_count
 
@@ -32,7 +44,9 @@ class DotoriService:
 
         if user_dotori.dotori_count < product_price:
             return False
-        print(f"구매 완료, {user_id}님의 도토리: {user_dotori.dotori_count} -> {user_dotori.dotori_count - product_price}")
+        print(
+            f"구매 완료, {user_id}님의 도토리: {user_dotori.dotori_count} -> {user_dotori.dotori_count - product_price}"
+        )
         user_dotori.decrement(product_price)
         db.session.commit()
         return True
@@ -43,7 +57,9 @@ class DotoriService:
         if not user_dotori:
             user_dotori = UserDotori(user_id=user_id)
             db.session.add(user_dotori)
-        print(f"도토리 추가, {user_id}님의 도토리: {user_dotori.dotori_count} -> {user_dotori.dotori_count + amount}")
+        print(
+            f"도토리 추가, {user_id}님의 도토리: {user_dotori.dotori_count} -> {user_dotori.dotori_count + amount}"
+        )
         user_dotori.increment(amount)
         db.session.commit()
         return user_dotori