models.py

import csv
import json
from dataclasses import dataclass, field, asdict
from typing import List, Optional
from datetime import datetime

from config import CSV_FIELDS


@dataclass
class RentalListing:
    property_id: str
    rental_id: str = ""
    property_type: str = ""
    address: str = ""
    city: str = ""
    state: str = ""
    zip_code: str = ""
    price_min: Optional[int] = None
    price_max: Optional[int] = None
    bedrooms_min: Optional[int] = None
    bedrooms_max: Optional[int] = None
    bathrooms_min: Optional[float] = None
    bathrooms_max: Optional[float] = None
    sqft_min: Optional[int] = None
    sqft_max: Optional[int] = None
    lat: Optional[float] = None
    lon: Optional[float] = None
    url: str = ""
    description: str = ""
    feed_source: str = ""
    last_updated: str = ""
    date_available: str = ""
    num_available_units: Optional[int] = None
    date_scraped: str = field(
        default_factory=lambda: datetime.utcnow().isoformat() + "Z"
    )

    def to_dict(self) -> dict:
        return asdict(self)


@dataclass
class ListingCollection:
    items: List[RentalListing] = field(default_factory=list)
    _seen_ids: set = field(default_factory=set, repr=False)

    def add(self, listing: RentalListing) -> bool:
        if listing.property_id and listing.property_id in self._seen_ids:
            return False
        if listing.property_id:
            self._seen_ids.add(listing.property_id)
        self.items.append(listing)
        return True

    def add_many(self, listings: List[RentalListing]) -> int:
        return sum(1 for item in listings if self.add(item))

    def __len__(self) -> int:
        return len(self.items)

    def __iter__(self):
        return iter(self.items)

    def to_csv(self, filepath: str) -> None:
        if not self.items:
            return
        with open(filepath, "w", newline="", encoding="utf-8") as f:
            writer = csv.DictWriter(f, fieldnames=CSV_FIELDS)
            writer.writeheader()
            for item in self.items:
                writer.writerow(item.to_dict())

    def to_json(self, filepath: str) -> None:
        with open(filepath, "w", encoding="utf-8") as f:
            json.dump(
                [item.to_dict() for item in self.items],
                f,
                indent=2,
                ensure_ascii=False,
            )