Add role-based filtering and imporve code

gateway/app/core/service_discovery.py

@@ -13,18 +13,18 @@ logger = logging.getLogger(__name__)
 
 class ServiceDiscovery:
     """Service discovery client"""
-    
+
     def __init__(self):
         self.consul_url = settings.CONSUL_URL if hasattr(settings, 'CONSUL_URL') else None
         self.service_cache: Dict[str, str] = {}
-        
+
     async def get_service_url(self, service_name: str) -> Optional[str]:
         """Get service URL from service discovery"""
-        
+
         # Return cached URL if available
         if service_name in self.service_cache:
             return self.service_cache[service_name]
-        
+
         # Try Consul if enabled
         if self.consul_url and getattr(settings, 'ENABLE_SERVICE_DISCOVERY', False):
             try:
@@ -34,10 +34,10 @@ class ServiceDiscovery:
                     return url
             except Exception as e:
                 logger.warning(f"Failed to get {service_name} from Consul: {e}")
-        
+
         # Fall back to environment variables
         return self._get_from_env(service_name)
-    
+
     async def _get_from_consul(self, service_name: str) -> Optional[str]:
         """Get service URL from Consul"""
         try:
@@ -45,7 +45,7 @@ class ServiceDiscovery:
                 response = await client.get(
                     f"{self.consul_url}/v1/health/service/{service_name}?passing=true"
                 )
-                
+
                 if response.status_code == 200:
                     services = response.json()
                     if services:
@@ -53,13 +53,13 @@ class ServiceDiscovery:
                         address = service['Service']['Address']
                         port = service['Service']['Port']
                         return f"http://{address}:{port}"
-                        
+
         except Exception as e:
             logger.error(f"Consul query failed: {e}")
-            
+
         return None
-    
+
     def _get_from_env(self, service_name: str) -> Optional[str]:
         """Get service URL from environment variables"""
         env_var = f"{service_name.upper().replace('-', '_')}_SERVICE_URL"
-        return getattr(settings, env_var, None)
+        return getattr(settings, env_var, None)