Fix release blockers and deployment build

apps/sources/services/scheduling.py

@@ -76,14 +76,10 @@ def calculate_failure_backoff_seconds(
 
 
 def calculate_success_jitter_seconds(source: Source) -> int:
-    return calculate_jitter_seconds(
-        source, max_seconds=settings.SOURCE_SUCCESS_JITTER_SECONDS
-    )
+    return calculate_jitter_seconds(source, max_seconds=settings.SOURCE_SUCCESS_JITTER_SECONDS)
 
 
-def acquire_source_lease(
-    *, source_id: int, worker_token: str, now=None
-) -> SourceLease | None:
+def acquire_source_lease(*, source_id: int, worker_token: str, now=None) -> SourceLease | None:
     now = now or timezone.now()
     with transaction.atomic():
         source = Source.objects.select_for_update().get(pk=source_id)
@@ -101,9 +97,8 @@ def acquire_source_lease(
                 return None
 
         lease = SourceLease.objects.select_for_update().filter(source=source).first()
-        if lease is not None and not lease.is_expired:
-            if lease.token != worker_token:
-                return None
+        if lease is not None and not lease.is_expired and lease.token != worker_token:
+            return None
 
         active_leases = SourceLease.objects.select_for_update().filter(
             source__domain=domain, expires_at__gt=now
@@ -118,7 +113,10 @@ def acquire_source_lease(
         lease.token = worker_token
         lease.worker_id = worker_token
         lease.expires_at = now + timedelta(seconds=_lease_ttl_seconds())
-        lease.save(update_fields=["token", "worker_id", "expires_at", "updated_at"])
+        if lease.pk is None:
+            lease.save()
+        else:
+            lease.save(update_fields=["token", "worker_id", "expires_at", "updated_at"])
         return lease
 
 
@@ -129,9 +127,11 @@ def release_source_lease(*, source_id: int, worker_token: str, now=None) -> bool
         if not source.domain:
             return False
 
-        lease = SourceLease.objects.select_for_update().filter(
-            source=source, token=worker_token
-        ).first()
+        lease = (
+            SourceLease.objects.select_for_update()
+            .filter(source=source, token=worker_token)
+            .first()
+        )
         if not lease:
             return False