Semaine 11

Semaine_11/Jour_01/inventaire_orm.py

@@ -0,0 +1,94 @@
+"""
+**Consignes :**
+
+1. Définissez deux modèles `Equipement` et `Interface` avec une relation **1-N** (`relationship` + `ForeignKey` + `back_populates`).
+
+2. Créez le moteur SQLite et générez les tables (`create_all`).
+
+3. Dans une `Session`, créez **un équipement avec ses deux interfaces en une seule fois**, puis `commit`.
+
+4. Récupérez les équipements avec `select()` et affichez, pour chacun, la liste de ses interfaces.
+"""
+
+from sqlalchemy import create_engine, select
+from sqlalchemy.orm import DeclarativeMeta, declarative_base, sessionmaker
+
+Base: DeclarativeMeta = declarative_base()
+db_url = "sqlite:///exo_2_1.db"
+
+def get_engine():
+    return create_engine(url= db_url)
+
+def get_session():
+    engine = get_engine()
+    Session = sessionmaker(bind= engine)
+    return Session()
+
+engine = get_engine()
+
+from random import choice, randint
+
+from sqlalchemy import ForeignKey, Column, Integer, Float, String, Boolean
+from sqlalchemy.orm import relationship
+
+class Equipment(Base):
+    __tablename__ = "equipments"
+
+    id = Column(Integer, primary_key= True, autoincrement= True)
+    hostname = Column(String, default= "[HOSTNAME]", nullable= False)
+    type = Column(String, default= "Server", nullable= False)
+    ip_address = Column(String, default= "255.255.255.0", nullable= False)
+    is_active = Column(Boolean, default= True, nullable= False)
+    latency = Column(Float, nullable= False)
+
+    interfaces = relationship("Interface", back_populates="equipment")
+
+class Interface(Base):
+    __tablename__ = "interfaces"
+
+    id = Column(Integer, primary_key= True, autoincrement= True)
+    port = Column(Integer, nullable= False)
+    service = Column(String, default= "[SERVICE]", nullable= False)
+
+    equipment_id = Column(Integer, ForeignKey("equipments.id"), nullable= False)
+
+    equipment = relationship("Equipment", back_populates="interfaces")
+
+Base.metadata.create_all(engine)
+
+session = get_session()
+
+def random_ipv4() -> str:
+    return ".".join(str(randint(0, 255)) for _ in range(4))
+
+def load_words(path: str = "wordlist.txt") -> list[str]:
+    with open(path, "r", encoding="utf-8") as f:
+        return [line.strip() for line in f if line.strip()]
+
+def random_hostname(words: list[str] | None = None) -> str:
+    if words is None:
+        words = load_words()
+    triplet = "-".join(choice(words) for _ in range(3))
+    number = f"{randint(1, 999):03d}"
+    return f"{triplet}-{number}"
+
+equipment = Equipment(
+    hostname=random_hostname(),
+    type="Server",
+    ip_address=random_ipv4(),
+    is_active=True,
+    latency=12.5,
+    interfaces=[
+        Interface(port=22, service="SSH"),
+        Interface(port=443, service="HTTPS"),
+    ],
+)
+
+session.add(equipment)
+session.commit()
+
+stmt = select(Equipment)
+for eq in session.execute(stmt).scalars():
+    print(f"{eq.hostname} ({eq.ip_address})")
+    for iface in eq.interfaces:
+        print(f"  - port {iface.port} : {iface.service}")