Skip to content

Commit 221dedf

Browse files
update to use upsert
1 parent 85147c5 commit 221dedf

File tree

1 file changed

+2
-8
lines changed

1 file changed

+2
-8
lines changed

src/sentry/preprod/tasks.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -195,8 +195,8 @@ def assemble_preprod_artifact_size_analysis(
195195
id=artifact_id,
196196
)
197197

198-
# Get or create PreprodArtifactSizeMetrics record
199-
size_metrics, created = PreprodArtifactSizeMetrics.objects.get_or_create(
198+
# Upsert PreprodArtifactSizeMetrics record
199+
size_metrics, created = PreprodArtifactSizeMetrics.objects.update_or_create(
200200
preprod_artifact=preprod_artifact,
201201
defaults={
202202
"analysis_file_id": assemble_result.bundle.id,
@@ -205,12 +205,6 @@ def assemble_preprod_artifact_size_analysis(
205205
},
206206
)
207207

208-
if not created:
209-
# Update existing record
210-
size_metrics.analysis_file_id = assemble_result.bundle.id
211-
size_metrics.state = PreprodArtifactSizeMetrics.SizeAnalysisState.COMPLETED
212-
size_metrics.save(update_fields=["analysis_file_id", "state", "date_updated"])
213-
214208
logger.info(
215209
"Created or updated preprod artifact size metrics with analysis file",
216210
extra={

0 commit comments

Comments
 (0)