denis
107 lines
4.0 KiB
Python
107 lines
4.0 KiB
Python
#
|
|
# 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}')
|