Remove obsolete cleanup from BundleAnalysis (#731)

graphql_api/actions/measurements.py

@@ -1,5 +1,5 @@
 from datetime import datetime
-from typing import Iterable, Mapping, Optional
+from typing import Any, Dict, Iterable, List, Optional
 
 from django.db.models import Max, QuerySet
 
@@ -16,7 +16,7 @@ def measurements_by_ids(
     before: datetime,
     after: Optional[datetime] = None,
     branch: Optional[str] = None,
-) -> Mapping[int, Iterable[dict]]:
+) -> Dict[int, List[Dict[str, Any]]]:
     queryset = MeasurementSummary.agg_by(interval).filter(
         name=measurable_name,
         owner_id=repository.author_id,
@@ -38,7 +38,7 @@ def measurements_by_ids(
     )
 
     # group by measurable_id
-    measurements: Mapping[int, Iterable[dict]] = {}
+    measurements: Dict[int, List[Dict[str, Any]]] = {}
     for measurement in queryset:
         measurable_id = measurement["measurable_id"]
         if measurable_id not in measurements:
@@ -51,7 +51,7 @@ def measurements_by_ids(
 def measurements_last_uploaded_before_start_date(
     repo_id: int,
     measurable_name: str,
-    measurable_ids: Iterable[str],
+    measurable_ids: List[int],
     start_date: datetime,
     branch: Optional[str] = None,
 ) -> QuerySet:

requirements.in

@@ -20,7 +20,7 @@ factory-boy
 fakeredis
 freezegun
 https://github.com/codecov/opentelem-python/archive/refs/tags/v0.0.4a1.tar.gz#egg=codecovopentelem
-https://github.com/codecov/shared/archive/14d8093f85adf1b99df763354c10ec0eaf86ee43.tar.gz#egg=shared
+https://github.com/codecov/shared/archive/58c6f4f0a0fa1a5d931466e0a63f5f15349546d5.tar.gz#egg=shared
 google-cloud-pubsub
 gunicorn>=22.0.0
 https://github.com/photocrowd/django-cursor-pagination/archive/f560902696b0c8509e4d95c10ba0d62700181d84.tar.gz

requirements.txt

@@ -205,6 +205,8 @@ googleapis-common-protos[grpc]==1.59.1
     #   grpcio-status
 graphql-core==3.2.3
     # via ariadne
+greenlet==3.0.3
+    # via sqlalchemy
 grpc-google-iam-v1==0.12.6
     # via google-cloud-pubsub
 grpcio==1.62.1
@@ -417,7 +419,7 @@ sentry-sdk[celery]==1.44.1
     #   shared
 setproctitle==1.1.10
     # via -r requirements.in
-shared @ https://github.com/codecov/shared/archive/14d8093f85adf1b99df763354c10ec0eaf86ee43.tar.gz
+shared @ https://github.com/codecov/shared/archive/58c6f4f0a0fa1a5d931466e0a63f5f15349546d5.tar.gz
     # via -r requirements.in
 simplejson==3.17.2
     # via -r requirements.in

services/bundle_analysis.py

@@ -3,7 +3,7 @@
 from dataclasses import dataclass
 from datetime import datetime
 from decimal import Decimal
-from typing import Any, Dict, Iterable, List, Optional
+from typing import Any, Dict, Iterable, List, Mapping, Optional
 
 from django.utils.functional import cached_property
 from shared.bundle_analysis import AssetReport as SharedAssetReport
@@ -84,7 +84,7 @@ def __init__(
         asset_type: BundleAnalysisMeasurementsAssetType,
         asset_name: Optional[str],
         interval: Interval,
-        after: datetime,
+        after: Optional[datetime],
         before: datetime,
     ):
         self.raw_measurements = raw_measurements
@@ -283,11 +283,6 @@ def is_cached(self) -> bool:
 class BundleAnalysisReport(object):
     def __init__(self, report: SharedBundleAnalysisReport):
         self.report = report
-        self.cleanup()
-
-    def cleanup(self) -> None:
-        if self.report and self.report.db_session:
-            self.report.db_session.close()
 
     def bundle(
         self, name: str, filters: Dict[str, List[str]]
@@ -323,13 +318,6 @@ def __init__(
             head_report_key,
         )
         self.head_report = self.comparison.head_report
-        self.cleanup()
-
-    def cleanup(self) -> None:
-        if self.comparison.head_report and self.comparison.head_report.db_session:
-            self.comparison.head_report.db_session.close()
-        if self.comparison.base_report and self.comparison.base_report.db_session:
-            self.comparison.base_report.db_session.close()
 
     @cached_property
     def bundles(self) -> List["BundleComparison"]:
@@ -391,7 +379,7 @@ def __init__(
 
     def _compute_measurements(
         self, measurable_name: str, measurable_ids: List[str]
-    ) -> dict[int, Iterable[dict[Any, Any]]]:
+    ) -> Dict[int, List[Dict[str, Any]]]:
         all_measurements = measurements_by_ids(
             repository=self.repository,
             measurable_name=measurable_name,