!111: tests: aircox.management (#114)

!111

Co-authored-by: bkfox <thomas bkfox net>
Reviewed-on: rc/aircox#114

aircox/management/commands/diffusions.py

@@ -9,59 +9,13 @@ import logging
 from argparse import RawTextHelpFormatter
 
 from django.core.management.base import BaseCommand
-from django.db import transaction
 from django.utils import timezone as tz
 
-from aircox.models import Diffusion, Schedule
+from aircox.controllers.diffusions import Diffusions
 
 logger = logging.getLogger("aircox.commands")
 
 
-class Actions:
-    date = None
-
-    def __init__(self, date):
-        self.date = date or datetime.date.today()
-
-    def update(self):
-        episodes, diffusions = [], []
-        for schedule in Schedule.objects.filter(
-            program__active=True, initial__isnull=True
-        ):
-            eps, diffs = schedule.diffusions_of_month(self.date)
-            if eps:
-                episodes += eps
-            if diffs:
-                diffusions += diffs
-
-            logger.info(
-                "[update] %s: %d episodes, %d diffusions and reruns",
-                str(schedule),
-                len(eps),
-                len(diffs),
-            )
-
-        with transaction.atomic():
-            logger.info(
-                "[update] save %d episodes and %d diffusions",
-                len(episodes),
-                len(diffusions),
-            )
-            for episode in episodes:
-                episode.save()
-            for diffusion in diffusions:
-                # force episode id's update
-                diffusion.episode = diffusion.episode
-                diffusion.save()
-
-    def clean(self):
-        qs = Diffusion.objects.filter(
-            type=Diffusion.TYPE_UNCONFIRMED, start__lt=self.date
-        )
-        logger.info("[clean] %d diffusions will be removed", qs.count())
-        qs.delete()
-
-
 class Command(BaseCommand):
     help = __doc__
 
@@ -116,7 +70,7 @@ class Command(BaseCommand):
                 date += tz.timedelta(days=28)
             date = date.replace(day=1)
 
-        actions = Actions(date)
+        actions = Diffusions(date)
         if options.get("update"):
             actions.update()
         if options.get("clean"):