feat: add metrics for copied and migrated projects (#897)

Co-authored-by: Tasko Olevski <olevski90@gmail.com>

Author: rokroskar

components/renku_data_services/base_models/metrics.py

@@ -54,7 +54,7 @@ async def session_launcher_created(
         """Send session launcher created event to metrics."""
         ...
 
-    async def project_created(self, user: APIUser) -> None:
+    async def project_created(self, user: APIUser, metadata: MetricsMetadata) -> None:
         """Send project created event to metrics."""
         ...
 
@@ -93,3 +93,11 @@ async def user_requested_session_launch(self, user: APIUser, metadata: MetricsMe
     async def user_requested_session_resume(self, user: APIUser, metadata: MetricsMetadata) -> None:
         """Send event about user requesting session resume."""
         ...
+
+
+class ProjectCreationType(StrEnum):
+    """The different types of project creation metrics."""
+
+    new = "new"
+    migrated = "migrated"
+    copied = "copied"

components/renku_data_services/metrics/core.py

@@ -53,9 +53,9 @@ async def session_launcher_created(
             {"environment_kind": environment_kind, "environment_image_source": environment_image_source},
         )
 
-    async def project_created(self, user: APIUser) -> None:
+    async def project_created(self, user: APIUser, metadata: MetricsMetadata) -> None:
         """Store project created event in staging table."""
-        await self._store_event(MetricsEvent.project_created, user)
+        await self._store_event(MetricsEvent.project_created, user, metadata)
 
     async def code_repo_linked_to_project(self, user: APIUser) -> None:
         """Store code repo linked to project event in staging table."""

components/renku_data_services/project/blueprints.py

@@ -20,7 +20,7 @@
 from renku_data_services.base_api.misc import validate_body_root_model, validate_query
 from renku_data_services.base_api.pagination import PaginationRequest, paginate
 from renku_data_services.base_models.core import Slug
-from renku_data_services.base_models.metrics import MetricsService
+from renku_data_services.base_models.metrics import MetricsService, ProjectCreationType
 from renku_data_services.base_models.validation import validate_and_dump, validated_json
 from renku_data_services.data_connectors.db import DataConnectorRepository
 from renku_data_services.errors import errors
@@ -82,7 +82,7 @@ def post(self) -> BlueprintFactoryResponse:
         async def _post(_: Request, user: base_models.APIUser, body: apispec.ProjectPost) -> JSONResponse:
             new_project = validate_unsaved_project(body, created_by=user.id or "")
             result = await self.project_repo.insert_project(user, new_project)
-            await self.metrics.project_created(user)
+            await self.metrics.project_created(user, metadata={"project_creation_kind": ProjectCreationType.new.value})
             if len(result.repositories) > 0:
                 await self.metrics.code_repo_linked_to_project(user)
             return validated_json(apispec.Project, self._dump_project(result), status=201)
@@ -131,6 +131,9 @@ async def _post_migration(
             result = await self.project_migration_repo.migrate_v1_project(
                 user, project=new_project, project_v1_id=v1_id, session_launcher=body.session_launcher
             )
+            await self.metrics.project_created(
+                user, metadata={"project_creation_kind": ProjectCreationType.migrated.value}
+            )
             return validated_json(apispec.Project, self._dump_project(result), status=201)
 
         return "/renku_v1_projects/<v1_id:int>/migrations", ["POST"], _post_migration
@@ -181,6 +184,9 @@ async def _copy(
                 session_repo=self.session_repo,
                 data_connector_repo=self.data_connector_repo,
             )
+            await self.metrics.project_created(
+                user, metadata={"project_creation_kind": ProjectCreationType.copied.value}
+            )
             return validated_json(apispec.Project, self._dump_project(project), status=201)
 
         return "/projects/<project_id:ulid>/copies", ["POST"], _copy

test/bases/renku_data_services/data_api/test_metrics.py

@@ -7,6 +7,7 @@
 import pytest_asyncio
 from sanic_testing.testing import SanicASGITestClient
 
+from renku_data_services.base_models.metrics import ProjectCreationType
 from renku_data_services.metrics.core import StagingMetricsService
 
 
@@ -40,7 +41,7 @@ async def test_metrics_are_stored(sanic_metrics_client, app_manager, create_proj
     project_created = events[0]
     assert re.match(r"^[0-7][0-9A-HJKMNP-TV-Z]{25}$", str(project_created.id))
     assert project_created.event == "project_created"
-    assert project_created.metadata_ is None
+    assert project_created.metadata_ == {"project_creation_kind": ProjectCreationType.new.value}
 
     session_launcher_created = events[1]
     assert re.match(r"^[0-7][0-9A-HJKMNP-TV-Z]{25}$", str(session_launcher_created.id))