Improve performance and add type hints

chzzk/tasks.py

@@ -5,7 +5,7 @@ import logging
 from celery import shared_task
 from django.core.management import call_command
 
-logger = logging.getLogger("ttvdrops.tasks")
+logger: logging.Logger = logging.getLogger("ttvdrops.tasks")
 
 
 @shared_task(bind=True, queue="imports", max_retries=3, default_retry_delay=60)

chzzk/tests/test_management_commands.py

@@ -136,7 +136,7 @@ class ImportChzzkCampaignRangeCommandTest(TestCase):
         stdout = StringIO()
         stderr = StringIO()
 
-        def side_effect(command: str, *args: str, **kwargs: object) -> None:
+        def side_effect(command: str, *args: str, **kwargs: StringIO) -> None:
             if "4" in args:
                 msg = "Campaign 4 not found"
                 raise CommandError(msg)

chzzk/views.py

@@ -1,7 +1,6 @@
 from typing import TYPE_CHECKING
 
 from django.db.models import Q
-from django.db.models.query import QuerySet
 from django.shortcuts import get_object_or_404
 from django.shortcuts import render
 from django.urls import reverse
@@ -16,9 +15,9 @@ from twitch.feeds import TTVDropsBaseFeed
 if TYPE_CHECKING:
     import datetime
 
+    from django.db.models.query import QuerySet
     from django.http import HttpResponse
     from django.http.request import HttpRequest
-    from pytest_django.asserts import QuerySet
 
 
 def dashboard_view(request: HttpRequest) -> HttpResponse: