Handle Build as a dict (APIBuild)

Author: humitos

readthedocs/organizations/signals.py

@@ -90,6 +90,5 @@ def mark_organization_assets_not_cleaned(sender, build, **kwargs):
     trigger a Celery task that will be executed in the web and mark the
     organization assets as not cleaned.
     """
-    organization = build.project.organizations.first()
-    if build.state == BUILD_STATE_FINISHED and build.success and organization:
-        mark_organization_assets_not_cleaned_task.delay(build)
+    if build.state == BUILD_STATE_FINISHED:
+        mark_organization_assets_not_cleaned_task.delay(build.id)

readthedocs/organizations/tasks.py

@@ -7,7 +7,15 @@
 
 
 @app.task(queue='web')
-def mark_organization_assets_not_cleaned(organization):
-    log.info("Marking organization as not cleaned. organization=%s", organization.slug)
-    organization.artifacts_cleaned = False
-    organization.save()
+def mark_organization_assets_not_cleaned(build_pk):
+    try:
+        build = Build.objects.get(pk=build_pk)
+    except Build.DoesNotExist:
+        log.info("Build does not exist. build=%s", build_pk)
+        return
+
+    organization = build.project.organizations.first()
+    if organization:
+        log.info("Marking organization as not cleaned. organization=%s", organization.slug)
+        organization.artifacts_cleaned = False
+        organization.save()