solution_py-tickets-and-orders_4

mate-academy · Dec 10, 2023 · b17bf25 · b17bf25
1 parent 49f5a32
commit b17bf25
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 12 deletions.
diff --git a/services/movie_session.py b/services/movie_session.py
@@ -1,5 +1,6 @@
 from django.db.models import QuerySet
 
+
 from db.models import MovieSession
 from db.models import Ticket
 
@@ -46,13 +47,7 @@ def delete_movie_session_by_id(session_id: int) -> None:
 
 
 def get_taken_seats(movie_session_id: int) -> list[dict]:
- tickets = Ticket.objects.filter(
+ tickets = list(Ticket.objects.filter(
  movie_session__id=movie_session_id
- ).values("row", "seat")
- return [
- {
- "row": ticket["row"],
- "seat": ticket["seat"]
- }
- for ticket in tickets
- ]
+ ).values("row", "seat"))
+ return tickets
diff --git a/services/order.py b/services/order.py
@@ -26,7 +26,7 @@ def create_order(
 
 
 def get_orders(username: str = None) -> Order:
+ orders = Order.objects.select_related("user")
  if username:
- return (Order.objects.select_related("user").
- filter(user__username=username))
- return Order.objects.all()
+ orders = orders.filter(user__username=username)
+ return orders