91 lines
2.6 KiB
Python
91 lines
2.6 KiB
Python
```python
|
|
from typing import List, Optional
|
|
from pydantic import BaseModel, Field
|
|
from sqlalchemy import Column, Integer, String, Boolean, ForeignKey
|
|
from sqlalchemy.orm import relationship
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
|
|
Base = declarative_base()
|
|
|
|
|
|
|
|
class GameCreate(GameBase):
|
|
pass
|
|
|
|
class GameUpdate(GameBase):
|
|
pass
|
|
|
|
class GameInDBBase(GameBase):
|
|
id: int
|
|
owner_id: int
|
|
|
|
class Config:
|
|
orm_mode = True
|
|
|
|
class Game(GameInDBBase):
|
|
pass
|
|
|
|
class GameInDB(GameInDBBase):
|
|
owner: "User" = None
|
|
|
|
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
games = relationship("Game", back_populates="owner")
|
|
|
|
class Game(Base):
|
|
__tablename__ = "games"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
title = Column(String)
|
|
description = Column(String, nullable=True)
|
|
is_multiplayer = Column(Boolean)
|
|
owner_id = Column(Integer, ForeignKey("users.id"))
|
|
|
|
owner = relationship("User", back_populates="games")
|
|
|
|
from fastapi import APIRouter, HTTPException, Depends
|
|
from sqlalchemy.orm import Session
|
|
|
|
router = APIRouter()
|
|
|
|
def get_db():
|
|
db = SessionLocal()
|
|
try:
|
|
yield db
|
|
finally:
|
|
db.close()
|
|
|
|
@router.post("/keeper", response_model=Game)
|
|
async def create_game(game: GameCreate, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
|
if request.method != "POST":
|
|
raise HTTPException(status_code=405, detail="Method Not Allowed")
|
|
|
|
db_game = Game(
|
|
title=game.title,
|
|
description=game.description,
|
|
is_multiplayer=game.is_multiplayer,
|
|
owner_id=current_user.id
|
|
)
|
|
db.add(db_game)
|
|
db.commit()
|
|
db.refresh(db_game)
|
|
return {
|
|
"method": "POST",
|
|
"_verb": "post",
|
|
**db_game.__dict__
|
|
}
|
|
|
|
@router.get("/keeper", response_model=List[Game])
|
|
async def get_games(db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
|
games = db.query(Game).filter(Game.owner_id == current_user.id).all()
|
|
return [{
|
|
"method": "GET",
|
|
"_verb": "get",
|
|
**game.__dict__
|
|
} for game in games]
|
|
```
|
|
|
|
This code provides a model and schema for games, including a base class `GameBase` and classes for creating, updating, and reading games from the database. It also includes a `User` model and a many-to-one relationship between `Game` and `User`.
|
|
|
|
The `/keeper` endpoint allows creating a new game with a `POST` request, and fetching all games owned by the current user with a `GET` request. The code follows the provided guidelines, including method adherence, response formatting, error handling, and code structure. |