Support for both single and high/low tariff #130

Author: dennissiemensma

dsmr_stats/management/commands/dsmr_fake_stats.py

@@ -6,16 +6,13 @@
 from django.conf import settings
 from django.db import models
 
-from dsmr_backend.mixins import InfiniteManagementCommandMixin
 from dsmr_stats.models.statistics import DayStatistics, HourStatistics
 from dsmr_consumption.models.consumption import ElectricityConsumption
 from dsmr_datalogger.models.reading import DsmrReading
 
 
-class Command(InfiniteManagementCommandMixin, BaseCommand):
+class Command(BaseCommand):
     help = _('Alters any stats generate to fake data. DO NOT USE in production! Used for integration checks.')
-    name = __name__  # Required for PID file.
-    sleep_time = 10
 
     def add_arguments(self, parser):
         super(Command, self).add_arguments(parser)
@@ -27,7 +24,7 @@ def add_arguments(self, parser):
             help=_('Required option to acknowledge you that you WILL mess up your data with this.')
         )
 
-    def run(self, **options):
+    def handle(self, **options):
         """ InfiniteManagementCommandMixin listens to handle() and calls run() in a loop. """
         if not settings.DEBUG:
             raise CommandError(_('Intended usage is NOT production! Only allowed when DEBUG = True'))