odapi/database/crud.py

from datetime import datetime
from app import db
from app.models import *
from app.schema import *
from marshmallow import ValidationError
from pydoc import locate


def insert_data(schema, data):
    validate_data(schema=schema, data=data)
    model = locate("app.models." + schema)
    instance = model(**data)
    db.session.add(instance)
    db.session.commit()


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


def save_otp(mobile, otp):
    db.session.query(table="Users").filter_by(mobile=mobile).update(dict(otp=otp))
    db.session.commit()


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


def fetch_stored_otp(mobile):
    user = db.session.query(table="Users").filter_by(mobile=mobile)
    otp = user.otp
    return otp


def validate_account(mobile):
    timestamp = datetime.now()
    db.session.query(table="Users").filter_by(mobile=mobile).update(
        dict(otp_valid_time=timestamp)
    )
    db.session.query(table="Users").filter_by(mobile=mobile).update(dict(status=1))


def verify_otp(mobile, otp):
    stored_otp = fetch_stored_otp(mobile=mobile)
    if stored_otp == otp:
        validate_account(mobile=mobile)
        return True
    return False