singlestoredb 0.10.1__cp38-abi3-win32.whl → 0.10.2__cp38-abi3-win32.whl

singlestoredb/__init__.py

@@ -13,7 +13,7 @@ Examples
 
 """
 
-__version__ = '0.10.1'
+__version__ = '0.10.2'
 
 from .alchemy import create_engine
 from .config import options, get_option, set_option, describe_option

singlestoredb/fusion/handlers/utils.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python
 import datetime
+import os
 from typing import Any
 from typing import Dict
 from typing import Optional
@@ -26,15 +27,17 @@ def get_workspace_group(params: Dict[str, Any]) -> WorkspaceGroup:
     """Find a workspace group matching group_id or group_name."""
     manager = get_workspace_manager()
 
-    if params['group_name']:
+    group_name = params.get('group_name') or \
+        (params.get('in_group') or {}).get('group_name')
+    if group_name:
         workspace_groups = [
             x for x in manager.workspace_groups
-            if x.name == params['group_name']
+            if x.name == group_name
         ]
 
         if not workspace_groups:
             raise KeyError(
-                'no workspace group found with name "{}"'.format(params['group_name']),
+                'no workspace group found with name "{}"'.format(group_name),
             )
 
         if len(workspace_groups) > 1:
@@ -45,4 +48,17 @@ def get_workspace_group(params: Dict[str, Any]) -> WorkspaceGroup:
 
         return workspace_groups[0]
 
-    return manager.get_workspace_group(params['group_id'])
+    group_id = params.get('group_id') or \
+        (params.get('in_group') or {}).get('group_id')
+    if group_id:
+        return manager.get_workspace_group(group_id)
+
+    if os.environ.get('SINGLESTOREDB_WORKSPACE_GROUP'):
+        return manager.get_workspace_group(
+            os.environ['SINGLESTOREDB_WORKSPACE_GROUP'],
+        )
+
+    if os.environ.get('SINGLESTOREDB_CLUSTER'):
+        raise ValueError('clusters and shared workspaces are not currently supported')
+
+    raise KeyError('no workspace group was specified')

singlestoredb/fusion/handlers/workspace.py

@@ -80,8 +80,10 @@ ShowWorkspaceGroupsHandler.register()
 
 class ShowWorkspacesHandler(SQLHandler):
     """
-    SHOW WORKSPACES IN GROUP { group_id | group_name }
-        [ <like> ] [ <extended> ] [ <order-by> ] [ <limit> ];
+    SHOW WORKSPACES [ in_group ] [ <like> ] [ <extended> ] [ <order-by> ] [ <limit> ];
+
+    # Workspace group
+    in_group = IN GROUP { group_id | group_name }
 
     # ID of group
     group_id = ID '<group-id>'
@@ -198,10 +200,12 @@ CreateWorkspaceGroupHandler.register()
 
 class CreateWorkspaceHandler(SQLHandler):
     """
-    CREATE WORKSPACE [ if_not_exists ] workspace_name
-        IN GROUP { group_id | group_name }
+    CREATE WORKSPACE [ if_not_exists ] workspace_name [ in_group ]
         WITH SIZE size [ wait_on_active ];
 
+    # Create workspace in workspace group
+    in_group = IN GROUP { group_id | group_name }
+
     # Only run command if workspace doesn't already exist
     if_not_exists = IF NOT EXISTS
 
@@ -264,15 +268,15 @@ class DropWorkspaceGroupHandler(SQLHandler):
     """
 
     def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
-        manager = get_workspace_manager()
-
         try:
-            name_or_id = params['group_name'] or params['group_id']
-            wg = manager.workspace_groups[name_or_id]
-            wg.terminate(wait_on_terminated=params['wait_on_terminated'])
+            workspace_group = get_workspace_group(params)
+            if workspace_group.terminated_at is not None:
+                raise KeyError
+            workspace_group.terminate(wait_on_terminated=params['wait_on_terminated'])
 
         except KeyError:
             if not params['if_exists']:
+                name_or_id = params['group_id'] or params['group_name']
                 raise KeyError(f"could not find workspace group '{name_or_id}'")
 
         return None
@@ -284,7 +288,10 @@ DropWorkspaceGroupHandler.register()
 class DropWorkspaceHandler(SQLHandler):
     """
     DROP WORKSPACE [ if_exists ] { workspace_id | workspace_name }
-        IN GROUP { group_id | group_name } [ wait_on_terminated ];
+        [ in_group ] [ wait_on_terminated ];
+
+    # Workspace group
+    in_group = IN GROUP { group_id | group_name }
 
     # Only drop workspace if it exists
     if_exists = IF EXISTS
@@ -307,16 +314,15 @@ class DropWorkspaceHandler(SQLHandler):
     """
 
     def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
-        manager = get_workspace_manager()
-
         try:
+            workspace_group = get_workspace_group(params)
             workspace_name_or_id = params['workspace_name'] or params['workspace_id']
-            group_name_or_id = params['group_name'] or params['group_id']
-            wg = manager.workspace_groups[group_name_or_id]
-            ws = wg.workspaces[workspace_name_or_id]
+            ws = workspace_group.workspaces[workspace_name_or_id]
             ws.terminate(wait_on_terminated=params['wait_on_terminated'])
 
         except KeyError:
+            group_name_or_id = params['in_group'].get('group_id', None) or \
+                               params['in_group'].get('group_name', None)
             if not params['if_exists']:
                 raise KeyError(
                     f"could not find workspace '{workspace_name_or_id}' "

{singlestoredb-0.10.1.dist-info → singlestoredb-0.10.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: singlestoredb
-Version: 0.10.1
+Version: 0.10.2
 Summary: Interface to the SingleStoreDB database and workspace management APIs
 Home-page: https://github.com/singlestore-labs/singlestoredb-python
 Author: SingleStore

{singlestoredb-0.10.1.dist-info → singlestoredb-0.10.2.dist-info}/RECORD

@@ -1,5 +1,5 @@
-_singlestoredb_accel.pyd,sha256=mCY7pUnRTRR7-MsBxCgVv4-xdev2CoBAFWRoD0qMaKk,42496
-singlestoredb/__init__.py,sha256=pfNhZinZq6ma34r0C7lVLA1LbrG9pJhwEQuyOdB1hVM,910
+_singlestoredb_accel.pyd,sha256=Q7DZfTnjn_9MIileaf3QbX7iZhD139iuixKmuJLiAys,42496
+singlestoredb/__init__.py,sha256=HVmH2GEx2_Vmnv0hvAFuRaB2yiP3MMmGY6cfiqJ2AFo,910
 singlestoredb/auth.py,sha256=RmYiH0Wlc2RXc4pTlRMysxtBI445ggCIwojWKC_eDLE,7844
 singlestoredb/config.py,sha256=Uq7LWKlIwMAJC3_G3j5dUIQVJf2JzbDVtRK6votjuDw,7746
 singlestoredb/connection.py,sha256=gSKmMN1d_ZSXVTmzfuHK3BXcLyNGiaSi2Vy6z7nymjg,45657
@@ -22,8 +22,8 @@ singlestoredb/fusion/registry.py,sha256=7wELKu09I6N4Oif4lpqAHCy8p_C4COn85hhtoCN5
 singlestoredb/fusion/result.py,sha256=EcFY5Qv43ySlQsfl_JB-I3ko7PzVdjuhhoKN96uHSAM,12171
 singlestoredb/fusion/handlers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 singlestoredb/fusion/handlers/stage.py,sha256=VEoHU3q5BX3lCl6pG9jVf-eviqkTZhyzTbO3sRKrMPs,5426
-singlestoredb/fusion/handlers/utils.py,sha256=GPDlvm-hU2zwTYczyITHcWwmBc4BwhiVN5JF3SeX2Nw,1476
-singlestoredb/fusion/handlers/workspace.py,sha256=Q6rEa0I74V3ujn_JHzUAROz8GxgLMjLWSaeiVpC5ZSs,9770
+singlestoredb/fusion/handlers/utils.py,sha256=H79iVSp0EZKyL3vI8OX8m8cBuFfU-7X9u5CjqnFBmRs,2045
+singlestoredb/fusion/handlers/workspace.py,sha256=b21Vi6zZPi8TyWaI7BcAF4L7KGU6xFNV3EAN7wF1fN4,10045
 singlestoredb/http/__init__.py,sha256=4cEDvLloGc3LSpU-PnIwacyu0n5oIIIE6xk2SPyWD_w,939
 singlestoredb/http/connection.py,sha256=vYP10ZsqrhK8RrLS17OVuMkpTsO0VZQu3bxx1FQzUwU,37839
 singlestoredb/management/__init__.py,sha256=kdKM-wErALWaEHYBLx_raHkJ8xUvUPflhOk8OBJyn58,173
@@ -98,8 +98,8 @@ singlestoredb/utils/debug.py,sha256=y7dnJeJGt3U_BWXz9pLt1qNQREpPtumYX_sk1DiqG6Y,
 singlestoredb/utils/mogrify.py,sha256=gCcn99-vgsGVjTUV7RHJ6hH4vCNrsGB_Xo4z8kiSPDQ,4201
 singlestoredb/utils/results.py,sha256=ely2XVAHHejObjLibS3UcrPOuCO2g5aRtA3PxAMtE-g,5432
 singlestoredb/utils/xdict.py,sha256=-wi1lSPTnY99fhVMBhPKJ8cCsQhNG4GMUfkEBDKYgCw,13321
-singlestoredb-0.10.1.dist-info/LICENSE,sha256=Bojenzui8aPNjlF3w4ojguDP7sTf8vFV_9Gc2UAG1sg,11542
-singlestoredb-0.10.1.dist-info/METADATA,sha256=iCAwLrDXbtWnwNl_MBAvGmu9vU0Ehm9WeD_pWrcf4mU,7794
-singlestoredb-0.10.1.dist-info/WHEEL,sha256=JjDd0FQBSruYmZy_b8ddnWkQmalS34MTxErrvze-qAs,96
-singlestoredb-0.10.1.dist-info/top_level.txt,sha256=SDtemIXf-Kp-_F2f_S6x0db33cHGOILdAEsIQZe2LZc,35
-singlestoredb-0.10.1.dist-info/RECORD,,
+singlestoredb-0.10.2.dist-info/LICENSE,sha256=Bojenzui8aPNjlF3w4ojguDP7sTf8vFV_9Gc2UAG1sg,11542
+singlestoredb-0.10.2.dist-info/METADATA,sha256=WJLPuZGqhIr4yEui0bNp7_4jhJkryaQ4-ZdHRkhkXus,7794
+singlestoredb-0.10.2.dist-info/WHEEL,sha256=JjDd0FQBSruYmZy_b8ddnWkQmalS34MTxErrvze-qAs,96
+singlestoredb-0.10.2.dist-info/top_level.txt,sha256=SDtemIXf-Kp-_F2f_S6x0db33cHGOILdAEsIQZe2LZc,35
+singlestoredb-0.10.2.dist-info/RECORD,,