Skip to content

Commit 9125545

Browse files
committed
Fix lint errors
1 parent a608fc1 commit 9125545

File tree

2 files changed

+13
-4
lines changed

2 files changed

+13
-4
lines changed

django_tasks/backends/database/admin.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,28 @@
11
from typing import List, Optional
22

33
from django.contrib import admin
4+
from django.db.models import QuerySet
45
from django.http import HttpRequest
56

67
from django_tasks.task import ResultStatus
78

89
from .models import DBTaskResult
910

1011

11-
def reenqueue(modeladmin: admin.ModelAdmin, request, queryset):
12+
def reenqueue(
13+
modeladmin: admin.ModelAdmin,
14+
request: HttpRequest,
15+
queryset: QuerySet[DBTaskResult],
16+
) -> None:
1217
tasks = queryset.update(status=ResultStatus.NEW)
1318
modeladmin.message_user(request, f"Rescheduled {tasks} tasks.", "SUCCESS")
1419

1520

16-
def duplicate(modeladmin: admin.ModelAdmin, request, queryset):
21+
def duplicate(
22+
modeladmin: admin.ModelAdmin,
23+
request: HttpRequest,
24+
queryset: QuerySet[DBTaskResult],
25+
) -> None:
1726
tasks = DBTaskResult.objects.bulk_create(
1827
old_task.duplicate() for old_task in queryset
1928
)

django_tasks/backends/database/models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import logging
22
import uuid
3-
from typing import TYPE_CHECKING, Any, Generic, Optional, TypeVar
3+
from typing import TYPE_CHECKING, Any, Generic, Optional, Self, TypeVar
44

55
from django.core.exceptions import SuspiciousOperation
66
from django.db import models
@@ -170,7 +170,7 @@ def set_failed(self, exc: BaseException) -> None:
170170
self.result = None
171171
self.save(update_fields=["status", "finished_at", "result"])
172172

173-
def duplicate(self):
173+
def duplicate(self) -> Self:
174174
return type(self)(
175175
args_kwargs=self.args_kwargs,
176176
priority=self.priority,

0 commit comments

Comments
 (0)