78 lines
No EOL
2.7 KiB
Python
78 lines
No EOL
2.7 KiB
Python
from sqlmodel import SQLModel, create_engine, Session, select
|
|
|
|
from app.models import example_model, project
|
|
from app.config import settings
|
|
|
|
engine = create_engine(
|
|
settings.database_url, echo=True,
|
|
connect_args={"check_same_thread": False},
|
|
)
|
|
|
|
def create_database():
|
|
SQLModel.metadata.create_all(engine)
|
|
|
|
def add_to_database(input: any):
|
|
with Session(engine) as session:
|
|
session.add(input)
|
|
session.commit()
|
|
|
|
def get_from_database(query: any) -> list:
|
|
with Session(engine) as session:
|
|
results = session.exec(select(query)).all()
|
|
return results
|
|
|
|
def get_from_database_where(query: any, variable: any, value: any, first_only: bool = False, offset: int | None = None, limit: int | None = None) -> list:
|
|
with Session(engine) as session:
|
|
statement = select(query).where(variable == value)
|
|
if offset:
|
|
statement = statement.offset(offset)
|
|
if limit:
|
|
statement = statement.limit(limit)
|
|
results = session.exec(statement)
|
|
if first_only:
|
|
results = results.first()
|
|
else:
|
|
results = results.all()
|
|
return results
|
|
|
|
def get_from_database_where_id(query: any, value: int):
|
|
with Session(engine) as session:
|
|
result = session.get(query, value)
|
|
return result
|
|
|
|
def get_from_database_where_join_filter(query: any, joined: any, variable: any, value: any):
|
|
with Session(engine) as session:
|
|
results = session.exec(select(query).join(joined).where(variable == value))
|
|
return results
|
|
|
|
def get_and_join_from_database(query: any, joined: any, isouter: bool = False):
|
|
with Session(engine) as session:
|
|
results = session.exec(select(query, joined).join(joined, isouter)).all()
|
|
return results
|
|
|
|
def update_database_where(query: any, variable: any, value: any, attribute: any, new_value: any):
|
|
with Session(engine) as session:
|
|
results = session.exec(select(query).where(variable == value))
|
|
|
|
for instance in results:
|
|
instance.setAttribute(attribute, new_value)
|
|
session.add(instance)
|
|
session.commit()
|
|
session.refresh(instance)
|
|
return results
|
|
|
|
def delete_from_database_where(query: any, variable: any, value: any):
|
|
with Session(engine) as session:
|
|
results = session.exec(select(query).where(variable == value))
|
|
|
|
for instance in results:
|
|
session.delete(instance)
|
|
session.commit()
|
|
|
|
remainder = session.exec(select(query).where(variable == value))
|
|
if remainder is None:
|
|
return {"message": "Successfully deleted matching query(ies)", "deleted": results.all()}
|
|
else:
|
|
return {"message": "Failed to delete all matching queries", "remaining": remainder.all()}
|
|
|
|
|