Update main.py

main.py

@@ -54,6 +54,25 @@ def modify_title(title: str) -> str:
     parts = title.split(' ')
     return ' '.join(parts[1:]) if len(parts) > 1 else title
 
+def get_sequence_number(role_title: str) -> int:
+    """Extract sequence number from role title for sorting"""
+    try:
+        return int(role_title.split()[0])
+    except (IndexError, ValueError):
+        return float('inf')  # Put roles without numbers at the end
+
+def clean_role_title(role_title: str) -> str:
+    """Remove sequence number from role title"""
+    try:
+        parts = role_title.split()
+        # Check if first part is a number
+        int(parts[0])
+        # If it is, return everything after the first part
+        return ' '.join(parts[1:])
+    except (IndexError, ValueError):
+        # If there's no number or invalid format, return original title
+        return role_title
+
 @app.post("/check-permission")
 async def check_permissions(request: PermissionRequest):
     try:
@@ -86,7 +105,7 @@ async def check_permissions(request: PermissionRequest):
                         modified_title = modify_title(resource_title)
 
                         permissions.append({
-                            "title": modified_title,  # Use modified title here
+                            "title": modified_title,
                             "link": resource,
                             "action": action
                         })
@@ -119,13 +138,24 @@ async def check_permissions(request: PermissionRequest):
                     seen.add(child_tuple)
                     unique_children.append(child)
 
+            # Store original title for sorting
             final_roles.append({
                 "title": role_name,
                 "value": role_name,
                 "icon": "PencilRuler",
-                "children": unique_children
+                "children": unique_children,
+                "original_title": role_name  # Store original title for sorting
             })
 
+        # Sort final_roles based on sequence number
+        final_roles.sort(key=lambda x: get_sequence_number(x["original_title"]))
+
+        # Clean up titles and remove original_title field
+        for role in final_roles:
+            role["title"] = clean_role_title(role["title"])
+            role["value"] = clean_role_title(role["value"])
+            del role["original_title"]
+
         return final_roles
 
     except Exception as e:
@@ -134,3 +164,6 @@ async def check_permissions(request: PermissionRequest):
             status_code=500,
             detail=f"Error checking permissions: {str(e)}"
         )
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=8000)