Пример #1
0
class Agreement(TimestampMixin, db.Model):
    __tablename__ = 'agreements'
    id = db.Column(db.Integer, primary_key=True)
    identifier = db.Column(db.String(50), unique=True, nullable=False, index=True)
    sign_date = db.Column(db.Date, nullable=False)
    date_from = db.Column(db.Date, nullable=False)
    date_to = db.Column(db.Date, nullable=False)
    price_value = db.Column(db.Float, nullable=False)
    price_period = db.Column(db.String(10), nullable=False)  #'day'/'month'
    payment_deadline = db.Column(db.Integer, nullable=False)
    deposit_value = db.Column(db.Float, default=0)
    description = db.Column(db.Text)
    flat_id = db.Column(db.Integer, db.ForeignKey('flats.id'), nullable=False)
    tenant_id = db.Column(db.Integer, db.ForeignKey('tenants.id'), nullable=False)
    flat = db.relationship('Flat', back_populates='agreements')
    tenant = db.relationship('Tenant', back_populates='agreements')
    settlements = db.relationship('Settlement', back_populates='agreement')

    def __repr__(self):
        return f'<agreement>: {self.identifier} - {self.flat} - {self.tenant}'

    @staticmethod
    def additional_validation(param: str, value: str) -> str:
        if param in ['sign_date', 'date_from', 'date_to']:
            try:
                value = datetime.strptime(value, '%d-%m-%Y').date()
            except ValueError:
                value = None
        return value
Пример #2
0
class Tenant(TimestampMixin, db.Model):
    __tablename__ = 'tenants'
    id = db.Column(db.Integer, primary_key=True)
    identifier = db.Column(db.String(255), unique=True, nullable=False, index=True)
    email = db.Column(db.String(255), unique=True, nullable=False)
    first_name = db.Column(db.String(100), nullable=False)
    last_name = db.Column(db.String(100), nullable=False)
    phone = db.Column(db.String(50), nullable=False)
    address = db.Column(db.String(255), nullable=False)
    description = db.Column(db.Text)
    password = db.Column(db.String(255), nullable=False)
    landlord_id = db.Column(db.Integer, db.ForeignKey('landlords.id'), nullable=False)
    landlord = db.relationship('Landlord', back_populates='tenants')
    agreements = db.relationship('Agreement', back_populates='tenant')

    def __repr__(self):
        return f'<tenant>: {self.first_name} {self.last_name}'

    @staticmethod
    def additional_validation(param: str, value: str) -> str:
        return value         

    def generate_jwt(self) -> bytes:
        jwt_expired_minutes = current_app.config.get('JWT_EXPIRED_MINUTES', 30)
        payload = {
            'id': self.id,
            'model': 'tenants',
            'exp': datetime.utcnow() + timedelta(minutes=jwt_expired_minutes)
        }

        return jwt.encode(payload, current_app.config.get('SECRET_KEY'))

    def is_password_valid(self, password: str) -> bool:
        return check_password_hash(self.password, password)
Пример #3
0
class Landlord(TimestampMixin, db.Model):
    __tablename__ = "landlords"
    id = db.Column(db.Integer, primary_key=True)
    identifier = db.Column(db.String(255),
                           unique=True,
                           nullable=False,
                           index=True)
    email = db.Column(db.String(255), unique=True, nullable=False)
    first_name = db.Column(db.String(100), nullable=False)
    last_name = db.Column(db.String(100), nullable=False)
    phone = db.Column(db.String(50), nullable=False)
    address = db.Column(db.String(255), nullable=False)
    description = db.Column(db.Text)
    password = db.Column(db.String(255), nullable=False)
    flats = db.relationship("Flat", back_populates="landlord")
    tenants = db.relationship("Tenant", back_populates="landlord")

    def __repr__(self):
        return f"<landlord>: {self.first_name} {self.last_name}"

    @staticmethod
    def additional_validation(param: str, value: str) -> str:
        return value

    def generate_jwt(self) -> bytes:
        jwt_expired_minutes = current_app.config.get("JWT_EXPIRED_MINUTES", 30)
        payload = {
            "id": self.id,
            "model": "landlords",
            "exp": datetime.utcnow() + timedelta(minutes=jwt_expired_minutes),
        }
        return jwt.encode(payload, current_app.config.get("SECRET_KEY"))

    def is_password_valid(self, password: str) -> bool:
        return check_password_hash(self.password, password)
Пример #4
0
class Flat(TimestampMixin, db.Model):
    __tablename__ = 'flats'
    id = db.Column(db.Integer, primary_key=True)
    identifier = db.Column(db.String(255), unique=True, nullable=False)
    address = db.Column(db.String(255), nullable=False)
    description = db.Column(db.Text)
    status = db.Column(db.String(50), default='active')  #active/inactive/sold
    landlord_id = db.Column(db.Integer, db.ForeignKey('landlords.id'), nullable=False)
    landlord = db.relationship('Landlord', back_populates='flats')
    agreements = db.relationship('Agreement', back_populates='flat')
    pictures = db.relationship('Picture', back_populates='flat')

    def __repr__(self):
        return f'<flat>: {self.id} {self.identifier}'

    @staticmethod
    def additional_validation(param: str, value: str) -> str:
        return value         
Пример #5
0
class Flat(TimestampMixin, db.Model):
    __tablename__ = "flats"
    id = db.Column(db.Integer, primary_key=True)
    identifier = db.Column(db.String(255), unique=True, nullable=False)
    address = db.Column(db.String(255), nullable=False)
    description = db.Column(db.Text)
    status = db.Column(db.String(50), default="active")  # active/inactive/sold
    landlord_id = db.Column(db.Integer,
                            db.ForeignKey("landlords.id"),
                            nullable=False)
    landlord = db.relationship("Landlord", back_populates="flats")
    agreements = db.relationship("Agreement", back_populates="flat")
    pictures = db.relationship("Picture", back_populates="flat")

    def __repr__(self):
        return f"<flat>: {self.id} {self.identifier}"

    @staticmethod
    def additional_validation(param: str, value: str) -> str:
        return value
Пример #6
0
class Picture(TimestampMixin, db.Model):
    __tablename__ = 'pictures'
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(50), unique=True, nullable = False)
    path = db.Column(db.String(255), nullable=False)
    description = db.Column(db.Text)
    flat_id = db.Column(db.Integer, db.ForeignKey('flats.id'), nullable=False)
    flat = db.relationship('Flat', back_populates='pictures')

    def __repr__(self):
        return f'<picture>: {self.id} - {self.name}'

    @staticmethod
    def additional_validation(param: str, value: str) -> str:
        return value         
Пример #7
0
class Settlement(TimestampMixin, db.Model):
    __tablename__ = 'settlements'
    id = db.Column(db.Integer, primary_key=True)
    type = db.Column(db.String(50), nullable=False)
    value = db.Column(db.Float, nullable=False)
    date = db.Column(db.Date, nullable=False, default=datetime.now().date())
    description = db.Column(db.Text)
    agreement_id = db.Column(db.Integer, db.ForeignKey('agreements.id'), nullable=False)
    agreement = db.relationship('Agreement', back_populates='settlements')

    def __repr__(self):
        return f'<settlement>: {self.id} {self.agreement}'

    @staticmethod
    def additional_validation(param: str, value: str) -> str:
        if param == 'date':
            try:
                value = datetime.strptime(value, '%d-%m-%Y').date()
            except ValueError:
                value = None
        return value