Add Validate project type

Author: frozenhelium

apps/project/graphql/inputs.py

@@ -10,6 +10,7 @@
 from apps.project.project_types.tile_map_service.compare import project as compare_project
 from apps.project.project_types.tile_map_service.completeness import project as completeness_project
 from apps.project.project_types.tile_map_service.find import project as find_project
+from apps.project.project_types.validate import project as validate_project
 from utils.geo.tile_server.models import TileServerCommonConfig, TileServerConfig, TileServerCustomConfig
 
 
@@ -26,6 +27,10 @@ class TileServerCommonConfigInput: ...
 class ProjectTileServerConfigInput: ...
 
 
+@strawberry.experimental.pydantic.input(model=validate_project.ValidateObjectSourceConfig, all_fields=True)
+class ValidateObjectSourceConfigInput: ...
+
+
 # Project Properties
 @strawberry.experimental.pydantic.input(model=compare_project.CompareProjectProperty, all_fields=True)
 class CompareProjectPropertyInput: ...
@@ -39,6 +44,10 @@ class FindProjectPropertyInput: ...
 class CompletenessProjectPropertyInput: ...
 
 
+@strawberry.experimental.pydantic.input(model=validate_project.ValidateProjectProperty, all_fields=True)
+class ValidateProjectPropertyInput: ...
+
+
 @strawberry_django.input(Organization)
 class OrganizationCreateInput(UserResourceCreateInputMixin):
     name: strawberry.auto
@@ -49,6 +58,7 @@ class ProjectTypeSpecificInput:
     compare: CompareProjectPropertyInput | None = strawberry.UNSET
     find: FindProjectPropertyInput | None = strawberry.UNSET
     completeness: CompletenessProjectPropertyInput | None = strawberry.UNSET
+    validate: ValidateProjectPropertyInput | None = strawberry.UNSET
 
 
 # NOTE: Make sure this matches with the serializers ../serializers.py

apps/project/graphql/types.py

@@ -8,6 +8,7 @@
 from apps.project.project_types.tile_map_service.compare import project as compare_project
 from apps.project.project_types.tile_map_service.completeness import project as completeness_project
 from apps.project.project_types.tile_map_service.find import project as find_project
+from apps.project.project_types.validate import project as validate_project
 from apps.tutorial.graphql.types import TutorialType
 from utils.geo.tile_server.models import TileServerCommonConfig, TileServerConfig, TileServerCustomConfig
 
@@ -31,6 +32,10 @@ class ProjectTileServerCommonConfig: ...
 class ProjectTileServerConfig: ...
 
 
+@strawberry.experimental.pydantic.type(model=validate_project.ValidateObjectSourceConfig, all_fields=True)
+class ValidateObjectSourceConfig: ...
+
+
 # Project Properties
 @strawberry.experimental.pydantic.type(model=compare_project.CompareProjectProperty, all_fields=True)
 class CompareProjectPropertyType: ...
@@ -40,6 +45,10 @@ class CompareProjectPropertyType: ...
 class FindProjectPropertyType: ...
 
 
+@strawberry.experimental.pydantic.type(model=validate_project.ValidateProjectProperty, all_fields=True)
+class ValidateProjectPropertyType: ...
+
+
 @strawberry.experimental.pydantic.type(model=completeness_project.CompletenessProjectProperty, all_fields=True)
 class CompletenessProjectPropertyType: ...
 
@@ -79,14 +88,22 @@ class ProjectType(UserResourceTypeMixin):
     async def project_type_specifics(
         self,
         project: strawberry.Parent[Project],
-    ) -> CompareProjectPropertyType | FindProjectPropertyType | CompletenessProjectPropertyType | None:
+    ) -> (
+        CompareProjectPropertyType
+        | FindProjectPropertyType
+        | ValidateProjectPropertyType
+        | CompletenessProjectPropertyType
+        | None
+    ):
         data = project.project_type_specifics
         if data is None:
             return None
         if project.project_type_enum == Project.Type.FIND:
             return typing.cast("FindProjectPropertyType", find_project.FindProjectProperty.model_validate(data))
         if project.project_type_enum == Project.Type.COMPARE:
             return typing.cast("CompareProjectPropertyType", compare_project.CompareProjectProperty.model_validate(data))
+        if project.project_type_enum == Project.Type.VALIDATE:
+            return typing.cast("ValidateProjectPropertyType", validate_project.ValidateProjectProperty.model_validate(data))
         if project.project_type_enum == Project.Type.COMPLETENESS:
             return typing.cast(
                 "CompletenessProjectPropertyType",

apps/project/migrations/0004_alter_projecttask_geometry.py

@@ -0,0 +1,19 @@
+# Generated by Django 5.1.6 on 2025-05-29 10:44
+
+import django.contrib.gis.db.models.fields
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('project', '0003_alter_organization_client_id_alter_project_client_id_and_more'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='projecttask',
+            name='geometry',
+            field=django.contrib.gis.db.models.fields.GeometryField(blank=True, default=None, null=True, srid=4326),
+        ),
+    ]

apps/project/models.py

@@ -46,8 +46,8 @@ class ProjectTypeEnum(models.IntegerChoices):
     FIND = 1, "Find"
     """ Find project type. Previously known as Classification / Build Area. """
 
-    # FOOTPRINT = 2, "Validate"
-    # """ Validate project type. Previously known as Footprint """
+    VALIDATE = 2, "Validate"
+    """ Validate project type. Previously known as Footprint """
 
     COMPARE = 3, "Compare"
     """ Compare project type. Previously known as Change Detection. """
@@ -393,9 +393,9 @@ class ProjectTask(models.Model):
         related_name="+",
     )
 
-    # NOTE(tnagorra): The geometry is only necessary for footprint project type
+    # NOTE(tnagorra): The geometry is only necessary for validate project type
     # FIXME(thenav56): Existing gid_models.MultiPolygonField(srid=4326, blank=True, null=True)
-    geometry = gis_models.GeometryField(null=True, blank=True, default=None, dim=3)
+    geometry = gis_models.GeometryField(null=True, blank=True, default=None, dim=2)
 
     # TODO(thenav56): Currently this field collects any data not stored by another fields, pulled from firebase.
     # Also, used in SQL queries

apps/project/project_types/base/project.py

@@ -29,8 +29,8 @@ def get_max_time_spend_percentile(project_type: ProjectTypeEnum) -> float:
             return 1.4
         case ProjectTypeEnum.COMPARE:
             return 11.2
-        # case ProjectTypeEnum.FOOTPRINT:
-        #     return 6.1
+        case ProjectTypeEnum.VALIDATE:
+            return 6.1
         # case ProjectTypeEnum.STREET:
         #     return 65
 

apps/project/project_types/store.py

@@ -5,8 +5,9 @@
 from .tile_map_service.compare.project import CompareProject
 from .tile_map_service.completeness.project import CompletenessProject
 from .tile_map_service.find.project import FindProject
+from .validate.project import ValidateProject
 
-type ProjectTypeHandlers = type[CompareProject | FindProject | CompletenessProject]
+type ProjectTypeHandlers = type[CompareProject | ValidateProject | FindProject | CompletenessProject]
 
 
 @typing.overload
@@ -21,6 +22,12 @@ def get_project_type_handler(
 ) -> type[CompareProject]: ...
 
 
+@typing.overload
+def get_project_type_handler(
+    project_type: typing.Literal[ProjectTypeEnum.VALIDATE],
+) -> type[ValidateProject]: ...
+
+
 @typing.overload
 def get_project_type_handler(
     project_type: typing.Literal[ProjectTypeEnum.COMPLETENESS],
@@ -33,5 +40,7 @@ def get_project_type_handler(project_type: ProjectTypeEnum) -> ProjectTypeHandle
             return FindProject
         case ProjectTypeEnum.COMPARE:
             return CompareProject
+        case ProjectTypeEnum.VALIDATE:
+            return ValidateProject
         case ProjectTypeEnum.COMPLETENESS:
             return CompletenessProject

apps/project/project_types/tile_map_service/base/project.py

@@ -231,7 +231,9 @@ def validate(self):
 
         extension = Path(aoi_asset.file.name).suffix
         with tempfile.NamedTemporaryFile(suffix=extension, dir=settings.TEMP_DIR) as temp_file:
-            temp_file.write(aoi_asset.file.read())
+            # FIXME(frozenhelium): close the aoi_asset file?
+            with aoi_asset.file.open() as aoi_file:
+                temp_file.write(aoi_file.read())
             temp_file.flush()
 
             aoi_geometry = tile_grouping.get_geometry_from_file(temp_file.name)

apps/project/project_types/tile_map_service/completeness/project.py

@@ -1,18 +1,37 @@
-from apps.project.project_types.tile_map_service.compare import project as compare
+import typing
 
+from apps.project.models import Project, ProjectTypeEnum
+from apps.project.project_types.tile_map_service.base import project as base_project
+from utils.geo.tile_server.models import TileServerConfig
+from utils.geo.tile_server.tile_server import AvailableTileServerTypeAlias, get_tile_server
 
-# NOTE: If we want to extend these property, we will need to update CompletenessProject
-class CompletenessProjectProperty(compare.CompareProjectProperty): ...
 
+class CompletenessProjectProperty(base_project.TileMapServiceProjectProperty):
+    tile_server_b_property: TileServerConfig
 
-# NOTE: If we want to extend these property, we will need to update CompletenessProject
-class CompletenessProjectTaskGroupProperty(compare.CompareProjectTaskGroupProperty): ...
 
+class CompletenessProjectTaskGroupProperty(base_project.TileMapServiceProjectTaskGroupProperty): ...
 
-# NOTE: If we want to extend these property, we will need to update CompletenessProject
-class CompletenessProjectTaskProperty(compare.CompareProjectTaskProperty): ...
+
+class CompletenessProjectTaskProperty(base_project.TileMapServiceProjectTaskProperty):
+    url_b: str
 
 
 class CompletenessProject(
-    compare.CompareProject,
-): ...
+    base_project.TileMapServiceBaseProject[
+        CompletenessProjectProperty,
+        CompletenessProjectTaskGroupProperty,
+        CompletenessProjectTaskProperty,
+    ],
+):
+    tile_server_b: AvailableTileServerTypeAlias
+
+    project_property_class = CompletenessProjectProperty
+    project_task_group_property_class = CompletenessProjectTaskGroupProperty
+    project_task_property_class = CompletenessProjectTaskProperty
+
+    def __init__(self, project: Project):
+        super().__init__(project)
+        if typing.TYPE_CHECKING:
+            assert project.project_type == ProjectTypeEnum.COMPLETENESS, f"{type(self)} is defined for COMPLETENESS"
+        self.tile_server_b = get_tile_server(self.project_type_specifics.tile_server_b_property)