chore: i am very sorry

Stella-IT · Mar 10, 2021 · d09c0a0 · d09c0a0
1 parent fe5ff47
commit d09c0a0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/API/v1/Interface/__init__.py b/API/v1/Interface/__init__.py
@@ -25,8 +25,8 @@ class SRCopyArgs(BaseModel):
 
 
 class QoSTypeArgs(BaseModel):
-    type: str
-    info: dict
+    type: Optional[str]
+    info: Optional[dict]
 
 
 class VBDCreateArgs(BaseModel):

diff --git a/API/v1/VIF/qos.py b/API/v1/VIF/qos.py
@@ -57,11 +57,11 @@ async def vif_get_qos_type_by_uuid(cluster_id: str, vif_uuid: str, data: QoSType
         vif: VIF = VIF.get_by_uuid(session=session, uuid=vif_uuid)
 
         result = True
-        if data["type"] is not None:
-            vif.set_qos_type(data["type"])
+        if data.type is not None:
+            vif.set_qos_type(data.type)
 
-        if data["info"] is not None:
-            vif.set_qos_info(data["info"])
+        if data.info is not None:
+            vif.set_qos_info(data.info)
 
         ret = dict(success=True, data=result)