Add a CSV export functionality in admin and add price fields (#41)

* Implement an action do download data in csv
* Refactor CSV download
* Move price to main models and add csv to bookshelf
* Update template and API
* Small refactoring

ram/roster/admin.py

@@ -1,6 +1,13 @@
+import html
+
+from django.conf import settings
 from django.contrib import admin
+from django.utils.html import strip_tags
+
 from adminsortable2.admin import SortableAdminBase, SortableInlineAdminMixin
 
+from ram.utils import generate_csv
+from portal.utils import get_site_conf
 from roster.models import (
     RollingClass,
     RollingClassProperty,
@@ -152,8 +159,6 @@ class RollingStockAdmin(SortableAdminBase, admin.ModelAdmin):
                     "era",
                     "description",
                     "production_year",
-                    "purchase_date",
-                    "notes",
                     "tags",
                 )
             },
@@ -168,6 +173,24 @@ class RollingStockAdmin(SortableAdminBase, admin.ModelAdmin):
                 )
             },
         ),
+        (
+            "Purchase data",
+            {
+                "fields": (
+                    "purchase_date",
+                    "price",
+                )
+            },
+        ),
+        (
+            "Notes",
+            {
+                "classes": ("collapse",),
+                "fields": (
+                    "notes",
+                )
+            },
+        ),
         (
             "Audit",
             {
@@ -179,3 +202,65 @@ class RollingStockAdmin(SortableAdminBase, admin.ModelAdmin):
             },
         ),
     )
+
+    def get_form(self, request, obj=None, **kwargs):
+        form = super().get_form(request, obj, **kwargs)
+        form.base_fields["price"].label = "Price ({})".format(
+            get_site_conf().currency
+        )
+        return form
+
+    def download_csv(modeladmin, request, queryset):
+        header = [
+            "Company",
+            "Identifier",
+            "Road Number",
+            "Manufacturer",
+            "Scale",
+            "Item Number",
+            "Set",
+            "Era",
+            "Description",
+            "Production Year",
+            "Notes",
+            "Tags",
+            "Decoder Interface",
+            "Decoder",
+            "Address",
+            "Purchase Date",
+            "Price ({})".format(get_site_conf().currency),
+            "Properties",
+        ]
+        data = []
+        for obj in queryset:
+            properties = settings.CSV_SEPARATOR_ALT.join(
+                "{}:{}".format(property.property.name, property.value)
+                for property in obj.property.all()
+            )
+            data.append([
+                obj.rolling_class.company.name,
+                obj.rolling_class.identifier,
+                obj.road_number,
+                obj.manufacturer.name,
+                obj.scale.scale,
+                obj.item_number,
+                obj.set,
+                obj.era,
+                html.unescape(strip_tags(obj.description)),
+                obj.production_year,
+                html.unescape(strip_tags(obj.notes)),
+                settings.CSV_SEPARATOR_ALT.join(
+                    t.name for t in obj.tags.all()
+                ),
+                obj.decoder_interface,
+                obj.decoder,
+                obj.address,
+                obj.purchase_date,
+                obj.price,
+                properties,
+            ])
+
+        return generate_csv(header, data, "rolling_stock.csv")
+
+    download_csv.short_description = "Download selected items as CSV"
+    actions = [download_csv]