UPDATE
This commit is contained in:
@@ -4,11 +4,23 @@ import django.db.models.deletion
|
||||
from django.db import migrations, models
|
||||
|
||||
|
||||
def _create_model_if_missing(app_label, model_name):
|
||||
def _operation(apps, schema_editor):
|
||||
model = apps.get_model(app_label, model_name)
|
||||
existing_tables = set(schema_editor.connection.introspection.table_names())
|
||||
if model._meta.db_table in existing_tables:
|
||||
return
|
||||
schema_editor.create_model(model)
|
||||
|
||||
return _operation
|
||||
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
initial = True
|
||||
atomic = False
|
||||
|
||||
dependencies = [
|
||||
("farm_hub", "0009_farmhub_irrigation_method_fields"),
|
||||
("farm_hub", "0001_initial"),
|
||||
]
|
||||
|
||||
operations = [
|
||||
@@ -33,24 +45,48 @@ class Migration(migrations.Migration):
|
||||
],
|
||||
options={"db_table": "sensor_catalogs", "ordering": ["code"]},
|
||||
),
|
||||
],
|
||||
),
|
||||
migrations.RunPython(
|
||||
_create_model_if_missing("device_hub", "SensorCatalog"),
|
||||
migrations.RunPython.noop,
|
||||
),
|
||||
migrations.SeparateDatabaseAndState(
|
||||
database_operations=[],
|
||||
state_operations=[
|
||||
migrations.CreateModel(
|
||||
name="FarmDevice",
|
||||
name="FarmSensor",
|
||||
fields=[
|
||||
("id", models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name="ID")),
|
||||
("uuid", models.UUIDField(db_index=True, default=uuid.uuid4, editable=False, unique=True)),
|
||||
("physical_device_uuid", models.UUIDField(db_index=True, default=uuid.uuid4, unique=True)),
|
||||
("name", models.CharField(max_length=255)),
|
||||
("sensor_type", models.CharField(blank=True, default="", max_length=255)),
|
||||
("is_active", models.BooleanField(default=True)),
|
||||
("specifications", models.JSONField(blank=True, default=dict)),
|
||||
("power_source", models.JSONField(blank=True, default=dict)),
|
||||
("customization", models.JSONField(blank=True, default=dict)),
|
||||
("created_at", models.DateTimeField(auto_now_add=True)),
|
||||
("updated_at", models.DateTimeField(auto_now=True)),
|
||||
("farm", models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name="sensors", to="farm_hub.farmhub")),
|
||||
("sensor_catalog", models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.PROTECT, related_name="farm_devices", to="device_hub.sensorcatalog")),
|
||||
(
|
||||
"farm",
|
||||
models.ForeignKey(
|
||||
on_delete=django.db.models.deletion.CASCADE,
|
||||
related_name="sensors",
|
||||
to="farm_hub.farmhub",
|
||||
),
|
||||
),
|
||||
],
|
||||
options={"db_table": "farm_sensors", "ordering": ["-created_at"]},
|
||||
),
|
||||
],
|
||||
),
|
||||
migrations.RunPython(
|
||||
_create_model_if_missing("device_hub", "FarmSensor"),
|
||||
migrations.RunPython.noop,
|
||||
),
|
||||
migrations.SeparateDatabaseAndState(
|
||||
database_operations=[],
|
||||
state_operations=[
|
||||
migrations.CreateModel(
|
||||
name="SensorExternalRequestLog",
|
||||
fields=[
|
||||
@@ -65,4 +101,8 @@ class Migration(migrations.Migration):
|
||||
),
|
||||
],
|
||||
),
|
||||
migrations.RunPython(
|
||||
_create_model_if_missing("device_hub", "SensorExternalRequestLog"),
|
||||
migrations.RunPython.noop,
|
||||
),
|
||||
]
|
||||
|
||||
@@ -1,10 +1,35 @@
|
||||
from django.db import migrations
|
||||
import uuid
|
||||
|
||||
import django.db.models.deletion
|
||||
from django.db import migrations, models
|
||||
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
dependencies = [
|
||||
("device_hub", "0003_absorb_sensor_external_api"),
|
||||
("farm_hub", "0003_farmsensor_catalog_and_physical_device"),
|
||||
]
|
||||
|
||||
operations = []
|
||||
|
||||
operations = [
|
||||
migrations.SeparateDatabaseAndState(
|
||||
database_operations=[],
|
||||
state_operations=[
|
||||
migrations.AddField(
|
||||
model_name="farmsensor",
|
||||
name="physical_device_uuid",
|
||||
field=models.UUIDField(db_index=True, default=uuid.uuid4, unique=True),
|
||||
),
|
||||
migrations.AddField(
|
||||
model_name="farmsensor",
|
||||
name="sensor_catalog",
|
||||
field=models.ForeignKey(
|
||||
blank=True,
|
||||
null=True,
|
||||
on_delete=django.db.models.deletion.PROTECT,
|
||||
related_name="farm_sensors",
|
||||
to="device_hub.sensorcatalog",
|
||||
),
|
||||
),
|
||||
],
|
||||
),
|
||||
]
|
||||
|
||||
+29
-18
@@ -9,23 +9,34 @@ class Migration(migrations.Migration):
|
||||
]
|
||||
|
||||
operations = [
|
||||
migrations.RenameModel(
|
||||
old_name="SensorCatalog",
|
||||
new_name="DeviceCatalog",
|
||||
),
|
||||
migrations.AddField(
|
||||
model_name="devicecatalog",
|
||||
name="device_communication_type",
|
||||
field=models.CharField(
|
||||
choices=[("output_only", "Output Only"), ("input_only", "Input Only")],
|
||||
db_index=True,
|
||||
default="output_only",
|
||||
max_length=32,
|
||||
),
|
||||
),
|
||||
migrations.AlterField(
|
||||
model_name="farmdevice",
|
||||
name="sensor_catalog",
|
||||
field=models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.PROTECT, related_name="farm_devices", to="device_hub.devicecatalog"),
|
||||
migrations.SeparateDatabaseAndState(
|
||||
database_operations=[],
|
||||
state_operations=[
|
||||
migrations.RenameModel(
|
||||
old_name="SensorCatalog",
|
||||
new_name="DeviceCatalog",
|
||||
),
|
||||
migrations.AddField(
|
||||
model_name="devicecatalog",
|
||||
name="device_communication_type",
|
||||
field=models.CharField(
|
||||
choices=[("output_only", "Output Only"), ("input_only", "Input Only")],
|
||||
db_index=True,
|
||||
default="output_only",
|
||||
max_length=32,
|
||||
),
|
||||
),
|
||||
migrations.AlterField(
|
||||
model_name="farmdevice",
|
||||
name="sensor_catalog",
|
||||
field=models.ForeignKey(
|
||||
blank=True,
|
||||
null=True,
|
||||
on_delete=django.db.models.deletion.PROTECT,
|
||||
related_name="farm_devices",
|
||||
to="device_hub.devicecatalog",
|
||||
),
|
||||
),
|
||||
],
|
||||
),
|
||||
]
|
||||
|
||||
@@ -1,23 +1,51 @@
|
||||
from django.db import migrations, models
|
||||
|
||||
|
||||
def ensure_device_catalogs_m2m_table(apps, schema_editor):
|
||||
FarmDevice = apps.get_model("device_hub", "FarmDevice")
|
||||
through_model = FarmDevice._meta.get_field("device_catalogs").remote_field.through
|
||||
existing_tables = set(schema_editor.connection.introspection.table_names())
|
||||
if through_model._meta.db_table not in existing_tables:
|
||||
schema_editor.create_model(through_model)
|
||||
|
||||
|
||||
def copy_sensor_catalog_to_device_catalogs(apps, schema_editor):
|
||||
FarmDevice = apps.get_model("device_hub", "FarmDevice")
|
||||
for farm_device in FarmDevice.objects.exclude(sensor_catalog__isnull=True).iterator():
|
||||
farm_device.device_catalogs.add(farm_device.sensor_catalog_id)
|
||||
through_model = FarmDevice._meta.get_field("device_catalogs").remote_field.through
|
||||
through_table = through_model._meta.db_table
|
||||
farm_device_column = through_model._meta.get_field("farmdevice").column
|
||||
device_catalog_column = through_model._meta.get_field("devicecatalog").column
|
||||
|
||||
with schema_editor.connection.cursor() as cursor:
|
||||
for farm_device_id, sensor_catalog_id in FarmDevice.objects.exclude(sensor_catalog__isnull=True).values_list("pk", "sensor_catalog_id").iterator():
|
||||
cursor.execute(
|
||||
f"""
|
||||
INSERT IGNORE INTO {schema_editor.quote_name(through_table)}
|
||||
({schema_editor.quote_name(farm_device_column)}, {schema_editor.quote_name(device_catalog_column)})
|
||||
VALUES (%s, %s)
|
||||
""",
|
||||
[farm_device_id, sensor_catalog_id],
|
||||
)
|
||||
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
atomic = False
|
||||
|
||||
dependencies = [
|
||||
("device_hub", "0007_devicecatalog_dynamic_fields"),
|
||||
]
|
||||
|
||||
operations = [
|
||||
migrations.AddField(
|
||||
model_name="farmdevice",
|
||||
name="device_catalogs",
|
||||
field=models.ManyToManyField(blank=True, related_name="composite_farm_devices", to="device_hub.devicecatalog"),
|
||||
migrations.SeparateDatabaseAndState(
|
||||
database_operations=[],
|
||||
state_operations=[
|
||||
migrations.AddField(
|
||||
model_name="farmdevice",
|
||||
name="device_catalogs",
|
||||
field=models.ManyToManyField(blank=True, related_name="composite_farm_devices", to="device_hub.devicecatalog"),
|
||||
),
|
||||
],
|
||||
),
|
||||
migrations.RunPython(ensure_device_catalogs_m2m_table, migrations.RunPython.noop),
|
||||
migrations.RunPython(copy_sensor_catalog_to_device_catalogs, migrations.RunPython.noop),
|
||||
]
|
||||
|
||||
@@ -0,0 +1,47 @@
|
||||
from django.db import migrations, models
|
||||
|
||||
|
||||
def add_column_if_missing(schema_editor, table_name, column_name, field):
|
||||
existing_columns = {
|
||||
column.name
|
||||
for column in schema_editor.connection.introspection.get_table_description(
|
||||
schema_editor.connection.cursor(),
|
||||
table_name,
|
||||
)
|
||||
}
|
||||
if column_name in existing_columns:
|
||||
return
|
||||
field.set_attributes_from_name(column_name)
|
||||
schema_editor.add_field(
|
||||
field.model,
|
||||
field,
|
||||
)
|
||||
|
||||
|
||||
def sync_devicecatalog_schema(apps, schema_editor):
|
||||
DeviceCatalog = apps.get_model("device_hub", "DeviceCatalog")
|
||||
table_name = DeviceCatalog._meta.db_table
|
||||
|
||||
fields = [
|
||||
DeviceCatalog._meta.get_field("device_communication_type"),
|
||||
DeviceCatalog._meta.get_field("payload_mapping"),
|
||||
DeviceCatalog._meta.get_field("display_schema"),
|
||||
DeviceCatalog._meta.get_field("supported_widgets"),
|
||||
DeviceCatalog._meta.get_field("commands_schema"),
|
||||
DeviceCatalog._meta.get_field("capabilities"),
|
||||
]
|
||||
|
||||
for field in fields:
|
||||
add_column_if_missing(schema_editor, table_name, field.column, field)
|
||||
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
atomic = False
|
||||
|
||||
dependencies = [
|
||||
("device_hub", "0008_farmdevice_device_catalogs"),
|
||||
]
|
||||
|
||||
operations = [
|
||||
migrations.RunPython(sync_devicecatalog_schema, migrations.RunPython.noop),
|
||||
]
|
||||
Reference in New Issue
Block a user