Handle poster rooms: change Main Hall A|B|C to Exhibit Hall (#40)

egeakman · web-flow · commit 91d6401037eb · 2024-06-12T16:07:15.000+03:00
diff --git a/src/misc.py b/src/misc.py
@@ -46,9 +46,7 @@ class Room(Enum):
     north_hall = "North Hall"
     south_hall_2a = "South Hall 2A"
     south_hall_2b = "South Hall 2B"
-    main_hall_a = "Main Hall A"
-    main_hall_b = "Main Hall B"
-    main_hall_c = "Main Hall C"
+    exhibit_hall = "Exhibit Hall"
 
 
 class EventType(Enum):
diff --git a/src/models/europython.py b/src/models/europython.py
@@ -2,7 +2,7 @@
 
 from datetime import date, datetime
 
-from pydantic import BaseModel, Field, computed_field, model_validator
+from pydantic import BaseModel, Field, computed_field, field_validator, model_validator
 
 from src.config import Config
 from src.misc import EventType, Room, SpeakerQuestion, SubmissionQuestion
@@ -143,6 +143,13 @@ class EuroPythonSession(BaseModel):
     prev_session: str | None = None
     slot_count: int = Field(..., exclude=True)
 
+    @field_validator("room", mode="before")
+    @classmethod
+    def handle_poster_room(cls, value) -> str | None:
+        if value and "Main Hall" in value:
+            return "Exhibit Hall"
+        return value
+
     @computed_field
     def website_url(self) -> str:
         return (
