Semaine 7

Semaine_07/Jour_04/bank/entities/clients.py

@@ -0,0 +1,43 @@
+from accounts import Account
+from services import client_service
+
+from sqlalchemy import ForeignKey, Column, Integer, String, Float
+from sqlalchemy.orm import relationship
+
+from database import Base, engine
+
+class Client(Base):
+    __tablename__ = "clients"
+
+    id = Column(Integer, primary_key= True, autoincrement= True)
+    first_name = Column(String(200), nullable= False)
+    last_name = Column(String(200), nullable= False)
+    city = Column(String(200))
+    salary = Column(Float)
+
+    bank_id = Column(Integer, ForeignKey("banks.id"), nullable= False)
+
+    bank = relationship("Bank", back_populates= "clients")
+    account = relationship("Account", back_populates= "clients", uselist= False) # type: ignore
+    services = relationship("Services", back_populates= "clients", secondary= client_service)
+
+    def __init__(self, first_name: str, last_name: str, salary: float, city: str, initial_deposit: int | float = 0) -> None:
+        self.first_name       = first_name
+        self.last_name        = last_name
+        self.salary           = salary
+        self.city             = city
+        self.account: Account = Account(initial_deposit = initial_deposit)
+
+    def __eq__(self, other: "Client") -> bool:
+        return self.first_name == other.first_name and\
+               self.last_name  == other.last_name and\
+               self.account    == other.account # type: ignore
+    
+    def __repr__(self) -> str:
+        return f"Client({self.first_name} {self.last_name})"
+
+class ClientNotFound(Exception):
+    def __init__(self, *args: object) -> None:
+        super().__init__("Client not found")
+
+Base.metadata.create_all(engine)