Extend functionalities

dcc/roster/admin.py

@@ -1,9 +1,9 @@
 from django.contrib import admin
-from roster.models import Cab
+from roster.models import RollingStock
 
 
-@admin.register(Cab)
-class CabAdmin(admin.ModelAdmin):
+@admin.register(RollingStock)
+class RollingStockAdmin(admin.ModelAdmin):
     readonly_fields = ('image_thumbnail', 'creation_time', 'updated_time',)
     list_display = ('identifier', 'address', 'manufacturer', 'company')
     list_filter = list_display
@@ -12,6 +12,7 @@ class CabAdmin(admin.ModelAdmin):
     fieldsets = (
         (None, {
             'fields': ('identifier',
+                       'tags',
                        'address',
                        'manufacturer',
                        'decoder',

dcc/roster/migrations/0017_rename_cab_rollingstock.py

@@ -0,0 +1,18 @@
+# Generated by Django 4.0.2 on 2022-03-06 19:42
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('metadata', '0003_metadata_extend'),
+        ('roster', '0016_alter_cab_image'),
+    ]
+
+    operations = [
+        migrations.RenameModel(
+            old_name='Cab',
+            new_name='RollingStock',
+        ),
+    ]

dcc/roster/migrations/0018_alter_rollingstock_options.py

@@ -0,0 +1,17 @@
+# Generated by Django 4.0.2 on 2022-03-06 19:50
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('roster', '0017_rename_cab_rollingstock'),
+    ]
+
+    operations = [
+        migrations.AlterModelOptions(
+            name='rollingstock',
+            options={'ordering': ['address', 'identifier'], 'verbose_name_plural': 'Rolling stock'},
+        ),
+    ]

dcc/roster/migrations/0019_rollingstock_tags.py

@@ -0,0 +1,19 @@
+# Generated by Django 4.0.2 on 2022-03-06 20:15
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('metadata', '0006_alter_tag_options'),
+        ('roster', '0018_alter_rollingstock_options'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='rollingstock',
+            name='tags',
+            field=models.ManyToManyField(related_name='rolling_stock', to='metadata.Tag'),
+        ),
+    ]

dcc/roster/migrations/0020_alter_rollingstock_address_alter_rollingstock_tags.py

@@ -0,0 +1,24 @@
+# Generated by Django 4.0.2 on 2022-03-06 20:16
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('metadata', '0006_alter_tag_options'),
+        ('roster', '0019_rollingstock_tags'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='rollingstock',
+            name='address',
+            field=models.SmallIntegerField(default=3, null=True),
+        ),
+        migrations.AlterField(
+            model_name='rollingstock',
+            name='tags',
+            field=models.ManyToManyField(null=True, related_name='rolling_stock', to='metadata.Tag'),
+        ),
+    ]

dcc/roster/migrations/0021_alter_rollingstock_tags.py

@@ -0,0 +1,19 @@
+# Generated by Django 4.0.2 on 2022-03-06 20:17
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('metadata', '0006_alter_tag_options'),
+        ('roster', '0020_alter_rollingstock_address_alter_rollingstock_tags'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='rollingstock',
+            name='tags',
+            field=models.ManyToManyField(blank=True, related_name='rolling_stock', to='metadata.Tag'),
+        ),
+    ]

dcc/roster/migrations/0022_alter_rollingstock_address.py

@@ -0,0 +1,18 @@
+# Generated by Django 4.0.2 on 2022-03-06 20:18
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('roster', '0021_alter_rollingstock_tags'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='rollingstock',
+            name='address',
+            field=models.SmallIntegerField(blank=True, default=3, null=True),
+        ),
+    ]

dcc/roster/models.py

@@ -4,7 +4,7 @@ from django.db import models
 # from django.dispatch import receiver
 
 from dcc.utils import get_image_preview
-from metadata.models import Manufacturer, Decoder, Company
+from metadata.models import Manufacturer, Decoder, Company, Tag
 
 # class OverwriteMixin(FileSystemStorage):
 #     def get_available_name(self, name, max_length):
@@ -12,12 +12,16 @@ from metadata.models import Manufacturer, Decoder, Company
 #         return name
 
 
-class Cab(models.Model):
+class RollingStock(models.Model):
     uuid = models.UUIDField(
         primary_key=True, default=uuid4,
         editable=False)
     identifier = models.CharField(max_length=128, unique=False)
-    address = models.SmallIntegerField(default=3)
+    tags = models.ManyToManyField(
+        Tag,
+        related_name='rolling_stock',
+        blank=True)
+    address = models.SmallIntegerField(default=3, null=True, blank=True)
     manufacturer = models.ForeignKey(
         Manufacturer, on_delete=models.CASCADE,
         null=True, blank=True)
@@ -42,6 +46,7 @@ class Cab(models.Model):
 
     class Meta:
         ordering = ['address', 'identifier']
+        verbose_name_plural = "Rolling stock"
 
     def __str__(self):
         return "{0} {1}".format(self.manufacturer, self.identifier)

dcc/roster/serializers.py

@@ -1,15 +1,15 @@
 from rest_framework import serializers
-from roster.models import Cab
+from roster.models import RollingStock
 from metadata.serializers import (
     ManufacturerSerializer, CompanySerializer, DecoderSerializer)
 
 
-class CabSerializer(serializers.ModelSerializer):
+class RollingStockSerializer(serializers.ModelSerializer):
     manufacturer = ManufacturerSerializer()
     decoder = DecoderSerializer()
     company = CompanySerializer()
 
     class Meta:
-        model = Cab
+        model = RollingStock
         fields = "__all__"
         read_only_fields = ("creation_time", "updated_time")

dcc/roster/views.py

@@ -1,27 +1,27 @@
 from rest_framework.generics import ListAPIView, RetrieveAPIView
 
-from roster.models import Cab
-from roster.serializers import CabSerializer
+from roster.models import RollingStock
+from roster.serializers import RollingStockSerializer
 
 
 class RosterList(ListAPIView):
-    queryset = Cab.objects.all()
-    serializer_class = CabSerializer
+    queryset = RollingStock.objects.all()
+    serializer_class = RollingStockSerializer
 
 
 class RosterGet(RetrieveAPIView):
-    queryset = Cab.objects.all()
-    serializer_class = CabSerializer
+    queryset = RollingStock.objects.all()
+    serializer_class = RollingStockSerializer
     lookup_field = 'uuid'
 
 
 class RosterAddress(RetrieveAPIView):
-    queryset = Cab.objects.all()
-    serializer_class = CabSerializer
+    queryset = RollingStock.objects.all()
+    serializer_class = RollingStockSerializer
     lookup_field = 'address'
 
 
 class RosterIdentifier(RetrieveAPIView):
-    queryset = Cab.objects.all()
-    serializer_class = CabSerializer
+    queryset = RollingStock.objects.all()
+    serializer_class = RollingStockSerializer
     lookup_field = 'identifier'