Use RestaurantPermit model for rest api

Author: msj

Makefile

@@ -1,10 +1,14 @@
 .PHONY: all
-all: data/raw/community-areas.geojson data/raw/chicago-restaurants.csv
+all: load_data
 
 .PHONY: clean
 clean:
 	rm data/raw/chicago-restaurants.csv data/raw/community-areas.geojson
 
+.PHONY: load_data
+load_data: data/raw/chicago-restaurants.csv data/raw/community-areas.geojson
+	python manage.py load_data data/raw/chicago-restaurants.csv
+
 data/raw/chicago-restaurants.csv:
 	curl "https://data.cityofchicago.org/api/v3/views/fr9j-f3pa/query.csv?query=SELECT%0A%20%20%60id%60%2C%0A%20%20%60permit_%60%2C%0A%20%20%60permit_type%60%2C%0A%20%20%60review_type%60%2C%0A%20%20%60application_start_date%60%2C%0A%20%20%60issue_date%60%2C%0A%20%20%60processing_time%60%2C%0A%20%20%60street_number%60%2C%0A%20%20%60street_direction%60%2C%0A%20%20%60street_name%60%2C%0A%20%20%60work_description%60%2C%0A%20%20%60building_fee_paid%60%2C%0A%20%20%60zoning_fee_paid%60%2C%0A%20%20%60other_fee_paid%60%2C%0A%20%20%60subtotal_paid%60%2C%0A%20%20%60building_fee_unpaid%60%2C%0A%20%20%60zoning_fee_unpaid%60%2C%0A%20%20%60other_fee_unpaid%60%2C%0A%20%20%60subtotal_unpaid%60%2C%0A%20%20%60building_fee_waived%60%2C%0A%20%20%60zoning_fee_waived%60%2C%0A%20%20%60other_fee_waived%60%2C%0A%20%20%60subtotal_waived%60%2C%0A%20%20%60total_fee%60%2C%0A%20%20%60reported_cost%60%2C%0A%20%20%60community_area%60%2C%0A%20%20%60census_tract%60%2C%0A%20%20%60ward%60%2C%0A%20%20%60latitude%60%2C%0A%20%20%60longitude%60%2C%0A%20%20%60location%60%0AWHERE%0A%20%20%60issue_date%60%0A%20%20%20%20BETWEEN%20%222016-01-01T00%3A00%3A00%22%20%3A%3A%20floating_timestamp%0A%20%20%20%20AND%20%222026-03-09T00%3A00%3A00%22%20%3A%3A%20floating_timestamp&app_token=$$CDP_APP_TOKEN" -o $@
 

docker-compose.yml

@@ -42,7 +42,7 @@ services:
 
   postgres:
     container_name: map-postgres
-    image: postgres
+    image: postgis/postgis
     healthcheck:
       test: ["CMD-SHELL", "pg_isready -U postgres"]
       interval: 10s

map/management/commands/load_data.py

@@ -0,0 +1,50 @@
+import csv
+from datetime import datetime
+
+from django.contrib.gis.geos.error import GEOSException
+from django.core.management.base import BaseCommand
+
+from map.models import RestaurantPermit
+
+
+class Command(BaseCommand):
+    help = "Load restaurant permit data from a CSV file"
+
+    def add_arguments(self, parser):
+        parser.add_argument(
+            "csv_file", type=str, help="The path to the CSV file to load"
+        )
+
+    def handle(self, *args, **kwargs):
+        RestaurantPermit.objects.all().delete()
+
+        csv_file = kwargs["csv_file"]
+
+        with open(csv_file, "r") as file:
+            reader = csv.DictReader(file)
+
+            for row in reader:
+                has_valid_dates = row["application_start_date"] and row["issue_date"]
+                if not has_valid_dates or not row["location"]:
+                    continue
+
+                try:
+                    restaurant = RestaurantPermit(
+                        permit_id=row["id"],
+                        permit_type=row["permit_type"],
+                        application_start_date=datetime.fromisoformat(
+                            row["application_start_date"]
+                        ),
+                        issue_date=datetime.fromisoformat(row["issue_date"]),
+                        work_description=row["work_description"],
+                        street_number=row["street_number"],
+                        street_direction=row["street_direction"],
+                        street_name=row["street_name"],
+                        location=row["location"],
+                    )
+                    restaurant.save()
+
+                except GEOSException:
+                    self.stdout.write(
+                        f'Invalid location for ID {row["id"]}. Skipping...'
+                    )

map/migrations/0001_initial.py

@@ -0,0 +1,30 @@
+# Generated by Django 6.0.3 on 2026-03-09 19:10
+
+import django.contrib.gis.db.models.fields
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Restaurant',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('permit_id', models.CharField(max_length=16)),
+                ('permit_type', models.CharField(max_length=64)),
+                ('application_start_date', models.DateField()),
+                ('issue_date', models.DateField()),
+                ('work_description', models.TextField(blank=True, null=True)),
+                ('street_number', models.CharField(max_length=16)),
+                ('street_direction', models.CharField(max_length=8)),
+                ('street_name', models.CharField(max_length=32)),
+                ('location', django.contrib.gis.db.models.fields.PointField(srid=4326)),
+            ],
+        ),
+    ]

map/migrations/0002_rename_restaurant_restaurantpermit.py

@@ -0,0 +1,17 @@
+# Generated by Django 6.0.3 on 2026-03-09 19:28
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('map', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.RenameModel(
+            old_name='Restaurant',
+            new_name='RestaurantPermit',
+        ),
+    ]

map/migrations/0003_alter_restaurantpermit_application_start_date.py

@@ -0,0 +1,18 @@
+# Generated by Django 6.0.3 on 2026-03-09 19:44
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('map', '0002_rename_restaurant_restaurantpermit'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='restaurantpermit',
+            name='application_start_date',
+            field=models.DateField(blank=True, null=True),
+        ),
+    ]

map/migrations/__init__.py

@@ -0,0 +1,14 @@
+from django.db import models
+from django.contrib.gis.db import models as gis_models
+
+
+class RestaurantPermit(models.Model):
+    permit_id = models.CharField(max_length=16, null=False, blank=False)
+    permit_type = models.CharField(max_length=64, null=False, blank=False)
+    application_start_date = models.DateField(null=True, blank=True)
+    issue_date = models.DateField(null=False, blank=False)
+    work_description = models.TextField(null=True, blank=True)
+    street_number = models.CharField(max_length=16, null=False, blank=False)
+    street_direction = models.CharField(max_length=8, null=False, blank=False)
+    street_name = models.CharField(max_length=32, null=False, blank=False)
+    location = gis_models.PointField(null=False, blank=False)

map/models.py

@@ -0,0 +1,9 @@
+from rest_framework import serializers
+
+from map.models import RestaurantPermit
+
+
+class RestaurantPermitSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = RestaurantPermit
+        fields = "__all__"

map/serializers.py

@@ -49,6 +49,7 @@
     "django.contrib.messages",
     "django.contrib.staticfiles",
     "django.contrib.postgres",
+    "rest_framework",
     "webpack_loader",
     "map",
     "storages",
@@ -105,6 +106,8 @@
     engine="django.db.backends.postgresql",
 )
 
+DATABASES["default"]["ENGINE"] = "django.contrib.gis.db.backends.postgis"
+
 
 # Caching
 # https://docs.djangoproject.com/en/stable/topics/cache/