Add keys_to_upper function

simonrw · simonrw · commit 29929039a517 · 2023-10-16T17:07:33.000+01:00
diff --git a/localstack/utils/objects.py b/localstack/utils/objects.py
@@ -4,7 +4,7 @@
 from typing import Any, Callable, Dict, Generic, List, Optional, Set, Type, TypeVar, Union
 
 from .collections import ensure_list
-from .strings import first_char_to_lower
+from .strings import first_char_to_lower, first_char_to_upper
 
 ComplexType = Union[List, Dict, object]
 
@@ -155,8 +155,10 @@ def recurse_object(obj: ComplexType, func: Callable, path: str = "") -> ComplexT
     return obj
 
 
-def keys_to_lower(obj: ComplexType, skip_children_of: List[str] = None) -> ComplexType:
-    """Recursively changes all dict keys to first character lowercase. Skip children
+def keys_to(
+    obj: ComplexType, op: Callable[[str], str], skip_children_of: List[str] = None
+) -> ComplexType:
+    """Recursively changes all dict keys to apply op. Skip children
     of any elements whose names are contained in skip_children_of (e.g., ['Tags'])"""
     skip_children_of = ensure_list(skip_children_of or [])
 
@@ -166,13 +168,21 @@ def fix_keys(o, path="", **kwargs):
         if isinstance(o, dict):
             for k, v in dict(o).items():
                 o.pop(k)
-                o[first_char_to_lower(k)] = v
+                o[op(k)] = v
         return o
 
     result = recurse_object(obj, fix_keys)
     return result
 
 
+def keys_to_lower(obj: ComplexType, skip_children_of: List[str] = None) -> ComplexType:
+    return keys_to(obj, first_char_to_lower, skip_children_of)
+
+
+def keys_to_upper(obj: ComplexType, skip_children_of: List[str] = None) -> ComplexType:
+    return keys_to(obj, first_char_to_upper, skip_children_of)
+
+
 def singleton_factory(factory: Callable[[], _T]) -> Callable[[], _T]:
     """
     Decorator for methods that create a particular value once and then return the same value in a thread safe way.
