from app import db
from database.models import *
from database.schema import *
from marshmallow import ValidationError


def insert_data(schema, data):
    instance = validate_data(schema=schema, data=data)
    db.session.add(instance)
    db.session.commit()


def delete_data(id):
    db.session.delete(data)
    db.session.commit()


def update_otp(user_id, otp):
    db.session.query(table="Users").filter_by(id=user_id).update(dict(otp=otp))
    db.session.commit()


def validate_data(schema, data):
    validation_schema = schema + "Schema()"
    instance = validation_schema
    try:
        output = instance.load(data).data
        return output
    except ValidationError as err:
        print(err.messages)