Merge pull request #113 from golony6449/feature/golony/session

django import export

Author: golony6449

session/admin.py

@@ -1,6 +1,9 @@
 from django.contrib import admin
 
+from import_export.admin import ImportExportModelAdmin
+
 from .models import Proposal, Session, Category
+from .resources import SessionResource
 
 
 @admin.register(Proposal)
@@ -20,7 +23,7 @@ class ProposalAdmin(admin.ModelAdmin):
 
 
 @admin.register(Session)
-class SessionAdmin(admin.ModelAdmin):
+class SessionAdmin(ImportExportModelAdmin):
     list_display = [
         "id",
         "title",
@@ -32,6 +35,7 @@ class SessionAdmin(admin.ModelAdmin):
     ]
     list_filter = ["difficulty", "duration", "language", "category"]
     search_fields = ["title", "user__username"]
+    resource_class = SessionResource
 
 
 @admin.register(Category)

session/resources.py

@@ -0,0 +1,15 @@
+import datetime
+
+from import_export import resources, fields
+from import_export.widgets import ForeignKeyWidget
+
+from session.models import Session, Category
+
+
+class SessionResource(resources.ModelResource):
+    def before_save_instance(self, instance: Session, using_transactions, dry_run):
+        instance.start_at = datetime.datetime.now()
+
+    class Meta:
+        model = Session
+        fields = ["id", "title", "difficulty", "duration", "language", "category"]

session/serializers.py

@@ -47,7 +47,7 @@ class Meta:
         fields = [
             "id",
             "title",
-            "brief",
+            "introduction",
             "difficulty",
             "duration",
             "language",
@@ -65,7 +65,12 @@ def get_category_name(obj: Session):
 
     def to_representation(self, instance: Session):
         response = super().to_representation(instance)
-        response["user"] = UserExtSerializer(instance.user.userext).data
+
+        if instance.user is not None:
+            response["user"] = UserExtSerializer(instance.user.userext).data
+        else:
+            response["user"] = None
+
         return response