Implemented py-lockdown-cafe

app/cafe.py

@@ -0,0 +1,29 @@
+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
+                         | NotVaccinatedError
+                         | OutdatedVaccineError
+                         | NotWearingMaskError):
+
+        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(masks_to_buy=0)
+
+        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) -> 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,25 @@
-# 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}"