RAI-12688 - Removing EngineSize enumeration to give more flexibility on engine specification (#125)

* removing EngineSize enumeration to give more flexibility on engine specification

* updated CHANGELOG

Author: Anatoli Kurtsevich

CHANGELOG.md

@@ -1,5 +1,9 @@
 # Changelog
 
+## v0.6.12
+
+* `create_engine` and `create_engine_wait` accept engine size as a string (e.g. "XS", "M", etc).
+
 ## v0.6.11
 
 * Log warnings if failed to read or write to the local access token cache.

examples/create_engine.py

@@ -18,13 +18,12 @@
 import json
 from urllib.request import HTTPError
 from railib import api, config, show
-from railib.api import EngineSize
 
 
 def run(engine: str, size: str, profile: str):
     cfg = config.read(profile=profile)
     ctx = api.Context(**cfg)
-    api.create_engine_wait(ctx, engine, EngineSize(size))
+    api.create_engine(ctx, engine, size)
     print(json.dumps(api.get_engine(ctx, engine), indent=2))
 
 

railib/api.py

@@ -39,16 +39,6 @@
 logger = logging.getLogger(__package__)
 
 
-# Engine sizes
-@unique
-class EngineSize(str, Enum):
-    XS = "XS"
-    S = "S"
-    M = "M"
-    L = "L"
-    XL = "XL"
-
-
 # Database modes
 @unique
 class Mode(str, Enum):
@@ -113,7 +103,6 @@ class Permission(str, Enum):
 
 __all__ = [
     "Context",
-    "EngineSize",
     "Mode",
     "Role",
     "Permission",
@@ -371,14 +360,14 @@ def is_engine_term_state(state: str) -> bool:
     return state == "PROVISIONED" or ("FAILED" in state)
 
 
-def create_engine(ctx: Context, engine: str, size: EngineSize = EngineSize.XS, **kwargs):
-    data = {"region": ctx.region, "name": engine, "size": size.value}
+def create_engine(ctx: Context, engine: str, size: str = "XS", **kwargs):
+    data = {"region": ctx.region, "name": engine, "size": size}
     url = _mkurl(ctx, PATH_ENGINE)
     rsp = rest.put(ctx, url, data, **kwargs)
     return json.loads(rsp.read())
 
 
-def create_engine_wait(ctx: Context, engine: str, size: EngineSize = EngineSize.XS, **kwargs):
+def create_engine_wait(ctx: Context, engine: str, size: str = "XS", **kwargs):
     create_engine(ctx, engine, size, **kwargs)
     poll_with_specified_overhead(
         lambda: is_engine_term_state(get_engine(ctx, engine)["state"]),