hoelee/tubesync
1
0
Fork 0
mirror of https://github.com/meeb/tubesync.git synced 2025-06-23 13:36:35 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Add explicit transactions for certain tasks

Browse Source
This commit is contained in:
tcely 2025-03-09 21:47:48 -04:00 committed by GitHub
parent ed3c7c6e9d
commit 58472f7785
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tubesync/sync/tasks.py
View File

@ -17,6 +17,7 @@ from django.conf import settings
from django.core.files.base import ContentFile from django.core.files.base import ContentFile
from django.core.files.uploadedfile import SimpleUploadedFile from django.core.files.uploadedfile import SimpleUploadedFile
from django.utils import timezone from django.utils import timezone
from django.db.tansaction import atomic
from django.db.utils import IntegrityError from django.db.utils import IntegrityError
from django.utils.translation import gettext_lazy as _ from django.utils.translation import gettext_lazy as _
from background_task import background from background_task import background
@ -179,6 +180,7 @@ def cleanup_removed_media(source, videos):
@background(schedule=300, remove_existing_tasks=True) @background(schedule=300, remove_existing_tasks=True)
@atomic(durable=True)
def index_source_task(source_id): def index_source_task(source_id):
''' '''
Indexes media available from a Source object. Indexes media available from a Source object.
@ -221,7 +223,8 @@ def index_source_task(source_id):
if published_dt is not None: if published_dt is not None:
media.published = published_dt media.published = published_dt
try: try:
media.save() with atomic():
media.save()
log.debug(f'Indexed media: {source} / {media}') log.debug(f'Indexed media: {source} / {media}')
# log the new media instances # log the new media instances
new_media_instance = ( new_media_instance = (
@ -611,9 +614,10 @@ def save_all_media_for_source(source_id):
# Trigger the post_save signal for each media item linked to this source as various # Trigger the post_save signal for each media item linked to this source as various
# flags may need to be recalculated # flags may need to be recalculated
for media in mqs: with atomic():
if media.uuid not in already_saved: for media in mqs:
media.save() if media.uuid not in already_saved:
media.save()
@background(schedule=60, remove_existing_tasks=True) @background(schedule=60, remove_existing_tasks=True)
@ -626,6 +630,7 @@ def rename_media(media_id):
@background(schedule=300, remove_existing_tasks=True) @background(schedule=300, remove_existing_tasks=True)
@atomic(durable=True)
def rename_all_media_for_source(source_id): def rename_all_media_for_source(source_id):
try: try:
source = Source.objects.get(pk=source_id) source = Source.objects.get(pk=source_id)
@ -653,7 +658,8 @@ def rename_all_media_for_source(source_id):
downloaded=True, downloaded=True,
) )
for media in mqs: for media in mqs:
media.rename_files() with atomic():
media.rename_files()
@background(schedule=60, remove_existing_tasks=True) @background(schedule=60, remove_existing_tasks=True)