VivianDee
54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
from datetime import datetime, timezone
|
|
from app.extensions import db
|
|
from app.models.charge import Charge
|
|
from sqlalchemy.orm import relationship
|
|
|
|
class Offer(db.Model):
|
|
__tablename__ = 'offers'
|
|
|
|
id = db.Column(db.String, primary_key=True)
|
|
product_id = db.Column(db.String, nullable=False)
|
|
min_amount = db.Column(db.Float, nullable=False)
|
|
max_amount = db.Column(db.Float, nullable=False)
|
|
tenor = db.Column(db.Integer, nullable=False)
|
|
created_at = db.Column(db.DateTime, default=datetime.now(timezone.utc))
|
|
updated_at = db.Column(db.DateTime, default=datetime.now(timezone.utc), onupdate=datetime.now(timezone.utc))
|
|
|
|
charges = relationship(
|
|
"Charge",
|
|
primaryjoin="Offer.id == Charge.offer_id",
|
|
foreign_keys="Charge.offer_id",
|
|
back_populates="offer",
|
|
)
|
|
|
|
@classmethod
|
|
def get_all_offers(cls):
|
|
"""
|
|
Return all offers in dictionary format.
|
|
"""
|
|
offers = cls.query.all()
|
|
|
|
if not offers:
|
|
raise ValueError(f"No available offers")
|
|
return offers
|
|
|
|
@classmethod
|
|
def is_valid_offer(cls, offer_id):
|
|
offer = cls.query.filter_by(id=str(offer_id)).first()
|
|
|
|
|
|
if not offer:
|
|
return False
|
|
return offer
|
|
|
|
def to_dict(self):
|
|
return {
|
|
"offerId": self.id,
|
|
"productId": self.product_id,
|
|
"minAmount": self.min_amount,
|
|
"maxAmount": self.max_amount,
|
|
"tenor": self.tenor
|
|
}
|
|
|
|
def __repr__(self):
|
|
return f'<LoanOffer {self.id}>' |