ominfra 0.0.0.dev207__tar.gz → 0.0.0.dev209__tar.gz

{ominfra-0.0.0.dev207/ominfra.egg-info → ominfra-0.0.0.dev209}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ominfra
-Version: 0.0.0.dev207
+Version: 0.0.0.dev209
 Summary: ominfra
 Author: wrmsr
 License: BSD-3-Clause
@@ -12,9 +12,9 @@ Classifier: Operating System :: OS Independent
 Classifier: Operating System :: POSIX
 Requires-Python: >=3.12
 License-File: LICENSE
-Requires-Dist: omdev==0.0.0.dev207
-Requires-Dist: omlish==0.0.0.dev207
-Requires-Dist: omserv==0.0.0.dev207
+Requires-Dist: omdev==0.0.0.dev209
+Requires-Dist: omlish==0.0.0.dev209
+Requires-Dist: omserv==0.0.0.dev209
 Provides-Extra: all
 Requires-Dist: paramiko~=3.5; extra == "all"
 Requires-Dist: asyncssh~=2.18; extra == "all"

{ominfra-0.0.0.dev207 → ominfra-0.0.0.dev209}/ominfra/clouds/aws/models/gen/cli.py

@@ -4,6 +4,7 @@ import os.path
 import sys
 import typing as ta
 
+from omlish import check
 from omlish import lang
 from omlish import marshal as msh
 from omlish.argparse import all as ap
@@ -30,14 +31,17 @@ class Cli(ap.Cli):
     ) -> str:
         service_model = ModelGen.load_service_model(service_name)
 
+        shape_names_seq = check.unique(shape_names or ())
+        operation_names_seq = check.unique(operation_names or ())
+
         bmg = ModelGen(
             service_model,
             shape_names=ModelGen.get_referenced_shape_names(
                 service_model,
-                shape_names=shape_names or (),
-                operation_names=operation_names or (),
+                shape_names=shape_names_seq,
+                operation_names=operation_names_seq,
             ),
-            operation_names=operation_names or (),
+            operation_names=operation_names_seq,
         )
 
         return bmg.gen_module()

{ominfra-0.0.0.dev207 → ominfra-0.0.0.dev209}/ominfra/clouds/aws/models/gen/gen.py

@@ -6,6 +6,7 @@ TODO:
 import builtins
 import dataclasses as dc
 import io
+import keyword
 import typing as ta
 
 from omlish import check
@@ -182,7 +183,7 @@ class ModelGen:
             pass
 
         if name in self._shape_names:
-            name = self.sanitize_class_name(name)
+            name = self.sanitize_global_name(name, upper=True)
             if not unquoted_names:
                 return f"'{name}'"
             else:
@@ -249,10 +250,17 @@ class ModelGen:
 
     #
 
-    def sanitize_class_name(self, n: str) -> str:
+    def sanitize_local_name(self, n: str) -> str:
+        if n in keyword.kwlist:
+            n += '_'
+        return n
+
+    def sanitize_global_name(self, n: str, *, upper: bool = False) -> str:
         if hasattr(builtins, n):
             n += '_'
-        return n[0].upper() + n[1:]
+        if upper:
+            n = n[0].upper() + n[1:]
+        return n
 
     #
 
@@ -292,7 +300,7 @@ class ModelGen:
     ) -> ShapeSrc:
         shape: botocore.model.Shape = self._service_model.shape_for(name)
 
-        san_name = self.sanitize_class_name(shape.name)
+        san_name = self.sanitize_global_name(shape.name, upper=True)
 
         if isinstance(shape, botocore.model.StructureShape):
             lines: list[str] = []
@@ -315,7 +323,7 @@ class ModelGen:
             for i, (mn, ms) in enumerate(shape.members.items()):
                 if i:
                     lines.append('')
-                fn = self.demangle_name(mn)
+                fn = self.sanitize_local_name(self.demangle_name(mn))
                 mds = [
                     f'member_name={mn!r}',
                 ]