feat: add more features

Author: GoToProd

LICENSE

@@ -0,0 +1,24 @@
+import os
+
+from sqlmodel import SQLModel
+
+from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
+from sqlalchemy.orm import sessionmaker
+
+
+DATABASE_URL = "postgresql+asyncpg://postgres:postgres@localhost:5432/"
+engine = create_async_engine(DATABASE_URL, echo=True, future=True)
+
+
+async def init_db():
+    async with engine.begin() as conn:
+        # await conn.run_sync(SQLModel.metadata.drop_all)
+        await conn.run_sync(SQLModel.metadata.create_all)
+
+
+async def get_session() -> AsyncSession:
+    async_session = sessionmaker(
+        engine, class_= AsyncSession, expire_on_commit=False,
+    )
+    async with async_session() as session:
+        yield session

README.md

@@ -0,0 +1,50 @@
+from fastapi import FastAPI, Depends, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+
+from sqlmodel import select
+from sqlmodel.ext.asyncio.session import AsyncSession
+
+from db import init_db, get_session
+from models import Overlay, OverlayCreate
+
+from schemas import OverlaySchema
+
+app = FastAPI()
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["http://localhost:5173"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+@app.on_event("startup")
+async def on_startup():
+    await init_db()
+
+@app.get("/")
+async def root():
+    return {"message": "Hello World"}
+
+@app.get("/overlays", response_model=list[Overlay])
+async def get_overlays(session: AsyncSession = Depends(get_session)):
+    result = await session.execute(select(Overlay))
+    overlays = result.scalars().all()
+    return [Overlay(nickname=overlay.nickname, tag=overlay.tag, uuid=overlay.uuid) for overlay in overlays]
+
+@app.get("/overlay/{overlay_id}", response_model=OverlaySchema)
+async def get_overlay(overlay_id: str, session: AsyncSession = Depends(get_session)):
+    result = await session.execute(select(Overlay).filter(Overlay.uuid == overlay_id))
+    overlay = result.scalars().first()
+    if overlay is None:
+        raise HTTPException(status_code=404, detail="Overlay not found")
+    return overlay
+
+@app.post("/overlay")
+async def add_overlay(overlay: OverlayCreate, session: AsyncSession = Depends(get_session)):
+    overlay = Overlay(nickname=overlay.nickname, tag=overlay.tag)
+    session.add(overlay)
+    await session.commit()
+    await session.refresh(overlay)
+    return overlay

backend/db.py

@@ -0,0 +1,31 @@
+from sqlmodel import SQLModel, Field
+import uuid as uuid_pkg
+
+
+class OverlayBase(SQLModel):
+    nickname: str
+    tag: str
+    # hdevKey: str
+    # textColor: str
+    # primaryColor: str
+    # bgColor: str
+    # progressRankColor: str
+    # progressRankBgColor: str
+    # alphaBg: bool
+    # alphaGradBg: bool
+    # wlStat: bool
+    # progressRank: bool
+    # lastMatchPoints: bool
+
+
+class Overlay(OverlayBase, table=True):
+    uuid: uuid_pkg.UUID = Field(
+        default_factory=uuid_pkg.uuid4,
+        primary_key=True,
+        index=True,
+        nullable=False,
+    )
+
+
+class OverlayCreate(OverlayBase):
+    pass

backend/main.py

@@ -0,0 +1,11 @@
+from uuid import UUID
+
+from pydantic import BaseModel
+
+class OverlaySchema(BaseModel):
+    uuid: UUID
+    nickname: str
+    tag: str
+
+    class Config:
+        orm_mode = True  # Включает поддержку ORM для SQLAlchemy моделей

backend/models.py

@@ -0,0 +1,13 @@
+import ast
+from os import environ
+from pathlib import Path
+
+from dotenv import load_dotenv
+
+BASE_DIR = Path(__file__).resolve().parent.parent.parent
+
+dotenv_file = BASE_DIR / '.env'
+if dotenv_file.is_file():
+    load_dotenv(dotenv_file)
+
+DEBUG = ast.literal_eval(environ.get('DEBUG'))