forked from polka_billy/porridger
HTTP exceptions added instead of True, False responces
This commit is contained in:
59
back/api.py
59
back/api.py
@ -11,7 +11,6 @@ from typing import Any, Annotated, List, Union
|
||||
from starlette.staticfiles import StaticFiles
|
||||
from sqlalchemy.orm import Session
|
||||
from sqlalchemy import select
|
||||
from dotenv import load_dotenv, dotenv_values
|
||||
|
||||
import requests
|
||||
from uuid import uuid4
|
||||
@ -26,6 +25,8 @@ import os
|
||||
|
||||
from . import add_poems_and_filters, auth_utils, orm_models, pydantic_schemas
|
||||
|
||||
from .config import TRASHBOXES_BASE_URL, TRASHBOXES_TOKEN
|
||||
|
||||
# создаем приложение Fastapi
|
||||
app = FastAPI()
|
||||
|
||||
@ -42,14 +43,10 @@ if not os.path.exists("./uploads"):
|
||||
# создаем эндпоинт для хранения файлов пользователя
|
||||
app.mount("/uploads", StaticFiles(directory = "./uploads"))
|
||||
|
||||
# load_dotenv("unimportant.env")
|
||||
# ACCESS_TOKEN_EXPIRE_MINUTES = os.getenv("ACCESS_TOKEN_EXPIRE_MINUTES")
|
||||
ACCESS_TOKEN_EXPIRE_MINUTES = 1440
|
||||
|
||||
# эндпоинт для возвращения согласия в pdf
|
||||
@app.get("/privacy_policy.pdf")
|
||||
async def privacy_policy():
|
||||
return FileResponse("privacy_policy.pdf")
|
||||
return FileResponse("./privacy_policy.pdf")
|
||||
|
||||
# получение списка объявлений
|
||||
@app.get("/api/announcements", response_model=List[pydantic_schemas.Announcement])#адрес объявлений
|
||||
@ -163,7 +160,7 @@ async def create_user(nickname: Annotated[str, Form()], password: Annotated[str,
|
||||
new_user = orm_models.User(nickname=nickname, hashed_password=auth_utils.get_password_hash(password),
|
||||
name=name, surname=surname, reg_date=datetime.date.today())
|
||||
# добавляем в бд
|
||||
await db.add(new_user)
|
||||
db.add(new_user)
|
||||
await db.commit()
|
||||
await db.refresh(new_user) # обновляем состояние объекта
|
||||
return {"Success": True}
|
||||
@ -187,7 +184,7 @@ async def login_for_access_token(
|
||||
# задаем временной интервал, в течение которого токен можно использовать
|
||||
access_token_expires = auth_utils.timedelta(minutes=auth_utils.ACCESS_TOKEN_EXPIRE_MINUTES)
|
||||
# создаем токен
|
||||
access_token = await auth_utils.create_access_token(
|
||||
access_token = auth_utils.create_access_token(
|
||||
data={"user_id": user.id}, expires_delta=access_token_expires
|
||||
)
|
||||
return {"access_token":access_token}
|
||||
@ -195,7 +192,7 @@ async def login_for_access_token(
|
||||
|
||||
# получаем данные успешно вошедшего пользователя
|
||||
@app.get("/api/users/me", response_model=pydantic_schemas.User) #
|
||||
async def read_users_me(current_user: Annotated[pydantic_schemas.User, Depends(auth_utils.get_current_active_user)]):
|
||||
def read_users_me(current_user: Annotated[pydantic_schemas.User, Depends(auth_utils.get_current_active_user)]):
|
||||
return current_user
|
||||
|
||||
|
||||
@ -231,9 +228,9 @@ async def poems_to_front(db: Annotated[Session, Depends(auth_utils.get_session)]
|
||||
num_of_poems = len(query.scalars().all())
|
||||
# если стихов в бд нет
|
||||
if num_of_poems < 1:
|
||||
add_poems_and_filters.add_poems_to_db(db) # добавляем поэмы в базу данных
|
||||
await add_poems_and_filters.add_poems_to_db(db) # добавляем поэмы в базу данных
|
||||
# после добавления стихов снова определяем кол-во стихов в бд
|
||||
query = await db.execute(select(orm_models.Poems))
|
||||
query = await db.execute(select(orm_models.Poems))
|
||||
num_of_poems = len(query.scalars().all())
|
||||
rand_id = random.randint(1, num_of_poems) # генерируем номер стихотворения
|
||||
#poem = db.query(orm_models.Poems).filter(orm_models.Poems.id == rand_id).first() # находим стих в бд
|
||||
@ -243,13 +240,22 @@ async def poems_to_front(db: Annotated[Session, Depends(auth_utils.get_session)]
|
||||
raise HTTPException(status_code=404, detail="Poem not found")
|
||||
return poem
|
||||
|
||||
trashboxes_category = {
|
||||
"PORRIDGE": ["Опасные отходы", "Иное"],
|
||||
"conspects": ["Бумага"],
|
||||
"milk": ["Стекло", "Тетра Пак", "Иное"],
|
||||
"bred": ["Пластик", "Иное"],
|
||||
"wathing": ["Пластик", "Опасные отходы", "Иное"],
|
||||
"cloth": ["Одежда"],
|
||||
"fruits_vegatables": ["Иное"],
|
||||
"other_things": ["Металл", "Бумага", "Стекло", "Иное", "Тетра Пак", "Батарейки", "Крышечки", "Шины",
|
||||
"Опасные отходы", "Лампочки", "Пластик"]
|
||||
}
|
||||
|
||||
@app.get("/api/trashbox", response_model=List[pydantic_schemas.TrashboxResponse])
|
||||
async def get_trashboxes(data: pydantic_schemas.TrashboxRequest = Depends()):#крутая функция для работы с api
|
||||
async def get_trashboxes(data: pydantic_schemas.TrashboxRequest = Depends()): #крутая функция для работы с api
|
||||
# json, передаваемый стороннему API
|
||||
BASE_URL= "https://geointelect2.gate.petersburg.ru"
|
||||
my_token="eyJhbGciOiJSUzI1NiIsInR5cCIgOiAiSldUIiwia2lkIiA6ICJhU1RaZm42bHpTdURYcUttRkg1SzN5UDFhT0FxUkhTNm9OendMUExaTXhFIn0.eyJleHAiOjE3ODgzODM1NzQsImlhdCI6MTY5MzY4OTE3NCwianRpIjoiNTIxNTE4ODYtZjJiMy00MWQ2LWE2OTYtN2JjNjg1YzRmZDBjIiwiaXNzIjoiaHR0cHM6Ly9rYy5wZXRlcnNidXJnLnJ1L3JlYWxtcy9lZ3MtYXBpIiwiYXVkIjoiYWNjb3VudCIsInN1YiI6ImJjYjQ2NzljLTU3ZGItNDU5ZC1iNWUxLWRlOGI4Yzg5MTMwMyIsInR5cCI6IkJlYXJlciIsImF6cCI6ImFkbWluLXJlc3QtY2xpZW50Iiwic2Vzc2lvbl9zdGF0ZSI6IjMyNmZlYjNjLWJhMDktNDU4NC05M2JiLTgzZjMyMTRjOTdkNSIsImFjciI6IjEiLCJhbGxvd2VkLW9yaWdpbnMiOlsiLyoiXSwicmVhbG1fYWNjZXNzIjp7InJvbGVzIjpbImRlZmF1bHQtcm9sZXMtZWdzLWFwaSIsIm9mZmxpbmVfYWNjZXNzIiwidW1hX2F1dGhvcml6YXRpb24iXX0sInJlc291cmNlX2FjY2VzcyI6eyJhY2NvdW50Ijp7InJvbGVzIjpbIm1hbmFnZS1hY2NvdW50IiwibWFuYWdlLWFjY291bnQtbGlua3MiLCJ2aWV3LXByb2ZpbGUiXX19LCJzY29wZSI6ImVtYWlsIHByb2ZpbGUiLCJzaWQiOiIzMjZmZWIzYy1iYTA5LTQ1ODQtOTNiYi04M2YzMjE0Yzk3ZDUiLCJlbWFpbF92ZXJpZmllZCI6ZmFsc2UsIm5hbWUiOiLQktC70LDQtNC40LzQuNGAINCv0LrQvtCy0LvQtdCyIiwicHJlZmVycmVkX3VzZXJuYW1lIjoiZTBmYzc2OGRhOTA4MjNiODgwZGQzOGVhMDJjMmQ5NTciLCJnaXZlbl9uYW1lIjoi0JLQu9Cw0LTQuNC80LjRgCIsImZhbWlseV9uYW1lIjoi0K_QutC-0LLQu9C10LIifQ.SpFC8EufKYV4tHWSvaeypZ-j-Lqhpva3tvilMba_Yh79EzT1EUOomeT5pC7PNJPlfsuXvOyutPBXG9Fpo7tazY2K5ymxQubYrsYt1m6KrPAJYAvDVGiZl5762-7zH1NoENgxLOjNc8eqN_cZBlcqImPyGiGAIqXaZ_Bwns-Jx5qHrYipFOQ3yXtehvy8-l84ZFhO0VhUHoZZnlfXz5x3OuOTVLU1-NjUclZEYc7t3ZlOl5m-r8jAfAdTnY-V-JDfLMbwqjouAAzauUNm4-1uiZjntrShuBsYoJcVaXcW8cSWnVk5NgWQNxwHoel4AhGIGBWPQ6gmJw2jkXBBjzXbqA"
|
||||
head = {'Authorization': 'Bearer {}'.format(my_token)}
|
||||
head = {'Authorization': 'Bearer ' + TRASHBOXES_TOKEN}
|
||||
# Данные пользователя (местоположение, количество мусорок, которое пользователь хочет видеть)
|
||||
my_data={
|
||||
'x' : f"{data.Lng}",
|
||||
@ -257,27 +263,12 @@ async def get_trashboxes(data: pydantic_schemas.TrashboxRequest = Depends()):#к
|
||||
'limit' : '1'
|
||||
}
|
||||
# Перевод категории с фронта на категорию с сайта
|
||||
match data.Category:
|
||||
case "PORRIDGE":
|
||||
list_of_category = ["Опасные отходы", "Иное"]
|
||||
case "conspects":
|
||||
list_of_category = ["Бумага"]
|
||||
case "milk":
|
||||
list_of_category = ["Стекло", "Тетра Пак", "Иное"]
|
||||
case "bred":
|
||||
list_of_category = ["Пластик", "Иное"]
|
||||
case "wathing":
|
||||
list_of_category = ["Пластик", "Опасные отходы", "Иное"]
|
||||
case "cloth":
|
||||
list_of_category = ["Одежда"]
|
||||
case "fruits_vegatables":
|
||||
list_of_category = ["Иное"]
|
||||
case "other_things":
|
||||
list_of_category = ["Металл", "Бумага", "Стекло", "Иное", "Тетра Пак", "Батарейки", "Крышечки", "Шины",
|
||||
"Опасные отходы", "Лампочки", "Пластик"]
|
||||
list_of_category = trashboxes_category[data.Category]
|
||||
|
||||
# Получение ответа от стороннего апи
|
||||
response = requests.post(f"{BASE_URL}/nearest_recycling/get", headers=head, data=my_data, timeout=10)
|
||||
response = requests.post(TRASHBOXES_BASE_URL + "/nearest_recycling/get", headers=head, data=my_data, timeout=10)
|
||||
infos = response.json()
|
||||
|
||||
# Чтение ответа
|
||||
trashboxes = []
|
||||
for trashbox in infos["results"]:
|
||||
|
Reference in New Issue
Block a user