Add endpoint to get zip code by ID

.gitignore

@@ -1,3 +1,3 @@
 .sqlhistory
 .dolt
-server/zip_codes.db
+server/zip_codes.dbserver/__pycache__/

server/main.py

@@ -162,6 +162,17 @@ def get_nearest_zip(
 ):
     return find_nearest(lat, lng)
 
+@app.get("/id/{intId}", response_model=ZipCodeData)
+def get_zip_by_id(intId: int):
+    cursor = state.db_connection.cursor()
+    cursor.execute("SELECT * FROM zip_codes WHERE id = ?", (intId,))
+    row = cursor.fetchone()
+
+    if not row:
+        raise HTTPException(status_code=404, detail="Zip code not found")
+
+    return dict(row)
+
 @app.get("/{query}", response_model=ZipCodeData)
 def get_zip_or_coords(query: str):
     # Check if input looks like coordinates "lat,lng"

server/test_main.py

@@ -137,6 +137,27 @@ def test_get_zips_invalid_params():
     assert response.status_code == 400
     assert "Invalid order" in response.json()["detail"]
 
+def test_get_zip_by_id():
+    # First get a valid ID from the random endpoint
+    response = client.get("/random")
+    assert response.status_code == 200
+    random_zip_data = response.json()
+    valid_id = random_zip_data["id"]
+
+    # Query by that ID
+    response = client.get(f"/id/{valid_id}")
+    assert response.status_code == 200
+    data = response.json()
+    assert data["id"] == valid_id
+    assert data["zip"] == random_zip_data["zip"]
+
+def test_get_zip_by_id_invalid():
+    # Assuming 0 or negative IDs don't exist, or a very large number
+    # Based on previous schema check, id is INTEGER PRIMARY KEY, usually starting at 1.
+    response = client.get("/id/0")
+    assert response.status_code == 404
+    assert response.json()["detail"] == "Zip code not found"
+
 def test_get_zips_city_state_filter():
     response = client.get("/zips?city_and_state_only=true")
     assert response.status_code == 200