osrd_schemas: add ruff

Signed-off-by: Jean SIMARD <[email protected]>

Author: woshilapin

.github/workflows/build.yml

@@ -395,23 +395,11 @@ jobs:
         run: |
           cd python/osrd_schemas
           poetry install
-      - name: Flake8
+      - name: Ruff
         run: |
           cd python/osrd_schemas
-          poetry run pflake8 --config ./pyproject.toml --output-file flake8.xml --format junit-xml
-      - name: Publish flake8 report
-        uses: mikepenz/action-junit-report@v5
-        if: failure()
-        with:
-          report_paths: flake8.xml
-      - name: Black
-        run: |
-          cd python/osrd_schemas
-          poetry run black . --check
-      - name: Isort
-        run: |
-          cd python/osrd_schemas
-          poetry run isort . --check
+          poetry run ruff check
+          poetry run ruff format --check
       - name: Pytype
         run: |
           cd python/osrd_schemas

python/osrd_schemas/osrd_schemas/external_generated_inputs.py

@@ -10,14 +10,20 @@ class ElectricalProfile(BaseModel):
     There should be one value per power class, on every electrified track."""
 
     value: str = Field(description="Category of power loss along the range")
-    power_class: str = Field(description="Category of rolling stock power usage this profile applies to")
-    track_ranges: List[TrackRange] = Field(description="List of locations where this profile is applied")
+    power_class: str = Field(
+        description="Category of rolling stock power usage this profile applies to"
+    )
+    track_ranges: List[TrackRange] = Field(
+        description="List of locations where this profile is applied"
+    )
 
 
 class ElectricalProfileSet(BaseModel):
     """This class is used to represent a set of electrical profiles, to use in simulation along an infrastructure."""
 
-    levels: List[ElectricalProfile] = Field(description="The list of electrical profiles")
+    levels: List[ElectricalProfile] = Field(
+        description="The list of electrical profiles"
+    )
     level_order: Dict[str, List[str]] = Field(
         description="""A mapping from electrification modes to the electrical profile levels
                     in decreasing order of magnitude"""

python/osrd_schemas/osrd_schemas/generated.py

@@ -13,16 +13,28 @@ class ObjectReference(BaseModel):
 # TRAITS
 class InfraErrorTrait(BaseModel):
     is_warning: Literal[False] = Field(default=False)
-    obj_id: str = Field(description="Identifier of the object that caused the error", max_length=255)
-    obj_type: str = Field(description="Type of the object that caused the error", max_length=32)
-    field: str = Field(description="Field of the object that caused the error", max_length=255)
+    obj_id: str = Field(
+        description="Identifier of the object that caused the error", max_length=255
+    )
+    obj_type: str = Field(
+        description="Type of the object that caused the error", max_length=32
+    )
+    field: str = Field(
+        description="Field of the object that caused the error", max_length=255
+    )
 
 
 class InfraWarningTrait(BaseModel):
     is_warning: Literal[True] = Field(default=True)
-    obj_id: str = Field(description="Identifier of the object that caused the warning", max_length=255)
-    obj_type: str = Field(description="Type of the object that caused the warning", max_length=32)
-    field: str = Field(description="Field of the object that caused the warning", max_length=255)
+    obj_id: str = Field(
+        description="Identifier of the object that caused the warning", max_length=255
+    )
+    obj_type: str = Field(
+        description="Type of the object that caused the warning", max_length=32
+    )
+    field: str = Field(
+        description="Field of the object that caused the warning", max_length=255
+    )
 
 
 # Errors
@@ -53,7 +65,9 @@ class InvalidRoute(InfraErrorTrait):
 
 
 class NodeEndpointsNotUnique(InfraErrorTrait):
-    error_type: Literal["node_endpoints_not_unique"] = Field(default="node_endpoints_not_unique")
+    error_type: Literal["node_endpoints_not_unique"] = Field(
+        default="node_endpoints_not_unique"
+    )
 
 
 class UnknownPortName(InfraErrorTrait):
@@ -94,16 +108,22 @@ class MissingBufferStop(InfraWarningTrait):
 
 
 class OddBufferStopLocation(InfraWarningTrait):
-    error_type: Literal["odd_buffer_stop_location"] = Field(default="odd_buffer_stop_location")
+    error_type: Literal["odd_buffer_stop_location"] = Field(
+        default="odd_buffer_stop_location"
+    )
 
 
 class OverlappingSpeedSections(InfraWarningTrait):
-    error_type: Literal["overlapping_speed_sections"] = Field(default="overlapping_speed_sections")
+    error_type: Literal["overlapping_speed_sections"] = Field(
+        default="overlapping_speed_sections"
+    )
     reference: ObjectReference
 
 
 class OverlappingElectrifications(InfraWarningTrait):
-    error_type: Literal["overlapping_electrifications"] = Field(default="overlapping_electrifications")
+    error_type: Literal["overlapping_electrifications"] = Field(
+        default="overlapping_electrifications"
+    )
     reference: ObjectReference