SQLAlchemy 1.0->SQLAlchemy 2.x
This commit is contained in:
@@ -1,4 +1,4 @@
|
||||
from sqlalchemy import asc, desc
|
||||
from sqlalchemy import asc, desc, select
|
||||
from sqlalchemy.orm import Session
|
||||
from sqlalchemy.orm.query import Query
|
||||
from typing import Type, List, Optional
|
||||
@@ -46,11 +46,12 @@ class crudbase:
|
||||
and_conditions.append(column == value)
|
||||
|
||||
if and_conditions:
|
||||
query = query.filter(*and_conditions)
|
||||
query = query.where(and_(*and_conditions))
|
||||
if or_conditions:
|
||||
query = query.filter(or_(*or_conditions))
|
||||
return query
|
||||
query = query.where(or_(*or_conditions))
|
||||
|
||||
return query
|
||||
|
||||
def _apply_sorting(self, query: Query, sort_by: Optional[str], sort_order: Optional[str]) -> Query:
|
||||
if sort_by:
|
||||
column = getattr(self.model, sort_by, None)
|
||||
@@ -61,12 +62,11 @@ class crudbase:
|
||||
query = query.order_by(asc(column))
|
||||
return query
|
||||
|
||||
def get_all(self, db: Session) -> Query:
|
||||
return db.query(self.model)
|
||||
def get_all(self) -> Query:
|
||||
return select(self.model)
|
||||
|
||||
|
||||
def get(self, db: Session, item_id: int) -> Optional[models.Base]:
|
||||
return db.query(self.model).get(item_id)
|
||||
return db.execute(select(self.model).filter(self.model.id == item_id)).scalar_one_or_none()
|
||||
|
||||
def create(self, db: Session, obj_in: BaseModel) -> models.Base:
|
||||
db_obj = self.model(**obj_in.model_dump())
|
||||
@@ -76,7 +76,7 @@ class crudbase:
|
||||
return db_obj
|
||||
|
||||
def update(self, db: Session, item_id: int, obj_in: BaseModel) -> Optional[models.Base]:
|
||||
db_obj = db.query(self.model).filter(self.model.id == item_id).first()
|
||||
db_obj = self.get(db,item_id)
|
||||
if db_obj:
|
||||
for key, value in obj_in.model_dump(exclude_unset=True).items():
|
||||
setattr(db_obj, key, value)
|
||||
@@ -86,16 +86,16 @@ class crudbase:
|
||||
return None
|
||||
|
||||
def delete(self, db: Session, item_id: int) -> Optional[models.Base]:
|
||||
db_obj = db.query(self.model).get(item_id)
|
||||
db_obj = self.get(db,item_id)
|
||||
if db_obj:
|
||||
db.delete(db_obj)
|
||||
db.commit()
|
||||
return db_obj
|
||||
return None
|
||||
|
||||
def get_by_conditions(self, db: Session, filters: Optional[dict] = None, sort_by: Optional[str] = None,
|
||||
def get_by_conditions(self, filters: Optional[dict] = None, sort_by: Optional[str] = None,
|
||||
sort_order: Optional[str] = "asc") -> Query:
|
||||
query = db.query(self.model)
|
||||
query = select(self.model)
|
||||
if filters:
|
||||
query = self._apply_filters(query, filters)
|
||||
if sort_by:
|
||||
|
||||
Reference in New Issue
Block a user