polka_billy/porridger
1
0
Fork 1
Code Issues 20 Pull Requests Packages Releases Activity

filter_ann almost fixed

Browse Source
This commit is contained in:
DmitryGantimurov 2023-09-02 19:04:30 +03:00
parent f744cce713
commit 37d219c516
2 changed files with 35 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

60
back/add_poems_and_filters.py
View File

@ -3,7 +3,7 @@ from sqlalchemy.sql import text, literal_column
from sqlalchemy.ext.asyncio import AsyncSession
from typing import Annotated
from fastapi import Depends
from sqlalchemy import select
from sqlalchemy import select, or_, and_
from . import auth_utils, orm_models, pydantic_schemas
import datetime
@ -49,35 +49,43 @@ async def add_poems_to_db(async_db: AsyncSession):
async def filter_ann(schema: pydantic_schemas.SortAnnouncements, db: AsyncSession):
"""Функция для последовательного применения различных фильтров (через схему SortAnnouncements)"""
# fields = schema.__dict__ # параметры передоваемой схемы SortAnnouncements (ключи и значения)
# # проходим по названиям фильтров и их значениям
# for name, filt_val in fields.items():
# # создаем выражение sql запроса
# statement = text("SELECT * FROM announcements ")
# # выбираем все строки
# query = await db.execute(statement)
# res = query.all()
# # если фильтр задан
# if filt_val is not None:
# to_intersect_query = await db.execute(select(orm_models.Announcement).where(literal_column(f"announcements.{name}") == filt_val))
# to_intersect = to_intersect_query.all()
# res = res.intersect(to_intersect)
fields = schema.__dict__ # параметры передоваемой схемы SortAnnouncements (ключи и значения)
# проходим по названиям фильтров и их значениям
# выбираем все строки
query = await db.execute(select(orm_models.Announcement))
res = set(query.scalars().all())
for name, filt_val in fields.items():
# res = await db.execute(statement)
# если фильтр задан
if filt_val is not None:
filter_query = await db.execute(select(orm_models.Announcement).where(literal_column(f"announcements.{name}") == filt_val))
filtered = set(filter_query.scalars().all())
res = res.intersection(filtered)
# # отфильтровываем подходящие объявления
# res = await db.execute(
# select(orm_models.Announcement).where(
# ((schema.obsolete == None) | ((schema.obsolete != None) & (orm_models.Announcement.obsolete == schema.obsolete)))
# & ((schema.user_id == None) | ((schema.user_id != None) & (orm_models.Announcement.user_id == schema.user_id)))
# & ((schema.metro == None) | ((schema.metro != None) & (orm_models.Announcement.metro == schema.metro)))
# & ((schema.category == None) | ((schema.category != None) & (orm_models.Announcement.category == schema.category)))
# )
# )
# отфильтровываем подходящие объявления
res = await db.execute(select(orm_models.Announcement)
.where(schema.obsolete != None and orm_models.Announcement.obsolete == schema.obsolete
and schema.user_id != None and orm_models.Announcement.user_id == schema.user_id
and schema.metro != None and orm_models.Announcement.metro == schema.metro
and )
# .where(schema.user_id != None and orm_models.Announcement.user_id == schema.user_id)
# .where(schema.metro != None and orm_models.Announcement.metro == schema.metro)
# .where(schema.category != None and orm_models.Announcement.category == schema.category)
)
# statement = text("SELECT * FROM announcements "
# "WHERE announcements.obsolete = :obsolete "
# "AND announcements.user_id == :user_id "
# "AND announcements.metro == :metro "
# "AND announcements.category == :category")
# "INTERSECT"
# "SELECT * FROM announcements "
# "WHERE announcements.user_id == :user_id "
# "INTERSECT"
# "SELECT * FROM announcements "
# "WHERE announcements.metro == :metro "
# "INTERSECT"
# "SELECT * FROM announcements "
# "WHERE announcements.category == :category")
# res = await db.execute(statement,
# {"obsolete": schema.obsolete,
@ -85,9 +93,9 @@ async def filter_ann(schema: pydantic_schemas.SortAnnouncements, db: AsyncSessio
# "metro": schema.metro,
# "category": schema.category}
# )
# возвращаем все подходящие объявления
return res.scalars().all()
# возвращаем все подходящие объявления
return res
async def check_obsolete(db: AsyncSession, current_date: datetime.date):

2
back/db.py
View File

@ -5,7 +5,7 @@ from sqlalchemy.ext.declarative import declarative_base
SQLALCHEMY_DATABASE_URL = "sqlite+aiosqlite:///./sql_app.db"
# SQLALCHEMY_DATABASE_URL = "postgresql+asyncpg://postgres:D560c34V112Ak@localhost/porridger"
engine = create_async_engine(SQLALCHEMY_DATABASE_URL, echo=True)
engine = create_async_engine(SQLALCHEMY_DATABASE_URL)
SessionLocal = sessionmaker(bind=engine, class_=AsyncSession, expire_on_commit=False)