fix: radar camera add enabled

closes open-v2x#222
qiaolei1 · May 10, 2023 · beb5dda · beb5dda
1 parent 7e4fb36
commit beb5dda
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 1 deletion.
diff --git a/dandelion/alembic/versions/26c910b0342f_radar_camera.py b/dandelion/alembic/versions/26c910b0342f_radar_camera.py
@@ -35,6 +35,7 @@ def upgrade():
         sa.Column("rsu_id", sa.Integer(), nullable=True),
         sa.Column("status", sa.Boolean(), nullable=False),
         sa.Column("desc", sa.String(length=255), nullable=False),
+        sa.Column("enabled", sa.Boolean(), nullable=True),
         sa.ForeignKeyConstraint(
             ["rsu_id"],
             ["rsu.id"],

diff --git a/dandelion/models/radar_camera.py b/dandelion/models/radar_camera.py
@@ -37,6 +37,7 @@ class RadarCamera(Base, DandelionBase):
 
     status = Column(Boolean, nullable=False, default=False)
     desc = Column(String(255), nullable=False, default="")
+    enabled = Column(Boolean, nullable=True, default=True)
 
     def __repr__(self) -> str:
         return f"<RadarCamera(sn='{self.sn}', name='{self.name}')>"
@@ -59,4 +60,5 @@ def to_all_dict(self):
             rsuName=Optional_util.none(self.rsu).map(lambda v: v.rsu_name).get(),
             desc=self.desc,
             createTime=self.create_time,
+            enabled=self.enabled,
         )
diff --git a/dandelion/schemas/radar_camera.py b/dandelion/schemas/radar_camera.py
@@ -63,6 +63,7 @@ class RadarCameraUpdate(BaseModel):
     )
     rsu_id: Optional[int] = Field(None, alias="rsuID", description="RSU ID")
     desc: Optional[str] = Field("", alias="desc", description="Description")
+    enabled: Optional[bool] = Field(None, alias="enabled", description="enabled")
 
 
 class RadarCameraInDBBase(RadarCameraBase):
@@ -77,6 +78,7 @@ class RadarCamera(RadarCameraInDBBase):
     status: bool = Field(..., alias="status", description="Status")
     rsu_name: Optional[str] = Field(None, alias="rsuName", description="RSU Name")
     create_time: datetime = Field(..., alias="createTime", description="Create Time")
+    enabled: bool = Field(..., alias="enabled", description="Enabled radar camera or not")
 
 
 class RadarCameras(BaseModel):

diff --git a/swagger.json b/swagger.json
@@ -16044,7 +16044,8 @@
                     "videoStreamAddress",
                     "id",
                     "status",
-                    "createTime"
+                    "createTime",
+                    "enabled"
                 ],
                 "type": "object",
                 "properties": {
@@ -16129,6 +16130,11 @@
                         "type": "string",
                         "description": "Create Time",
                         "format": "date-time"
+                    },
+                    "enabled": {
+                        "title": "Enabled",
+                        "type": "boolean",
+                        "description": "Enabled radar camera or not"
                     }
                 }
             },
@@ -16275,6 +16281,11 @@
                         "type": "string",
                         "description": "Description",
                         "default": ""
+                    },
+                    "enabled": {
+                        "title": "Enabled",
+                        "type": "boolean",
+                        "description": "enabled"
                     }
                 }
             },