fix(enum): remove some dangling .name calls after enum migration

zyv · Nov 3, 2023 · c445099 · c445099
1 parent cedc3a7
commit c445099
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/logbook/management/commands/import_flightlog.py b/logbook/management/commands/import_flightlog.py
@@ -110,7 +110,7 @@ def handle(self, *args, **options):
                     "departure_time": departure_time,
                     "arrival_time": arrival_time,
                     "landings": int(row[Fields.LANDINGS]),
-                    "time_function": time_function.name,
+                    "time_function": time_function,
                     "pilot": pilot,
                     "copilot": copilot,
                     "launch_type": launch_type,

diff --git a/logbook/templates/logbook/logentry_list.html b/logbook/templates/logbook/logentry_list.html
@@ -17,12 +17,12 @@ <h1>Log entries</h1>
                 <th>From<br>To</th>
                 <th>Departure<br>Arrival</th>
                 {% for aircraft_type in AircraftType reversed %}
-                    <th class="text-center">{{ aircraft_type.name }}</th>
+                    <th class="text-center">{{ aircraft_type }}</th>
                 {% endfor %}
                 <th class="text-center">Landings</th>
                 <th>Name</th>
                 {% for function_type in FunctionType %}
-                    <th class="text-center">{{ function_type.name }}</th>
+                    <th class="text-center">{{ function_type }}</th>
                 {% endfor %}
                 <th>Remarks</th>
             </tr>
@@ -36,7 +36,7 @@ <h1>Log entries</h1>
                     <td>{{ obj.departure_time | time:"H:i" }}<br>{{ obj.arrival_time | time:"H:i" }}</td>
                     {% for aircraft_type in AircraftType reversed %}
                         <td class="text-center">
-                            {% if obj.aircraft.type == aircraft_type.name %}
+                            {% if obj.aircraft.type == aircraft_type %}
                                 {{ obj.arrival_time | subtract:obj.departure_time | duration:"%h:%M" }}
                             {% elif obj is not None %}
                                 -
@@ -47,7 +47,7 @@ <h1>Log entries</h1>
                     <td>{% if not obj.pilot.self %}{{ obj.pilot.first_name.0 }}. {{ obj.pilot.last_name }}{% else %}Self{% endif %}</td>
                     {% for function_type in FunctionType %}
                         <td class="text-center">
-                            {% if obj.time_function == function_type.name %}
+                            {% if obj.time_function == function_type %}
                                 {{ obj.arrival_time | subtract:obj.departure_time | duration:"%h:%M" }}
                             {% elif obj is not None %}
                                 -

diff --git a/logbook/views/entries.py b/logbook/views/entries.py
@@ -69,7 +69,7 @@ def parse_pilot(name: str) -> Pilot:
             departure_time=departure_time,
             arrival_time=arrival_time,
             landings=int(flight_data["landingcount"]),
-            time_function=time_function.name,
+            time_function=time_function,
             pilot=pilot,
             copilot=copilot,
             remarks=html.unescape(flight_data["comment"]),