class UserToken(BaseModel):
"""
Token Model for storing JWT tokens
"""
__tablename__ = 'user_token'
id = db.Column(db.INTEGER, primary_key=True, autoincrement=True)
user_id = db.Column(db.INTEGER, db.ForeignKey('user.id', ondelete='CASCADE'))
token = db.Column(db.String(1000), unique=True, nullable=False)
refresh_token = db.Column(db.String(1000), unique=True, nullable=False)
expired_at = db.Column(db.DateTime, nullable=False)
def __init__(self, token, refresh_token, user_id: int):
self.user_id = user_id
self.token = token
self.refresh_token = refresh_token
self.expired_at = datetime.utcnow() + timedelta(days=365)
def __repr__(self):
return '<UserToken: token: {}>'.format(self.token)
@classmethod
def is_token_valid(cls, auth_token):
res = cls.query.filter(cls.token == auth_token,
cls.expired_at >= datetime.utcnow(),
cls.is_deleted.is_(False)).first()
return res if True else False
@classmethod
def delete_token(cls, auth_token):
user_token = cls.query.filter_by(token=auth_token).first()
if user_token:
user_token.is_deleted = True
user_token.expired_at = datetime.utcnow()
user_token.update()
class UserQAnswerModel(BaseModel):
__tablename__ = 'user_question_releation'
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
question_id = db.Column(db.Integer,
db.ForeignKey('question.id'),
nullable=False)
answer_id = db.Column(db.Integer,
db.ForeignKey('question_option.id'),
nullable=False)
question = db.relationship('QuestionModel', uselist=True, lazy=True)
@classmethod
def get_parser_submit_qanswer(cls):
parser = reqparse.RequestParser(bundle_errors=True, trim=True)
parser.add_argument('question_id', required=True, type=str)
parser.add_argument('answer_id', required=True, type=str)
return parser
class FavEntityModel(BaseModel):
__tablename__ = 'fav_entity'
user_id = db.Column(db.ForeignKey('user.id'), nullable=False, index=True)
entity_id = db.Column(db.String(45), nullable=False)
entity_type = db.Column(db.INT, nullable=False, default=0)
# product = db.relationship('ProductModel')
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
class QOptionsModel(BaseModel):
__tablename__ = 'question_option'
question_id = db.Column(db.Integer,
db.ForeignKey('question.id'),
nullable=False)
option = db.Column(db.String(80), nullable=False)
option_icon = db.Column(db.String(100), nullable=False)
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
class UserRole(db.Model):
__tablename__ = 'user_role'
id = db.Column(db.Integer(), primary_key=True, autoincrement=True)
role_type = db.Column(db.String)
user_id = db.Column(db.Integer(),
db.ForeignKey('user.id', ondelete='CASCADE'),
unique=True)
# def __init__(self, role_type: str, user_id: int):
# if RoleType.has_value(role_type):
# self.role_type = role_type
# self.user_id = user_id
def __init__(self, role: RoleType, user_id: int):
self.role_type = role.value
self.user_id = user_id
def __init__(self, role: RoleType):
self.role_type = role.value
class NotificaionModel(BaseModel):
__tablename__ = 'notification'
title = db.Column(db.String(100), nullable=False)
body = db.Column(db.String(250), nullable=False)
user_id = db.Column(db.Integer(),
db.ForeignKey('user.id', ondelete='CASCADE'),
unique=True)
# sender_id = db.Column(db.Integer(), db.ForeignKey('user.id', ondelete='CASCADE'), unique=True)
is_seen = db.Column(db.Boolean, nullable=False, default=0)
is_viewed = db.Column(db.Boolean, nullable=False, default=0)
type = db.Column(db.Integer, nullable=False, default=0)
entity_id = db.Column(db.Integer, nullable=True, default=0)
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
@classmethod
def get_parser_fetch_notification(cls):
from api.common.base.parsers import base_pagination_parser
parser = base_pagination_parser.copy()
return parser
class BookingModel(BaseModel):
__tablename__ = 'booking'
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
booking_id = db.Column(db.String(45), nullable=False)
providerConfirmationId = db.Column(
db.String(45),
nullable=False) # GDS Confirmation Number. If you call the
# Provider, this Reference may be asked
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
# associated_record =
hotel_name = db.Column(db.String(45), nullable=False)
checking_date = db.Column(db.DateTime, nullable=False)
checkout_date = db.Column(db.DateTime, nullable=False)
number_of_guest = db.Column(db.Integer, nullable=False)
number_of_room = db.Column(db.Integer, nullable=False)
city = db.Column(db.String(30), nullable=False)
address = db.Column(db.String(45), nullable=False)
booking_status = db.Column(
db.Integer, nullable=False,
default=BookingStatus.CONFIRM.value) # booking_status
# @classmethod
# def save_booking_info(cls, data: dict, ):
# booking = BookingModel(**data)
@classmethod
def _get_parser_hotel_booking_details(cls):
from flask_restplus import reqparse
parser = reqparse.RequestParser(bundle_errors=True, trim=True)
parser.add_argument('hotel_name', type=str, required=True)
parser.add_argument('checking_date',
type=str,
required=True,
help='Date Of Birth (YYYY-MM-DD)')
parser.add_argument('checkout_date',
type=str,
required=True,
help='Date Of Birth (YYYY-MM-DD)')
# parser.add_argument('number_of_guest', type=str, required=True)
parser.add_argument('number_of_room', type=str, required=True)
parser.add_argument('city', type=str, required=True)
parser.add_argument('address', type=str, required=True)
return parser
@classmethod
def get_parser_booking(cls):
from flask_restplus import reqparse
parser = BookingModel._get_parser_hotel_booking_details()
parser.add_argument('offerId', type=str, required=True)
guests = '''sample - [
{
"name": {
"title": "MR",
"firstName": "BOB",
"lastName": "SMITH"
},
"contact": {
"phone": "+33679278416",
"email": "*****@*****.**"
}
}
]'''
parser.add_argument('guests',
type=list,
required=True,
location='json',
help=guests)
payments = '''{ "payments": [
{
"method": "creditCard",
"card": {
"vendorCode": "VI",
"cardNumber": "4111111111111111",
"expiryDate": "2023-01"
}
}
]}'''
parser.add_argument('payments',
type=list,
required=False,
location='json',
help=payments)
help = '''offerId, guests, payments and
optional rooms for the repartition (when used the rooms array items must match the shopping offer
roomQuantity)'''
return parser