second commit

2026-03-10 13:10:32 +01:00
parent b5cc0106c6
commit 3aac7ff9b9
101 changed files with 22046 additions and 1 deletions
@@ -0,0 +1,106 @@
+#
+# encoding: utf-8
+#import re
+#from django.utils.translation import gettext_lazy as _
+from django.conf import settings
+from django.core.management import BaseCommand
+from django_celery_beat.models import IntervalSchedule, CrontabSchedule, SolarSchedule
+from django.contrib.auth import get_user_model
+from relay import models
+from modules import utils
+
+
+def create_user():
+    try:
+        User = get_user_model()
+        if User.objects.count() == 0:
+            for email, username, password, is_superuser in settings.ADMINS:
+                if is_superuser:
+                    User.objects.create_superuser(
+                        email=email,
+                        username=username,
+                        password=password,
+                        is_active=True,
+                        is_superuser=is_superuser,
+                    )
+                else:
+                    User.objects.create_user(
+                        email=email,
+                        username=username,
+                        password=password,
+                        is_active=True,
+                    )
+                print(f'Creating {username} user with {password} password and email: {email} superuser: {is_superuser}')
+    except Exception as e:
+        print(f'Creating user error {e}')
+
+def create_schedule():
+    try:
+        for n in [1, 2, 5, 10, 15, 20, 30, 40, 45, 50]:
+            IntervalSchedule.objects.create(every=n, period='minutes')
+        print('Creating IntervalSchedule minutes')
+
+        for n in [1, 2, 4, 6, 8, 10, 12, 14, 16, 18, 20, 22]:
+            IntervalSchedule.objects.create(every=n, period='hours')
+        print('Creating IntervalSchedule hours')
+
+        for n in [1, 2, 3, 7, 15, 30, 60, 90, 180, 360]:
+            IntervalSchedule.objects.create(every=n, period='days')
+        print('Creating IntervalSchedule days')
+
+        CrontabSchedule.objects.create(minute='0', hour='4', day_of_week='*', timezone=settings.TIME_ZONE)
+        CrontabSchedule.objects.create(minute='0', hour='4', day_of_week='*/3', timezone=settings.TIME_ZONE)
+        CrontabSchedule.objects.create(minute='15', hour='4', day_of_week='*/3', timezone=settings.TIME_ZONE)
+        CrontabSchedule.objects.create(minute='30', hour='4', day_of_week='*/3', timezone=settings.TIME_ZONE)
+        CrontabSchedule.objects.create(minute='0', hour='18', day_of_week='*', timezone=settings.TIME_ZONE)
+        CrontabSchedule.objects.create(minute='15', hour='18', day_of_week='*/3', timezone=settings.TIME_ZONE)
+        CrontabSchedule.objects.create(minute='30', hour='18', day_of_week='*/3', timezone=settings.TIME_ZONE)
+        print('Creating CrontabSchedule')
+
+        SolarSchedule.objects.create(event='sunset', latitude=43.8, longitude=2.3)
+        SolarSchedule.objects.create(event='sunrise', latitude=43.8, longitude=2.3)
+        print('Creating SolarSchedule')
+
+    except Exception as e:
+        print(f'Creating schedule error {e}')
+
+
+def create_device_type():
+    try:
+        data = dict(
+            name = 'Relais à sortie décalée',
+            code = 'relay_shiftout',
+        )
+        models.DeviceType.objects.create(**data)
+        print(f'Creating DeviceType')
+    except Exception as e:
+        print(f'Creating DeviceType error {e}')
+
+
+def create_relay():
+    try:
+        device_type = models.DeviceType.objects.filter(code__exact='relay_shiftout').first()
+        for i in range(settings.CTL_RELAY_NUMBER):
+            data = dict(
+                label = f'Relay:{i}',
+                uuid = utils.get_device_uuid(12),
+                device_type = device_type,
+                index = i,
+            )
+            models.Device.objects.create(**data)
+            print('Creating Device', data)
+
+    except Exception as e:
+        print(f'Creating Device error {e}')
+
+
+class Command(BaseCommand):
+
+    def handle(self, *args, **options):
+        try:
+            create_user()
+            create_schedule()
+            create_device_type()
+            create_relay()
+        except Exception as e:
+            print(f'Creating monitor error {e}')