Refactoring

lib/Elemente/Angetriebene_Kurve.py

@@ -0,0 +1,48 @@
+
+from pydantic import BaseModel, Field, field_validator
+
+class Angetriebene_Kurve(BaseModel):
+    teileid: str
+    x: float
+    y:float
+    drehung: float = Field(default=0.0,description="Rotation der Kurve")
+    hoehe0: float = Field(default=0.0,description="Hoehe Anfang der Kurve")
+    hoehe1: float = Field(default=0.0,description="Hoehe Ende der Kurve")
+    kurvenrichtung: str = Field(description="Kurvenrichtung der Kurve")
+    antriebNebenStrecke: str =Field(description="wo die Angetriebene Strecke ist abhängig von der kurvenrichtung")
+    winkel: int = Field(description="Der Winkel der Kurve")
+    
+    @property
+    def antrieb(self):
+        if self.antriebNebenStrecke == "Aussen":
+            self.antriebNebenStrecke = "außen"
+        elif self.antriebNebenStrecke == "Innen": 
+            self.antriebNebenStrecke = "innen"
+        return self.antriebNebenStrecke
+    @property
+    def hight_zwischen(self):
+        return ((self.hoehe0 + self.hoehe1) /2)   
+    @classmethod
+    def from_merkmale(cls, teileid: str, x: float, y: float, merkmale: dict) -> 'Angetriebene_Kurve':
+        hoehe0 = float(merkmale.get("Höhe Anfang")) * 1000
+        hoehe1 = float(merkmale.get("Höhe Ende")) * 1000
+        winkel = int(merkmale.get("Kurvenwinkel"))
+        kurvenrichtung = merkmale.get("Kurvenrichtung")
+        antriebNebenstrecke = merkmale.get("AntriebNebenStrecke")
+        try:
+            drehung = float(merkmale.get("Drehung"))
+        except Exception as e:
+            drehung = 0.0
+        
+        return cls(
+            teileid = teileid,
+            x = x,
+            y = y,
+            drehung = drehung,
+            hoehe0 = hoehe0,
+            hoehe1 = hoehe1,
+            kurvenrichtung = kurvenrichtung,
+            antriebNebenStrecke = antriebNebenstrecke,
+            winkel = winkel
+
+        )