Refactor rolling stock

dcc/roster/admin.py

@@ -1,27 +1,42 @@
 from django.contrib import admin
-from roster.models import RollingStock
+from roster.models import (
+    RollingStock, RollingStockImage, RollingStockDocument, Engine, Car,
+    Equipment, Other)
+
+
+class RollingStockDocInline(admin.TabularInline):
+    model = RollingStockDocument
+    min_num = 0
+    extra = 0
+
+
+class RollingStockImageInline(admin.TabularInline):
+    model = RollingStockImage
+    min_num = 0
+    extra = 0
+    readonly_fields = ('image_thumbnail',)
 
 
-@admin.register(RollingStock)
 class RollingStockAdmin(admin.ModelAdmin):
-    readonly_fields = ('image_thumbnail', 'creation_time', 'updated_time',)
-    list_display = ('identifier', 'address', 'manufacturer', 'company')
+    inlines = (RollingStockImageInline, RollingStockDocInline)
+    readonly_fields = ('creation_time', 'updated_time',)
+    list_display = ('identifier', 'manufacturer', 'sku', 'company')
     list_filter = list_display
     search_fields = list_display
 
     fieldsets = (
         (None, {
             'fields': ('identifier',
+                       'type',
                        'tags',
-                       'address',
                        'manufacturer',
+                       'sku',
                        'decoder',
+                       'address',
                        'company',
                        'epoch',
                        'production_year',
                        'purchase_date',
-                       'image',
-                       'image_thumbnail',
                        'notes')
         }),
         ('Audit', {
@@ -29,3 +44,23 @@ class RollingStockAdmin(admin.ModelAdmin):
             'fields': ('creation_time', 'updated_time',)
         }),
     )
+
+
+@admin.register(Engine)
+class Engine(RollingStockAdmin):
+    list_display = ('identifier', 'address', 'manufacturer', 'sku', 'company')
+
+
+@admin.register(Car)
+class Car(RollingStockAdmin):
+    pass
+
+
+@admin.register(Equipment)
+class Equipment(RollingStockAdmin):
+    pass
+
+
+@admin.register(Other)
+class Other(RollingStockAdmin):
+    pass