covot/database/crud.py

from sqlalchemy import or_
from sqlalchemy.orm import Query

from database import SessionLocal, engine, models
from database.models import User

db = SessionLocal()


def search_database(id) -> Query:
    """
    Returns the user associated with the id argument
    """
    return db.query(User).filter(
        or_(User.correo_institucional == id, User.numero_de_documento == id)
    )


def insert_data(data) -> None:
    """
    Inserts a new user into the database
    """
    item = User(**data)
    db.add(item)
    db.commit()
    db.refresh(item)


def create_database(data) -> None:
    """
    Creates the SQL database from scratch and populates it
    """
    models.Base.metadata.create_all(bind=engine)
    existing_row = search_database(data["correo_institucional"])
    if existing_row:
        return
    insert_data(data)


def save_attribute(attribute, data) -> None:
    """
    Updates the attribute value with the content of data
    """
    key = eval("User." + attribute)
    db.query(User).filter(
        or_(User.correo_institucional == id, User.numero_de_documento == id)
    ).update({key: data})


def verify_code(id, code) -> bool:
    """
    Verifies that two-factor authentification code matches the database record
    """
    db_record = search_database(id=id)
    valid_code = code == db_record
    if valid_code:
        return True
    return False