Skip to content

Commit 6fb4fe3

Browse files
Rup-Narayan-Rajbanshiranjan-stha
authored andcommitted
send file as input for glide transformer.
- Change data schema format for glide transformer.
1 parent 82eaecc commit 6fb4fe3

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

apps/etl/transform/sources/glide.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,11 @@
11
import json
22

3+
from pystac_monty.sources.common import DataType, File
34
from pystac_monty.sources.glide import GlideDataSource, GlideTransformer
45

56
from apps.etl.models import ExtractionData
67
from apps.etl.transform.sources.handler import BaseTransformerHandler
8+
from apps.etl.utils import write_into_temp_file
79
from main.celery import app
810

911

@@ -13,10 +15,12 @@ class GlideTransformHandler(BaseTransformerHandler[GlideTransformer, GlideDataSo
1315

1416
@classmethod
1517
def get_schema_data(cls, extraction_obj):
16-
with extraction_obj.resp_data.open() as file_data:
17-
data = file_data.read()
18+
with extraction_obj.resp_data.open("rb") as f:
19+
data = f.read()
20+
data_file = write_into_temp_file(data)
21+
data_source = {"source_url": extraction_obj.url, "source_data": File(path=data_file.name, data_type=DataType.FILE)}
1822

19-
return cls.transformer_schema(source_url=extraction_obj.url, data=data)
23+
return cls.transformer_schema(data_source)
2024

2125
@staticmethod
2226
@app.task

0 commit comments

Comments
 (0)