Merge commit '4cf6f91e954b770198578bcb8db5d98ac964bfed' as 'common'

2026-03-28 02:24:14 +00:00
parent 27fe957074 4cf6f91e95
commit b736e62d4f
66 changed files with 7044 additions and 0 deletions
@@ -0,0 +1,203 @@
+"""Generate Store and StoreLocation seed data."""
+
+import uuid
+from datetime import UTC, datetime
+
+from cartsnitch_common.constants import StoreSlug
+from cartsnitch_common.seed.config import NUM_LOCATIONS_PER_STORE
+
+# Fixed store definitions
+_STORE_DEFS: list[dict] = [
+    {
+        "name": "Meijer",
+        "slug": StoreSlug.MEIJER,
+        "logo_url": "https://www.meijer.com/favicon.ico",
+        "website_url": "https://www.meijer.com",
+    },
+    {
+        "name": "Kroger",
+        "slug": StoreSlug.KROGER,
+        "logo_url": "https://www.kroger.com/favicon.ico",
+        "website_url": "https://www.kroger.com",
+    },
+    {
+        "name": "Target",
+        "slug": StoreSlug.TARGET,
+        "logo_url": "https://www.target.com/favicon.ico",
+        "website_url": "https://www.target.com",
+    },
+]
+
+# SE Michigan locations per store (5 each = 15 total)
+_LOCATION_DEFS: dict[StoreSlug, list[dict]] = {
+    StoreSlug.MEIJER: [
+        {
+            "address": "3145 Ann Arbor-Saline Rd",
+            "city": "Ann Arbor",
+            "state": "MI",
+            "zip": "48103",
+            "lat": 42.2434,
+            "lng": -83.8102,
+        },
+        {
+            "address": "700 W Ellsworth Rd",
+            "city": "Ann Arbor",
+            "state": "MI",
+            "zip": "48108",
+            "lat": 42.2318,
+            "lng": -83.7581,
+        },
+        {
+            "address": "5100 Oakman Blvd",
+            "city": "Dearborn",
+            "state": "MI",
+            "zip": "48126",
+            "lat": 42.3223,
+            "lng": -83.1952,
+        },
+        {
+            "address": "15555 Northline Rd",
+            "city": "Southgate",
+            "state": "MI",
+            "zip": "48195",
+            "lat": 42.2089,
+            "lng": -83.1953,
+        },
+        {
+            "address": "2855 Washtenaw Ave",
+            "city": "Ypsilanti",
+            "state": "MI",
+            "zip": "48197",
+            "lat": 42.2461,
+            "lng": -83.6388,
+        },
+    ],
+    StoreSlug.KROGER: [
+        {
+            "address": "2010 W Stadium Blvd",
+            "city": "Ann Arbor",
+            "state": "MI",
+            "zip": "48103",
+            "lat": 42.2706,
+            "lng": -83.7807,
+        },
+        {
+            "address": "1100 S Main St",
+            "city": "Ann Arbor",
+            "state": "MI",
+            "zip": "48104",
+            "lat": 42.2555,
+            "lng": -83.7469,
+        },
+        {
+            "address": "23650 Michigan Ave",
+            "city": "Dearborn",
+            "state": "MI",
+            "zip": "48124",
+            "lat": 42.3221,
+            "lng": -83.2135,
+        },
+        {
+            "address": "14000 Michigan Ave",
+            "city": "Dearborn",
+            "state": "MI",
+            "zip": "48126",
+            "lat": 42.3281,
+            "lng": -83.1789,
+        },
+        {
+            "address": "3965 Packard St",
+            "city": "Ann Arbor",
+            "state": "MI",
+            "zip": "48108",
+            "lat": 42.2298,
+            "lng": -83.7196,
+        },
+    ],
+    StoreSlug.TARGET: [
+        {
+            "address": "3165 Ann Arbor-Saline Rd",
+            "city": "Ann Arbor",
+            "state": "MI",
+            "zip": "48103",
+            "lat": 42.2431,
+            "lng": -83.8097,
+        },
+        {
+            "address": "4001 Carpenter Rd",
+            "city": "Ypsilanti",
+            "state": "MI",
+            "zip": "48197",
+            "lat": 42.2373,
+            "lng": -83.6617,
+        },
+        {
+            "address": "16000 Ford Rd",
+            "city": "Dearborn",
+            "state": "MI",
+            "zip": "48126",
+            "lat": 42.3312,
+            "lng": -83.2098,
+        },
+        {
+            "address": "17300 Eureka Rd",
+            "city": "Southgate",
+            "state": "MI",
+            "zip": "48195",
+            "lat": 42.2001,
+            "lng": -83.2014,
+        },
+        {
+            "address": "2400 E Stadium Blvd",
+            "city": "Ann Arbor",
+            "state": "MI",
+            "zip": "48104",
+            "lat": 42.2624,
+            "lng": -83.7102,
+        },
+    ],
+}
+
+
+def generate_stores() -> list[dict]:
+    """Return 3 fixed store records."""
+    now = datetime.now(tz=UTC)
+    stores = []
+    for defn in _STORE_DEFS:
+        stores.append(
+            {
+                "id": uuid.uuid4(),
+                "name": defn["name"],
+                "slug": defn["slug"],
+                "logo_url": defn["logo_url"],
+                "website_url": defn["website_url"],
+                "created_at": now,
+                "updated_at": now,
+            }
+        )
+    return stores
+
+
+def generate_store_locations(stores: list[dict]) -> list[dict]:
+    """Return 5 locations per store (15 total)."""
+    now = datetime.now(tz=UTC)
+    slug_to_id = {s["slug"]: s["id"] for s in stores}
+    locations = []
+    for slug, loc_defs in _LOCATION_DEFS.items():
+        store_id = slug_to_id[slug]
+        for loc in loc_defs[:NUM_LOCATIONS_PER_STORE]:
+            locations.append(
+                {
+                    "id": uuid.uuid4(),
+                    "store_id": store_id,
+                    "address": loc["address"],
+                    "city": loc["city"],
+                    "state": loc["state"],
+                    "zip": loc["zip"],
+                    "lat": loc["lat"],
+                    "lng": loc["lng"],
+                    "created_at": now,
+                    "updated_at": now,
+                }
+            )
+    return locations