singlestoredb 1.12.0__cp38-abi3-macosx_10_9_universal2.whl → 1.12.1__cp38-abi3-macosx_10_9_universal2.whl

singlestoredb/__init__.py

@@ -13,7 +13,7 @@ Examples
 
 """
 
-__version__ = '1.12.0'
+__version__ = '1.12.1'
 
 from typing import Any
 

singlestoredb/fusion/handlers/utils.py

@@ -7,13 +7,11 @@ from typing import Optional
 from typing import Union
 
 from ...exceptions import ManagementError
+from ...management import files as mgmt_files
 from ...management import manage_workspaces
 from ...management.files import FilesManager
 from ...management.files import FileSpace
 from ...management.files import manage_files
-from ...management.files import MODELS_SPACE
-from ...management.files import PERSONAL_SPACE
-from ...management.files import SHARED_SPACE
 from ...management.workspace import StarterWorkspace
 from ...management.workspace import Workspace
 from ...management.workspace import WorkspaceGroup
@@ -190,6 +188,10 @@ def get_deployment(
         * params['group']['deployment_id']
         * params['in_deployment']['deployment_name']
         * params['in_deployment']['deployment_id']
+        * params['in']['in_group']['deployment_name']
+        * params['in']['in_group']['deployment_id']
+        * params['in']['in_deployment']['deployment_name']
+        * params['in']['in_deployment']['deployment_id']
 
     Or, from the SINGLESTOREDB_WORKSPACE_GROUP
     or SINGLESTOREDB_CLUSTER environment variables.
@@ -199,7 +201,9 @@ def get_deployment(
 
     deployment_name = params.get('deployment_name') or \
         (params.get('in_deployment') or {}).get('deployment_name') or \
-        (params.get('group') or {}).get('deployment_name')
+        (params.get('group') or {}).get('deployment_name') or \
+        ((params.get('in') or {}).get('in_group') or {}).get('deployment_name') or \
+        ((params.get('in') or {}).get('in_deployment') or {}).get('deployment_name')
     if deployment_name:
         workspace_groups = [
             x for x in manager.workspace_groups
@@ -239,7 +243,9 @@ def get_deployment(
 
     deployment_id = params.get('deployment_id') or \
         (params.get('in_deployment') or {}).get('deployment_id') or \
-        (params.get('group') or {}).get('deployment_id')
+        (params.get('group') or {}).get('deployment_id') or \
+        ((params.get('in') or {}).get('in_group') or {}).get('deployment_id') or \
+        ((params.get('in') or {}).get('in_deployment') or {}).get('deployment_id')
     if deployment_id:
         try:
             return manager.get_workspace_group(deployment_id)
@@ -298,11 +304,11 @@ def get_file_space(params: Dict[str, Any]) -> FileSpace:
     if file_location:
         file_location_lower_case = file_location.lower()
 
-        if file_location_lower_case == PERSONAL_SPACE:
+        if file_location_lower_case == mgmt_files.PERSONAL_SPACE:
             return manager.personal_space
-        elif file_location_lower_case == SHARED_SPACE:
+        elif file_location_lower_case == mgmt_files.SHARED_SPACE:
             return manager.shared_space
-        elif file_location_lower_case == MODELS_SPACE:
+        elif file_location_lower_case == mgmt_files.MODELS_SPACE:
             return manager.models_space
         else:
             raise ValueError(f'invalid file location: {file_location}')

singlestoredb/management/files.py

@@ -10,11 +10,9 @@ import re
 from abc import ABC
 from abc import abstractmethod
 from typing import Any
-from typing import BinaryIO
 from typing import Dict
 from typing import List
 from typing import Optional
-from typing import TextIO
 from typing import Union
 
 from .. import config
@@ -362,7 +360,7 @@ class FileLocation(ABC):
     @abstractmethod
     def upload_file(
         self,
-        local_path: Union[PathLike, TextIO, BinaryIO],
+        local_path: Union[PathLike, io.IOBase],
         path: PathLike,
         *,
         overwrite: bool = False,
@@ -385,7 +383,7 @@ class FileLocation(ABC):
     @abstractmethod
     def _upload(
         self,
-        content: Union[str, bytes, TextIO, BinaryIO],
+        content: Union[str, bytes, io.IOBase],
         path: PathLike,
         *,
         overwrite: bool = False,
@@ -628,7 +626,7 @@ class FileSpace(FileLocation):
 
     def upload_file(
         self,
-        local_path: Union[PathLike, TextIO, BinaryIO],
+        local_path: Union[PathLike, io.IOBase],
         path: PathLike,
         *,
         overwrite: bool = False,
@@ -646,7 +644,7 @@ class FileSpace(FileLocation):
             Should the ``path`` be overwritten if it exists already?
 
         """
-        if isinstance(local_path, (TextIO, BinaryIO)):
+        if isinstance(local_path, io.IOBase):
             pass
         elif not os.path.isfile(local_path):
             raise IsADirectoryError(f'local path is not a file: {local_path}')
@@ -657,8 +655,9 @@ class FileSpace(FileLocation):
 
             self.remove(path)
 
-        if isinstance(local_path, (TextIO, BinaryIO)):
+        if isinstance(local_path, io.IOBase):
             return self._upload(local_path, path, overwrite=overwrite)
+
         return self._upload(open(local_path, 'rb'), path, overwrite=overwrite)
 
     def upload_folder(
@@ -727,7 +726,7 @@ class FileSpace(FileLocation):
 
     def _upload(
         self,
-        content: Union[str, bytes, TextIO, BinaryIO],
+        content: Union[str, bytes, io.IOBase],
         path: PathLike,
         *,
         overwrite: bool = False,

singlestoredb/management/workspace.py

@@ -10,11 +10,9 @@ import re
 import time
 from collections.abc import Mapping
 from typing import Any
-from typing import BinaryIO
 from typing import Dict
 from typing import List
 from typing import Optional
-from typing import TextIO
 from typing import Union
 
 from .. import config
@@ -165,7 +163,7 @@ class Stage(FileLocation):
 
     def upload_file(
         self,
-        local_path: Union[PathLike, TextIO, BinaryIO],
+        local_path: Union[PathLike, io.IOBase],
         stage_path: PathLike,
         *,
         overwrite: bool = False,
@@ -183,7 +181,7 @@ class Stage(FileLocation):
             Should the ``stage_path`` be overwritten if it exists already?
 
         """
-        if isinstance(local_path, (TextIO, BinaryIO)):
+        if isinstance(local_path, io.IOBase):
             pass
         elif not os.path.isfile(local_path):
             raise IsADirectoryError(f'local path is not a file: {local_path}')
@@ -194,8 +192,9 @@ class Stage(FileLocation):
 
             self.remove(stage_path)
 
-        if isinstance(local_path, (TextIO, BinaryIO)):
+        if isinstance(local_path, io.IOBase):
             return self._upload(local_path, stage_path, overwrite=overwrite)
+
         return self._upload(open(local_path, 'rb'), stage_path, overwrite=overwrite)
 
     def upload_folder(
@@ -258,7 +257,7 @@ class Stage(FileLocation):
 
     def _upload(
         self,
-        content: Union[str, bytes, TextIO, BinaryIO],
+        content: Union[str, bytes, io.IOBase],
         stage_path: PathLike,
         *,
         overwrite: bool = False,

singlestoredb/tests/test_management.py

@@ -414,8 +414,12 @@ class TestStage(unittest.TestCase):
         with self.assertRaises(OSError):
             st.upload_file(TEST_DIR / 'test.sql', 'upload_test.sql')
 
-        # Force overwrite with new content
-        f = st.upload_file(TEST_DIR / 'test2.sql', 'upload_test.sql', overwrite=True)
+        # Force overwrite with new content; use file object this time
+        f = st.upload_file(
+            open(TEST_DIR / 'test2.sql', 'r'),
+            'upload_test.sql',
+            overwrite=True,
+        )
         assert str(f.path) == 'upload_test.sql'
         assert f.type == 'file'
 
@@ -1093,6 +1097,50 @@ class TestFileSpaces(unittest.TestCase):
             # Cleanup
             space.remove('upload_test.ipynb')
 
+    def test_upload_file_io(self):
+        for space in [self.personal_space, self.shared_space]:
+            root = space.info('/')
+            assert str(root.path) == '/'
+            assert root.type == 'directory'
+
+            # Upload files
+            f = space.upload_file(
+                open(TEST_DIR / 'test.ipynb', 'r'),
+                'upload_test.ipynb',
+            )
+            assert str(f.path) == 'upload_test.ipynb'
+            assert f.type == 'notebook'
+
+            # Download and compare to original
+            txt = f.download(encoding='utf-8')
+            assert txt == open(TEST_DIR / 'test.ipynb').read()
+
+            # Make sure we can't overwrite
+            with self.assertRaises(OSError):
+                space.upload_file(
+                    open(TEST_DIR / 'test.ipynb', 'r'),
+                    'upload_test.ipynb',
+                )
+
+            # Force overwrite with new content
+            f = space.upload_file(
+                open(TEST_DIR / 'test2.ipynb', 'r'),
+                'upload_test.ipynb', overwrite=True,
+            )
+            assert str(f.path) == 'upload_test.ipynb'
+            assert f.type == 'notebook'
+
+            # Verify new content
+            txt = f.download(encoding='utf-8')
+            assert txt == open(TEST_DIR / 'test2.ipynb').read()
+
+            # Make sure we can't upload a folder
+            with self.assertRaises(s2.ManagementError):
+                space.upload_folder(TEST_DIR, 'test')
+
+            # Cleanup
+            space.remove('upload_test.ipynb')
+
     def test_open(self):
         for space in [self.personal_space, self.shared_space]:
             # See if error is raised for non-existent file

{singlestoredb-1.12.0.dist-info → singlestoredb-1.12.1.dist-info}/METADATA

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

{singlestoredb-1.12.0.dist-info → singlestoredb-1.12.1.dist-info}/RECORD

@@ -1,15 +1,15 @@
-_singlestoredb_accel.abi3.so,sha256=WQfunCWdQbFRxYoFX0QjZVDGx_uQ44k40GhykF39_R0,206864
-singlestoredb-1.12.0.dist-info/RECORD,,
-singlestoredb-1.12.0.dist-info/LICENSE,sha256=Mlq78idURT-9G026aMYswwwnnrLcgzTLuXeAs5hjDLM,11341
-singlestoredb-1.12.0.dist-info/WHEEL,sha256=_VEguvlLpUd-c8RbFMA4yMIVNMBv2LhpxYLCEQ-Bogk,113
-singlestoredb-1.12.0.dist-info/entry_points.txt,sha256=bSLaTWB5zGjpVYPAaI46MkkDup0su-eb3uAhCNYuRV0,48
-singlestoredb-1.12.0.dist-info/top_level.txt,sha256=lA65Vf4qAMfg_s1oG3LEO90h4t1Z-SPDbRqkevI3bSY,40
-singlestoredb-1.12.0.dist-info/METADATA,sha256=n-AeePEEJP3mbVTbkAIrZVWLt6eZDNBiWFCJejn-OKQ,5636
+_singlestoredb_accel.abi3.so,sha256=hEcp6tY0Iwmx0nPwBkfSG2Yb9iWV4KUd65FPBatqYGc,206864
 sqlx/magic.py,sha256=JsS9_9aBFaOt91Torm1JPN0c8qB2QmYJmNSKtbSQIY0,3509
 sqlx/__init__.py,sha256=aBYiU8DZXCogvWu3yWafOz7bZS5WWwLZXj7oL0dXGyU,85
+singlestoredb-1.12.1.dist-info/RECORD,,
+singlestoredb-1.12.1.dist-info/LICENSE,sha256=Mlq78idURT-9G026aMYswwwnnrLcgzTLuXeAs5hjDLM,11341
+singlestoredb-1.12.1.dist-info/WHEEL,sha256=_VEguvlLpUd-c8RbFMA4yMIVNMBv2LhpxYLCEQ-Bogk,113
+singlestoredb-1.12.1.dist-info/entry_points.txt,sha256=bSLaTWB5zGjpVYPAaI46MkkDup0su-eb3uAhCNYuRV0,48
+singlestoredb-1.12.1.dist-info/top_level.txt,sha256=lA65Vf4qAMfg_s1oG3LEO90h4t1Z-SPDbRqkevI3bSY,40
+singlestoredb-1.12.1.dist-info/METADATA,sha256=wYTpEPYx7WpKtU6Jr7wQjunmlrA7UnsmNMRkbL73R40,5636
 singlestoredb/auth.py,sha256=u8D9tpKzrqa4ssaHjyZnGDX1q8XBpGtuoOkTkSv7B28,7599
 singlestoredb/config.py,sha256=rlF69SiclYyKghNRckX77Ls1ZT23RhSssO1cyYBiHmA,12589
-singlestoredb/__init__.py,sha256=rMEoyZJpkhiOhcoZLCItPxn7d2AppFdBTIwmK7t_Y7E,1649
+singlestoredb/__init__.py,sha256=4htLkEL7t7jjaZ4Qym7QvHrv4P7cYZdDGtYcMpmSGow,1649
 singlestoredb/types.py,sha256=FIqO1A7e0Gkk7ITmIysBy-P5S--ItbMSlYvblzqGS30,9969
 singlestoredb/connection.py,sha256=0HEpjBZXLqQwOTEfveMkgej1H3Kyof47prIHvJJZtoo,45831
 singlestoredb/pytest.py,sha256=OyF3BO9mgxenifYhOihnzGk8WzCJ_zN5_mxe8XyFPOc,9074
@@ -26,7 +26,7 @@ singlestoredb/fusion/handlers/models.py,sha256=xJPIG0_GgF-VrmPoIsU2U4AsS7ytDz8JM
 singlestoredb/fusion/handlers/job.py,sha256=r0KdOD55VUDw-SymC__5Mn-fzJTZE_xcBgH-O8DYVHc,21095
 singlestoredb/fusion/handlers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 singlestoredb/fusion/handlers/export.py,sha256=3moTJeqsHkDDpitUUAE6x95JYH2rmb28MALbO4x0dcc,8981
-singlestoredb/fusion/handlers/utils.py,sha256=mYRfGMFv2mxZVILeb_B0S55SWsb0gaye8RnjAyylbhE,10064
+singlestoredb/fusion/handlers/utils.py,sha256=dg5v25kCejgqg8smnxE4vqu_G3OJzJ4jX8pm2N1z6mc,10574
 singlestoredb/fusion/handlers/stage.py,sha256=kYVjbPys83kf3jX6jWwN8Ju0oEocKVZ3TIOt2HiC5Ew,14287
 singlestoredb/fusion/handlers/workspace.py,sha256=4xN2TFO4yF7KZB2Fcht7IuvoDdAT6fDfDLjixiHZN8w,27506
 singlestoredb/tests/test.sql,sha256=dfMehVCQ9wObSVTQKyQi-fRFDZeqRxV4Cj8doBCPEFM,17679
@@ -41,7 +41,7 @@ singlestoredb/tests/test2.ipynb,sha256=yd1PE1VK-DwiRd6mYS4_0cPBtuVkvcDtycvTwD-Yn
 singlestoredb/tests/test_ext_func_data.py,sha256=yTADD93nPxX6_rZXXLZaOWEI_yPvYyir9psn5PK9ctU,47695
 singlestoredb/tests/test_exceptions.py,sha256=tfr_8X2w1UmG4nkSBzWGB0C7ehrf1GAVgj6_ODaG-TM,1131
 singlestoredb/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-singlestoredb/tests/test_management.py,sha256=biqwuHfzaH5Jay1yj-nHnX7fhvLB9DPcjujmJR_T1vM,43480
+singlestoredb/tests/test_management.py,sha256=9WKaFChAWkKMm0kDhwTgjaJIAcga7mQZiF0x9BNI1Tc,45112
 singlestoredb/tests/test_udf.py,sha256=G6MgAzu5ZiMHmtHaGbWYXRZ-naEvwYzT5MRB900BU3I,30029
 singlestoredb/tests/test_http.py,sha256=RXasTqBWRn__omj0eLFTJYIbZjd0PPdIV2d4Cqz0MC8,8580
 singlestoredb/tests/utils.py,sha256=2A2tEdD3t8aXWUnHtAIcFlWrflsz2MlMcCbUDaAG29c,4995
@@ -56,7 +56,7 @@ singlestoredb/tests/ext_funcs/__init__.py,sha256=qZLnDI_Ck0tguVi-K-BKXDHAcC0jui3
 singlestoredb/magics/__init__.py,sha256=lZjkT3Webo9c1EQAzlRCRh6B2pckQH8uvNrrB__abcI,1210
 singlestoredb/magics/run_shared.py,sha256=SI8dCBRMaGn-xZU7dto4jsAqKBi-Ll14htUsMUSBpJM,1752
 singlestoredb/magics/run_personal.py,sha256=2f7u1T7iblxGzZurHNgNXLrPBvsvPADZKo_RD_IjYuE,1844
-singlestoredb/management/files.py,sha256=0sp3UDDHbiY5lP0QYkNgVcZ2v9zz3Kba3CeLU_xtE4Y,30575
+singlestoredb/management/files.py,sha256=_FDIs9rtG7cW6aIkXnYReLQaFxdix2iu7Do368hOoE8,30476
 singlestoredb/management/organization.py,sha256=hqMaM7H-naMjNbxDl_f7G_2o5TkiGKyzPhxuzDveJAw,5402
 singlestoredb/management/job.py,sha256=4-xLWzbE8odQogVVaFer80UEoTAZY1T28VZ9Ug4rbmM,24611
 singlestoredb/management/region.py,sha256=HnLcWUh7r_aLECliplCDHak4a_F3B7LOSXEYMW66qD0,1611
@@ -64,7 +64,7 @@ singlestoredb/management/__init__.py,sha256=ofNTPCdkZ1dS_aX2aUujd8aMHQi8Lle5Ced0
 singlestoredb/management/export.py,sha256=jJCe25ecH_LzKSDc7vS1-5DQaWFrZipeawLPpArByJE,5108
 singlestoredb/management/utils.py,sha256=P4fp8a7EwaYiag_hvpILcgwXtdFNYKKO70dsKjmxn1A,13171
 singlestoredb/management/cluster.py,sha256=h75grXSxq4Anr4RxwKxcZW4TkWJ4bFg_ql5iRWCNLdQ,14405
-singlestoredb/management/workspace.py,sha256=fNUiz3XNTGgXdOACsQz56Gox2qt9lOVGAtHK67IpMuc,56319
+singlestoredb/management/workspace.py,sha256=ze-eE-cO3JCrR3uttVFaBOndDbEE8_qWR2kzOjzbKaY,56234
 singlestoredb/management/manager.py,sha256=X29VEHlUEzmWvGo_bQMzo8a6d4nYMLE1CewlNBjrD7M,8851
 singlestoredb/management/billing_usage.py,sha256=9ighjIpcopgIyJOktBYQ6pahBZmWGHOPyyCW4gu9FGs,3735
 singlestoredb/server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0