Update geocoder in basehandler

Author: sudan45

apps/etl/transform/sources/noaa_ibtracs.py

@@ -4,7 +4,7 @@
 from pystac_monty.geocoding import GAULGeocoder
 from pystac_monty.sources.ibtracs import IBTrACSDataSource, IBTrACSTransformer
 
-from apps.etl.models import ExtractionData, Transform
+from apps.etl.models import ExtractionData
 from apps.etl.transform.sources.handler import BaseTransformerHandler
 from main.celery import app
 
@@ -14,7 +14,6 @@
 class IbtracsTransformHandler(BaseTransformerHandler):
     transformer = IBTrACSTransformer
     transformer_schema = IBTrACSDataSource
-    geocoder = GAULGeocoder(gpkg_path=None, service_base_url=settings.GEOCODER_URL)
 
     @classmethod
     def get_schema_data(cls, extraction_obj: ExtractionData):
@@ -23,37 +22,8 @@ def get_schema_data(cls, extraction_obj: ExtractionData):
 
         return cls.transformer_schema(source_url=extraction_obj.url, data=data.decode("utf-8"))
 
-    @classmethod
-    def handle_transformation(cls, extraction_id):
-        logger.info("Transformation started")
-        extraction_obj = ExtractionData.objects.filter(id=extraction_id).first()
-
-        transform_obj = Transform.objects.create(
-            extraction=extraction_obj,
-            status=Transform.Status.PENDING,
-        )
-
-        try:
-            schema = cls.get_schema_data(extraction_obj)
-            transformer = cls.transformer(schema, geocoder=cls.geocoder)
-            transformed_items = transformer.make_items()
-            print("...........", transformed_items[0].__dict__)
-
-            transform_obj.status = Transform.Status.SUCCESS
-            transform_obj.save(update_fields=["status"])
-
-            cls.load_stac_item_to_queue(transformed_items, transform_obj.id)
-
-            logger.info("Transformation ended")
-
-        except Exception as e:
-            logger.error("Transformation failed", exc_info=True, extra={"extraction_id": extraction_obj.id})
-            transform_obj.status = Transform.Status.FAILED
-            transform_obj.save(update_fields=["status"])
-            # FIXME: Check if this creates duplicate entry in Sentry. if yes, remove this.
-            raise e
-
     @staticmethod
     @app.task
     def task(extraction_id):
-        return IbtracsTransformHandler().handle_transformation(extraction_id)
+        geocoder = GAULGeocoder(gpkg_path=None, service_base_url=settings.GEOCODER_URL)
+        return IbtracsTransformHandler().handle_transformation(extraction_id, geocoder)