from sqlalchemy import or_

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

db = SessionLocal()


def search_database(id):
    return db.query(Users).filter(
        or_(Users.correo_institucional == id, Users.numero_de_documento == id)
    )


def insert_data(data):
    item = Users(**data)
    db.add(item)
    db.commit()
    db.refresh(item)


def create_database(data):
    models.Base.metadata.create_all(bind=engine)
    existing_row = search_database(data["correo_institucional"])
    if existing_row:
        return
    insert_data(data)