Prepare for production deployment

Author: mraible

functions/host-info/main.py

@@ -1,7 +1,4 @@
 from falconfoundry import FoundryFunction, FoundryRequest, FoundryResponse, FoundryAPIError
-# The following two lines are necessary to import utils
-import sys
-sys.path.append('../')
 from utils import validate_host_id, format_error_response
 from logging import Logger
 from typing import Dict

functions/utils.py functions/host-info/utils.pyfunctions/utils.py renamed to functions/host-info/utils.py

@@ -1,7 +1,4 @@
 from falconfoundry import FoundryFunction, FoundryRequest, FoundryResponse, FoundryAPIError
-# The following two lines are necessary to import utils
-import sys
-sys.path.append('../')
 from utils import validate_email, format_error_response
 from logging import Logger
 from typing import Dict

functions/user-management/main.py

@@ -0,0 +1,26 @@
+import re
+import json
+from typing import Dict, Any, Optional, Union, List
+from falconfoundry import FoundryAPIError
+
+def validate_host_id(host_id: str) -> bool:
+    """Validate that a host ID is in the correct format."""
+    if not host_id or not isinstance(host_id, str):
+        return False
+    return len(host_id) == 32 and all(c in '0123456789abcdef' for c in host_id.lower())
+
+def validate_email(email: str) -> bool:
+    """Validate email format."""
+    pattern = r'^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$'
+    return bool(re.match(pattern, email))
+
+def format_error_response(message: str, code: int = 400) -> list[Any]:
+    """Create a standardized error response."""
+    return [FoundryAPIError(code=code, message=message)]
+
+def safe_json_parse(data: str) -> Optional[Dict[str, Any]]:
+    """Safely parse JSON data."""
+    try:
+        return json.loads(data)
+    except (json.JSONDecodeError, TypeError):
+        return None