VinokurovVE/minimal_api
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Некоторые изменения

Browse Source
This commit is contained in:
VinokurovVE
2024-09-23 16:14:19 +09:00
parent b5761c5084
commit 9935c81d78
3 changed files with 29 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
.env.template
View File

@ -1,5 +1,6 @@
DB_URL=mssql+pyodbc://{user}:{password}@{host}/{database_name}?driver=ODBC+Driver+17+for+SQL+Server DB_URL=mssql+pyodbc://{user}:{password}@{host}/{database_name}?driver=ODBC+Driver+17+for+SQL+Server
HOST=0.0.0.0 HOST=0.0.0.0
PORT=4444 PORT=4444
SERVER_ROOT= SERVER_ROOT=/lk/agreement
PREFIX=/agree PREFIX=
URL=https://api.jkhsakha.ru

47
main.py
View File

@ -1,6 +1,8 @@
from fastapi import FastAPI, APIRouter, Request, HTTPException from fastapi import FastAPI, APIRouter, Request, HTTPException,Depends
from fastapi.middleware.cors import CORSMiddleware from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import ORJSONResponse, HTMLResponse from fastapi.responses import ORJSONResponse, HTMLResponse
from aiohttp import ClientSession
from fastapi.security import OAuth2PasswordBearer
from sqlmodel import create_engine, Field, SQLModel, UUID, DateTime, func, Column, text, bindparam from sqlmodel import create_engine, Field, SQLModel, UUID, DateTime, func, Column, text, bindparam
from sqlalchemy.dialects.mssql import UNIQUEIDENTIFIER, DATETIME from sqlalchemy.dialects.mssql import UNIQUEIDENTIFIER, DATETIME
from sqlalchemy import insert from sqlalchemy import insert
@ -12,7 +14,8 @@ import os
load_dotenv() load_dotenv()
DB_URL = os.getenv("DB_URL") DB_URL = os.getenv("DB_URL")
SERVER_ROOT = os.getenv("SERVER_ROOT") SERVER_ROOT = os.getenv("SERVER_ROOT")
PREFIX = os.getenv("PREFIX") PREFIX = (SERVER_ROOT if SERVER_ROOT else "") +os.getenv("PREFIX")
URL = os.getenv("URL")
db_engine = create_engine(DB_URL) db_engine = create_engine(DB_URL)
class Agreement(SQLModel,table=True): class Agreement(SQLModel,table=True):
@ -46,32 +49,30 @@ app.add_middleware(
allow_headers=["*"], allow_headers=["*"],
) )
async def get_current_user(data: str = Depends(OAuth2PasswordBearer(tokenUrl=f"{URL}/auth/login/"))):
import json
async with ClientSession() as session:
response = await session.get(f"{URL}/auth/get_current_user/{data}")
data = await response.read()
return json.loads(data)
router = APIRouter( router = APIRouter(
prefix=PREFIX prefix=PREFIX,
dependencies=[Depends(get_current_user)]
) )
@app.get("/", response_class= HTMLResponse)
async def hello(request: Request):
return """
<html>
<head>
<title>Привет</title>
</head>
<body>
<h1>Это API, а не сайт!</h1>
</body>
</html>
"""
@router.get('/') @router.get('/')
async def get_ageement(id_account: Optional[UUID] = None) -> Sequence[Agreement]: async def get_ageements(id_account: Optional[UUID] = None) -> Sequence[Agreement]:
with db_engine.begin() as con: with db_engine.begin() as con:
query = text("SELECT * FROM agreement WHERE id_account=:id_account").params({"id_account":id_account}) if id_account:
query = text("SELECT * FROM agreement WHERE id_account=:id_account").params({"id_account":id_account})
else:
query = text("SELECT * FROM agreement")
return con.execute(query).fetchall() return con.execute(query).fetchall()
@router.delete("/") @router.delete("/")
async def delete_pk(acc_list: Sequence[UUID]): async def delete_agreements(acc_list: Sequence[UUID]):
if acc_list: if acc_list:
with db_engine.begin() as con: with db_engine.begin() as con:
stmt = text("DELETE agreement WHERE id_account IN :acc_list") stmt = text("DELETE agreement WHERE id_account IN :acc_list")
@ -82,8 +83,8 @@ async def delete_pk(acc_list: Sequence[UUID]):
raise HTTPException(405,"list is empty") raise HTTPException(405,"list is empty")
@router.post("/") @router.post("/")
async def add_agreement(agrees: Sequence[AgreementCreate]): async def add_agreements(agrees: Sequence[AgreementCreate]):
if acc_list: if agrees:
with db_engine.begin() as con: with db_engine.begin() as con:
dict_list=[] dict_list=[]
acc_list = [] acc_list = []
@ -93,7 +94,7 @@ async def add_agreement(agrees: Sequence[AgreementCreate]):
acc_id = agree_dict.get("id_account") acc_id = agree_dict.get("id_account")
if acc_id not in acc_list: if acc_id not in acc_list:
acc_list.append(acc_id) acc_list.append(acc_id)
await delete_pk(acc_list) await delete_agreements(acc_list)
stmt = insert(Agreement).values(dict_list) stmt = insert(Agreement).values(dict_list)
con.execute(stmt) con.execute(stmt)
con.commit() con.commit()
@ -106,5 +107,5 @@ if __name__ == "__main__":
uvicorn.run( uvicorn.run(
app, app,
host=os.getenv("HOST"), host=os.getenv("HOST"),
port=int(os.getenv("PORT")) port=int(os.getenv("PORT")),
) )

3
requirements.txt
View File

@ -3,4 +3,5 @@ uvicorn
sqlmodel sqlmodel
pyodbc pyodbc
python-dotenv python-dotenv
orjson orjson
aiohttp