Add monitoring PF service
This commit is contained in:
103
app/models.py
Normal file
103
app/models.py
Normal file
@@ -0,0 +1,103 @@
|
||||
from datetime import datetime
|
||||
from enum import Enum
|
||||
|
||||
from sqlalchemy import DateTime, Enum as SAEnum, Float, ForeignKey, Integer, String, Text, UniqueConstraint
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
|
||||
from app.db import Base
|
||||
|
||||
|
||||
class DealType(str, Enum):
|
||||
SALE = "sale"
|
||||
RENT = "rent"
|
||||
|
||||
|
||||
class Source(str, Enum):
|
||||
PROPERTYFINDER = "propertyfinder"
|
||||
BAYUT = "bayut"
|
||||
|
||||
|
||||
class ListingStatus(str, Enum):
|
||||
ACTIVE = "active"
|
||||
REMOVED = "removed"
|
||||
|
||||
|
||||
class Employee(Base):
|
||||
__tablename__ = "employees"
|
||||
|
||||
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
||||
name: Mapped[str] = mapped_column(String(200))
|
||||
tg_chat_id: Mapped[str | None] = mapped_column(String(64), unique=True, nullable=True)
|
||||
tg_username: Mapped[str | None] = mapped_column(String(200), nullable=True)
|
||||
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||||
|
||||
projects: Mapped[list["Project"]] = relationship(back_populates="owner")
|
||||
|
||||
|
||||
class Project(Base):
|
||||
"""Наш проект — квартира, которую агентство рекламирует."""
|
||||
|
||||
__tablename__ = "projects"
|
||||
|
||||
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
||||
title: Mapped[str] = mapped_column(String(300))
|
||||
deal_type: Mapped[DealType] = mapped_column(SAEnum(DealType))
|
||||
our_price: Mapped[float | None] = mapped_column(Float, nullable=True)
|
||||
notes: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||||
|
||||
# Опциональные параметры — используются для подсказок похожих объявлений
|
||||
dld_permit: Mapped[str | None] = mapped_column(String(100), index=True, nullable=True)
|
||||
building: Mapped[str | None] = mapped_column(String(300), nullable=True)
|
||||
bedrooms: Mapped[int | None] = mapped_column(Integer, nullable=True)
|
||||
size_sqft: Mapped[float | None] = mapped_column(Float, nullable=True)
|
||||
our_url: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||||
|
||||
owner_id: Mapped[int] = mapped_column(ForeignKey("employees.id"))
|
||||
owner: Mapped[Employee] = relationship(back_populates="projects")
|
||||
|
||||
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||||
last_checked_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
|
||||
|
||||
listings: Mapped[list["CompetitorListing"]] = relationship(
|
||||
back_populates="project", cascade="all, delete-orphan"
|
||||
)
|
||||
|
||||
|
||||
class CompetitorListing(Base):
|
||||
"""Объявление конкурента, найденное на PF/Bayut по DLD permit нашего проекта."""
|
||||
|
||||
__tablename__ = "competitor_listings"
|
||||
__table_args__ = (UniqueConstraint("project_id", "source", "external_id", name="uq_listing"),)
|
||||
|
||||
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
||||
project_id: Mapped[int] = mapped_column(ForeignKey("projects.id"))
|
||||
project: Mapped[Project] = relationship(back_populates="listings")
|
||||
|
||||
source: Mapped[Source] = mapped_column(SAEnum(Source))
|
||||
external_id: Mapped[str] = mapped_column(String(100)) # ID на стороне PF/Bayut
|
||||
url: Mapped[str] = mapped_column(Text)
|
||||
title: Mapped[str | None] = mapped_column(String(500), nullable=True)
|
||||
agent_name: Mapped[str | None] = mapped_column(String(300), nullable=True)
|
||||
agency_name: Mapped[str | None] = mapped_column(String(300), nullable=True)
|
||||
|
||||
current_price: Mapped[float | None] = mapped_column(Float, nullable=True)
|
||||
currency: Mapped[str | None] = mapped_column(String(10), nullable=True, default="AED")
|
||||
status: Mapped[ListingStatus] = mapped_column(SAEnum(ListingStatus), default=ListingStatus.ACTIVE)
|
||||
|
||||
first_seen_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||||
last_seen_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||||
|
||||
price_history: Mapped[list["PriceHistory"]] = relationship(
|
||||
back_populates="listing", cascade="all, delete-orphan", order_by="PriceHistory.recorded_at.desc()"
|
||||
)
|
||||
|
||||
|
||||
class PriceHistory(Base):
|
||||
__tablename__ = "price_history"
|
||||
|
||||
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
||||
listing_id: Mapped[int] = mapped_column(ForeignKey("competitor_listings.id"))
|
||||
listing: Mapped[CompetitorListing] = relationship(back_populates="price_history")
|
||||
|
||||
price: Mapped[float | None] = mapped_column(Float, nullable=True)
|
||||
recorded_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||||
Reference in New Issue
Block a user