Implemented py-lockdown-cafe

app/cafe.py

@@ -0,0 +1,24 @@
+import datetime
+
+from app.errors import (
+    NotVaccinatedError,
+    OutdatedVaccineError,
+    NotWearingMaskError
+)
+
+
+class Cafe:
+    def __init__(self, name: str) -> None:
+        self.name = name
+
+    def visit_cafe(
+        self, visitor: dict
+    ) -> str:
+        if "vaccine" not in visitor:
+            raise NotVaccinatedError
+        current_date = datetime.datetime.now().date()
+        if visitor["vaccine"]["expiration_date"] < current_date:
+            raise OutdatedVaccineError
+        if not visitor.get("wearing_a_mask", False):
+            raise NotWearingMaskError
+        return f"Welcome to {self.name}"

app/errors.py

@@ -0,0 +1,20 @@
+class VaccineError(Exception):
+    def __str__(self) -> str:
+        return "All friends should be vaccinated"
+
+
+class NotVaccinatedError(VaccineError):
+    pass
+
+
+class OutdatedVaccineError(VaccineError):
+    pass
+
+
+class NotWearingMaskError(Exception):
+    def __init__(self, masks_to_buy: int = 0) -> None:
+        self.masks_to_buy = masks_to_buy
+        super().__init__()
+
+    def __str__(self) -> str:
+        return f"Friends should buy {self.masks_to_buy} masks"

app/main.py

@@ -1 +1,24 @@
-# write your code here
+from app.cafe import Cafe
+from app.errors import (
+    NotWearingMaskError,
+    VaccineError,
+)
+
+
+def go_to_cafe(friends: list, cafe: Cafe) -> str | VaccineError:
+    not_wearing_mask_error = None
+
+    for friend in friends:
+        try:
+            cafe.visit_cafe(friend)
+        except VaccineError as e:
+            return str(e)
+        except NotWearingMaskError as e:
+            if not not_wearing_mask_error:
+                not_wearing_mask_error = e
+            if not_wearing_mask_error:
+                not_wearing_mask_error.masks_to_buy += 1
+
+    if not_wearing_mask_error is not None:
+        return str(not_wearing_mask_error)
+    return f"Friends can go to {cafe.name}"