57 lines
1.4 KiB
Python
57 lines
1.4 KiB
Python
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
|