Skip to content

Commit

Permalink
fix classic metadata update (#366)
Browse files Browse the repository at this point in the history
  • Loading branch information
rsjr authored Sep 18, 2024
1 parent db19499 commit c0ca5c0
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions cognite/extractorutils/uploader/files.py
Original file line number Diff line number Diff line change
Expand Up @@ -255,18 +255,20 @@ def _upload_empty(
) -> tuple[FileMetadataOrCogniteExtractorFile, str]:
if isinstance(file_meta, CogniteExtractorFileApply):
node_id = self._apply_cognite_file(file_meta)
file_meta, url = self._create_cdm(instance_id=node_id)
file_meta_response, url = self._create_cdm(instance_id=node_id)
else:
file_meta, url = self.cdf_client.files.create(file_metadata=file_meta, overwrite=self.overwrite_existing)
file_meta_response, url = self.cdf_client.files.create(
file_metadata=file_meta, overwrite=self.overwrite_existing
)
# trigger update after creation (upsert =P)
basic_attributes = set(["externalId", "name"])
attr = set(file_meta.dump().keys())
diff = attr - basic_attributes

if len(diff) >= 1 and "externalId" in attr:
file_meta = self.cdf_client.files.update(file_meta)
file_meta_response = self.cdf_client.files.update(file_meta)

return file_meta, url
return file_meta_response, url

def _upload_bytes(self, size: int, file: BinaryIO, file_meta: FileMetadataOrCogniteExtractorFile) -> None:
file_meta, url = self._upload_empty(file_meta)
Expand Down

0 comments on commit c0ca5c0

Please sign in to comment.