- cullen merge

homeharvest/core/scrapers/models.py

@@ -1,7 +1,6 @@
 from dataclasses import dataclass
 from enum import Enum
-from typing import Tuple
-from datetime import datetime
+from typing import Optional
 
 
 class SiteName(Enum):
@@ -23,46 +22,13 @@ class ListingType(Enum):
     SOLD = "SOLD"
 
 
-class PropertyType(Enum):
-    HOUSE = "HOUSE"
-    BUILDING = "BUILDING"
-    CONDO = "CONDO"
-    TOWNHOUSE = "TOWNHOUSE"
-    SINGLE_FAMILY = "SINGLE_FAMILY"
-    MULTI_FAMILY = "MULTI_FAMILY"
-    MANUFACTURED = "MANUFACTURED"
-    NEW_CONSTRUCTION = "NEW_CONSTRUCTION"
-    APARTMENT = "APARTMENT"
-    APARTMENTS = "APARTMENTS"
-    LAND = "LAND"
-    LOT = "LOT"
-    OTHER = "OTHER"
-
-    BLANK = "BLANK"
-
-    @classmethod
-    def from_int_code(cls, code):
-        mapping = {
-            1: cls.HOUSE,
-            2: cls.CONDO,
-            3: cls.TOWNHOUSE,
-            4: cls.MULTI_FAMILY,
-            5: cls.LAND,
-            6: cls.OTHER,
-            8: cls.SINGLE_FAMILY,
-            13: cls.SINGLE_FAMILY,
-        }
-
-        return mapping.get(code, cls.BLANK)
-
-
 @dataclass
 class Address:
-    address_one: str | None = None
-    address_two: str | None = "#"
+    street: str | None = None
+    unit: str | None = None
     city: str | None = None
     state: str | None = None
-    zip_code: str | None = None
+    zip: str | None = None
 
 
 @dataclass
@@ -74,47 +40,31 @@ class Agent:
 
 @dataclass
 class Property:
-    property_url: str
-    site_name: SiteName
-    listing_type: ListingType
-    address: Address
-    property_type: PropertyType | None = None
-
-    # house for sale
-    tax_assessed_value: int | None = None
-    lot_area_value: float | None = None
-    lot_area_unit: str | None = None
-    stories: int | None = None
-    year_built: int | None = None
-    price_per_sqft: int | None = None
+    property_url: str | None = None
+    mls: str | None = None
     mls_id: str | None = None
+    status: str | None = None
+    style: str | None = None
 
-    agent: Agent | None = None
-    img_src: str | None = None
-    description: str | None = None
-    status_text: str | None = None
-    posted_time: datetime | None = None
+    beds: int | None = None
+    baths_full: int | None = None
+    baths_half: int | None = None
+    list_price: int | None = None
+    list_date: str | None = None
+    sold_price: int | None = None
+    last_sold_date: str | None = None
+    prc_sqft: float | None = None
+    est_sf: int | None = None
+    lot_sf: int | None = None
+    hoa_fee: int | None = None
 
-    # building for sale
-    bldg_name: str | None = None
-    area_min: int | None = None
-
-    beds_min: int | None = None
-    beds_max: int | None = None
-
-    baths_min: float | None = None
-    baths_max: float | None = None
-
-    sqft_min: int | None = None
-    sqft_max: int | None = None
-
-    price_min: int | None = None
-    price_max: int | None = None
-
-    unit_count: int | None = None
+    address: Address | None = None
 
+    yr_blt: int | None = None
     latitude: float | None = None
     longitude: float | None = None
 
-    sold_date: datetime | None = None
-    days_on_market: int | None = None
+    stories: int | None = None
+    prkg_gar: float | None = None
+
+    neighborhoods: Optional[str] = None