singlestoredb 1.14.0__cp38-abi3-win32.whl → 1.14.1__cp38-abi3-win32.whl

singlestoredb/__init__.py

@@ -13,7 +13,7 @@ Examples
 
 """
 
-__version__ = '1.14.0'
+__version__ = '1.14.1'
 
 from typing import Any
 

singlestoredb/fusion/handlers/export.py

@@ -1,10 +1,12 @@
 #!/usr/bin/env python3
+import datetime
 import json
 from typing import Any
 from typing import Dict
 from typing import Optional
 
 from .. import result
+from ...management.export import _get_exports
 from ...management.export import ExportService
 from ...management.export import ExportStatus
 from ..handler import SQLHandler
@@ -104,7 +106,100 @@ class CreateClusterIdentity(SQLHandler):
 CreateClusterIdentity.register(overwrite=True)
 
 
-class CreateExport(SQLHandler):
+def _start_export(params: Dict[str, Any]) -> Optional[FusionSQLResult]:
+    # From table
+    if isinstance(params['from_table'], str):
+        from_database = None
+        from_table = params['from_table']
+    else:
+        from_database, from_table = params['from_table']
+
+    # Catalog
+    catalog_config = json.loads(params['catalog'].get('catalog_config', '{}') or '{}')
+    catalog_creds = json.loads(params['catalog'].get('catalog_creds', '{}') or '{}')
+
+    # Storage
+    storage_config = json.loads(params['storage'].get('link_config', '{}') or '{}')
+    storage_creds = json.loads(params['storage'].get('link_creds', '{}') or '{}')
+
+    storage_config['provider'] = 'S3'
+
+    wsg = get_workspace_group({})
+
+    if from_database is None:
+        raise ValueError('database name must be specified for source table')
+
+    if wsg._manager is None:
+        raise TypeError('no workspace manager is associated with workspace group')
+
+    partition_by = []
+    if params['partition_by']:
+        for key in params['partition_by']:
+            transform = key['partition_key']['transform']['col_transform']
+            part = {}
+            part['transform'] = transform[0].lower()
+            part['name'] = transform[-1]['transform_col']
+            partition_by.append(part)
+
+    order_by = []
+    if params['order_by'] and params['order_by']['by']:
+        for key in params['order_by']['by']:
+            transform = key['transform']['col_transform']
+            order = {}
+            order['transform'] = transform[0].lower()
+            order['name'] = transform[-1]['transform_col']
+            order['direction'] = 'ascending'
+            order['null_order'] = 'nulls_first'
+            if key.get('direction'):
+                if 'desc' in key['direction'].lower():
+                    order['direction'] = 'descending'
+            if key.get('null_order'):
+                if 'last' in key['null_order'].lower():
+                    order['null_order'] = 'nulls_last'
+            order_by.append(order)
+
+    # Refresh interval
+    refresh_interval_delta = None
+    refresh_interval = params.get('refresh_interval', None)
+    if refresh_interval is not None:
+        value = int(refresh_interval['refresh_interval_value'])
+        time_unit = refresh_interval['refresh_interval_time_unit'].upper()
+        if value < 0:
+            raise ValueError('refresh interval must be greater than 0')
+        if time_unit == 'SECONDS':
+            refresh_interval_delta = datetime.timedelta(seconds=int(value))
+        elif time_unit == 'MINUTES':
+            refresh_interval_delta = datetime.timedelta(minutes=int(value))
+        elif time_unit == 'HOURS':
+            refresh_interval_delta = datetime.timedelta(hours=int(value))
+        elif time_unit == 'DAYS':
+            refresh_interval_delta = datetime.timedelta(days=int(value))
+        else:
+            raise ValueError('invalid refresh interval time unit')
+
+    out = ExportService(
+        wsg,
+        from_database,
+        from_table,
+        dict(**catalog_config, **catalog_creds),
+        dict(**storage_config, **storage_creds),
+        columns=None,
+        partition_by=partition_by or None,
+        order_by=order_by or None,
+        properties=json.loads(params['properties']) if params['properties'] else None,
+        incremental=params.get('incremental', False),
+        refresh_interval=int(refresh_interval_delta.total_seconds())
+        if refresh_interval_delta is not None else None,
+    ).start()
+
+    res = FusionSQLResult()
+    res.add_field('ExportID', result.STRING)
+    res.set_rows([(out.export_id,)])
+
+    return res
+
+
+class StartExport(SQLHandler):
     """
     START EXPORT
         from_table
@@ -150,7 +245,7 @@ class CreateExport(SQLHandler):
 
     Description
     -----------
-    Create an export configuration.
+    Start an export.
 
     Arguments
     ---------
@@ -180,7 +275,6 @@ class CreateExport(SQLHandler):
             LINK S3 CONFIG '{
                 "region": "us-east-1",
                 "endpoint_url": "s3://bucket-name"
-
             }'
         ;
 
@@ -189,77 +283,129 @@ class CreateExport(SQLHandler):
     _enabled = False
 
     def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
-        # From table
-        if isinstance(params['from_table'], str):
-            from_database = None
-            from_table = params['from_table']
-        else:
-            from_database, from_table = params['from_table']
+        return _start_export(params)
 
-        # Catalog
-        catalog_config = json.loads(params['catalog'].get('catalog_config', '{}') or '{}')
-        catalog_creds = json.loads(params['catalog'].get('catalog_creds', '{}') or '{}')
 
-        # Storage
-        storage_config = json.loads(params['storage'].get('link_config', '{}') or '{}')
-        storage_creds = json.loads(params['storage'].get('link_creds', '{}') or '{}')
+StartExport.register(overwrite=True)
 
-        storage_config['provider'] = 'S3'
 
-        wsg = get_workspace_group({})
+class StartIncrementalExport(SQLHandler):
+    """
+    START INCREMENTAL EXPORT
+        from_table
+        catalog
+        storage
+        [ partition_by ]
+        [ order_by ]
+        [ properties ]
+        [ refresh_interval ]
+    ;
 
-        if from_database is None:
-            raise ValueError('database name must be specified for source table')
+    # From table
+    from_table = FROM <table>
 
-        if wsg._manager is None:
-            raise TypeError('no workspace manager is associated with workspace group')
+    # Transforms
+    _col_transform = { VOID | IDENTITY | YEAR | MONTH | DAY | HOUR } ( _transform_col )
+    _transform_col = <column>
+    _arg_transform = { BUCKET | TRUNCATE } ( _transform_col <comma> _transform_arg )
+    _transform_arg = <integer>
+    transform = { _col_transform | _arg_transform }
 
-        partition_by = []
-        if params['partition_by']:
-            for key in params['partition_by']:
-                transform = key['partition_key']['transform']['col_transform']
-                part = {}
-                part['transform'] = transform[0].lower()
-                part['name'] = transform[-1]['transform_col']
-                partition_by.append(part)
-
-        order_by = []
-        if params['order_by'] and params['order_by']['by']:
-            for key in params['order_by']['by']:
-                transform = key['transform']['col_transform']
-                order = {}
-                order['transform'] = transform[0].lower()
-                order['name'] = transform[-1]['transform_col']
-                order['direction'] = 'ascending'
-                order['null_order'] = 'nulls_first'
-                if key.get('direction'):
-                    if 'desc' in key['direction'].lower():
-                        order['direction'] = 'descending'
-                if key.get('null_order'):
-                    if 'last' in key['null_order'].lower():
-                        order['null_order'] = 'nulls_last'
-                order_by.append(order)
+    # Partitions
+    partition_by = PARTITION BY partition_key,...
+    partition_key = transform
 
-        out = ExportService(
-            wsg,
-            from_database,
-            from_table,
-            dict(**catalog_config, **catalog_creds),
-            dict(**storage_config, **storage_creds),
-            columns=None,
-            partition_by=partition_by or None,
-            order_by=order_by or None,
-            properties=json.loads(params['properties']) if params['properties'] else None,
-        ).start()
+    # Sort order
+    order_by = ORDER BY sort_key,...
+    sort_key = transform [ direction ] [ null_order ]
+    direction = { ASC | DESC | ASCENDING | DESCENDING }
+    null_order = { NULLS_FIRST | NULLS_LAST }
 
-        res = FusionSQLResult()
-        res.add_field('ExportID', result.STRING)
-        res.set_rows([(out.export_id,)])
+    # Properties
+    properties = PROPERTIES '<json>'
 
-        return res
+    # Catolog
+    catalog = CATALOG [ _catalog_config ] [ _catalog_creds ]
+    _catalog_config = CONFIG '<catalog-config>'
+    _catalog_creds = CREDENTIALS '<catalog-creds>'
+
+    # Storage
+    storage = LINK [ _link_config ] [ _link_creds ]
+    _link_config = S3 CONFIG '<link-config>'
+    _link_creds = CREDENTIALS '<link-creds>'
+
+    # Refresh interval
+    refresh_interval = REFRESH INTERVAL _refresh_interval_value _refresh_interval_time_unit
+    _refresh_interval_value = <integer>
+    _refresh_interval_time_unit = { SECONDS | MINUTES | HOURS | DAYS }
+
+    Description
+    -----------
+    Start an incremental export.
+
+    Arguments
+    ---------
+    * ``<catalog-config>`` and ``<catalog-creds>``: The catalog configuration.
+    * ``<link-config>`` and ``<link-creds>``: The storage link configuration.
+
+    Remarks
+    -------
+    * ``FROM <table>`` specifies the SingleStore table to export. The same name will
+      be used for the exported table.
+    * ``CATALOG`` specifies the details of the catalog to connect to.
+    * ``LINK`` specifies the details of the data storage to connect to.
+    * ``REFRESH INTERVAL`` specifies the interval for refreshing the
+      incremental export. The default is 1 day.
+
+    Examples
+    --------
+    The following statement starts an export operation with the given
+    catalog and link configurations. The source table to export is
+    named "customer_data"::
 
+        START INCREMENTAL EXPORT FROM my_db.customer_data
+            CATALOG CONFIG '{
+                "catalog_type": "GLUE",
+                "table_format": "ICEBERG",
+                "catalog_id": "13983498723498",
+                "catalog_region": "us-east-1"
+            }'
+            LINK S3 CONFIG '{
+                "region": "us-east-1",
+                "endpoint_url": "s3://bucket-name"
+            }'
+            REFRESH INTERVAL 24 HOURS
+        ;
+
+    """  # noqa
+
+    _enabled = False
+
+    def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
+        params['incremental'] = True
+        return _start_export(params)
+
+
+StartIncrementalExport.register(overwrite=True)
+
+
+def _format_status(export_id: str, status: ExportStatus) -> Optional[FusionSQLResult]:
+    """Return the status of an export operation."""
+    info = status._info()
 
-CreateExport.register(overwrite=True)
+    res = FusionSQLResult()
+    res.add_field('ExportID', result.STRING)
+    res.add_field('Status', result.STRING)
+    res.add_field('Message', result.STRING)
+    res.set_rows([
+        (
+            export_id,
+            info.get('status', 'Unknown'),
+            info.get('statusMsg', ''),
+        ),
+    ])
+
+    return res
 
 
 class ShowExport(SQLHandler):
@@ -275,9 +421,29 @@ class ShowExport(SQLHandler):
 
     def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
         wsg = get_workspace_group({})
-        out = ExportStatus(params['export_id'], wsg)
+        return _format_status(
+            params['export_id'], ExportStatus(params['export_id'], wsg),
+        )
+
+
+ShowExport.register(overwrite=True)
+
+
+class ShowExports(SQLHandler):
+    """
+    SHOW EXPORTS [ scope ];
+
+    # Location of the export
+    scope = FOR '<scope>'
+
+    """
+
+    _enabled = False
+
+    def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
+        wsg = get_workspace_group({})
 
-        status = out._info()
+        exports = _get_exports(wsg, params.get('scope', 'all'))
 
         res = FusionSQLResult()
         res.add_field('ExportID', result.STRING)
@@ -285,13 +451,75 @@ class ShowExport(SQLHandler):
         res.add_field('Message', result.STRING)
         res.set_rows([
             (
-                params['export_id'],
-                status.get('status', 'Unknown'),
-                status.get('statusMsg', ''),
-            ),
+                info['egressID'],
+                info.get('status', 'Unknown'),
+                info.get('statusMsg', ''),
+            )
+            for info in [x._info() for x in exports]
         ])
 
         return res
 
 
-ShowExport.register(overwrite=True)
+ShowExports.register(overwrite=True)
+
+
+class SuspendExport(SQLHandler):
+    """
+    SUSPEND EXPORT export_id;
+
+    # ID of export
+    export_id = '<export-id>'
+
+    """
+
+    _enabled = False
+
+    def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
+        wsg = get_workspace_group({})
+        service = ExportService.from_export_id(wsg, params['export_id'])
+        return _format_status(params['export_id'], service.suspend())
+
+
+SuspendExport.register(overwrite=True)
+
+
+class ResumeExport(SQLHandler):
+    """
+    RESUME EXPORT export_id;
+
+    # ID of export
+    export_id = '<export-id>'
+
+    """
+
+    _enabled = False
+
+    def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
+        wsg = get_workspace_group({})
+        service = ExportService.from_export_id(wsg, params['export_id'])
+        return _format_status(params['export_id'], service.resume())
+
+
+ResumeExport.register(overwrite=True)
+
+
+class DropExport(SQLHandler):
+    """
+    DROP EXPORT export_id;
+
+    # ID of export
+    export_id = '<export-id>'
+
+    """
+
+    _enabled = False
+
+    def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
+        wsg = get_workspace_group({})
+        service = ExportService.from_export_id(wsg, params['export_id'])
+        service.drop()
+        return None
+
+
+DropExport.register(overwrite=True)

singlestoredb/http/connection.py

@@ -569,8 +569,10 @@ class Cursor(connection.Cursor):
 
         if res.status_code >= 400:
             if res.text:
-                if re.match(r'^Error\s+\d+:', res.text):
-                    code, msg = res.text.split(':', 1)
+                m = re.match(r'^Error\s+(\d+).*?:', res.text)
+                if m:
+                    code = m.group(1)
+                    msg = res.text.split(':', 1)[-1]
                     icode = int(code.split()[-1])
                 else:
                     icode = res.status_code

singlestoredb/management/export.py

@@ -27,6 +27,9 @@ class ExportService(object):
     partition_by: Optional[List[Dict[str, str]]]
     order_by: Optional[List[Dict[str, Dict[str, str]]]]
     properties: Optional[Dict[str, Any]]
+    incremental: bool
+    refresh_interval: Optional[int]
+    export_id: Optional[str]
 
     def __init__(
         self,
@@ -38,6 +41,8 @@ class ExportService(object):
         columns: Optional[List[str]] = None,
         partition_by: Optional[List[Dict[str, str]]] = None,
         order_by: Optional[List[Dict[str, Dict[str, str]]]] = None,
+        incremental: bool = False,
+        refresh_interval: Optional[int] = None,
         properties: Optional[Dict[str, Any]] = None,
     ):
         #: Workspace group
@@ -68,8 +73,30 @@ class ExportService(object):
         self.order_by = order_by or None
         self.properties = properties or None
 
+        self.incremental = incremental
+        self.refresh_interval = refresh_interval
+
+        self.export_id = None
+
         self._manager: Optional[WorkspaceManager] = workspace_group._manager
 
+    @classmethod
+    def from_export_id(
+        self,
+        workspace_group: WorkspaceGroup,
+        export_id: str,
+    ) -> ExportService:
+        """Create export service from export ID."""
+        out = ExportService(
+            workspace_group=workspace_group,
+            database='',
+            table='',
+            catalog_info={},
+            storage_info={},
+        )
+        out.export_id = export_id
+        return out
+
     def __str__(self) -> str:
         """Return string representation."""
         return vars_to_str(self)
@@ -98,6 +125,11 @@ class ExportService(object):
 
     def start(self, tags: Optional[List[str]] = None) -> 'ExportStatus':
         """Start the export process."""
+        if not self.table or not self.database:
+            raise ManagementError(
+                msg='Database and table must be set before starting the export.',
+            )
+
         if self._manager is None:
             raise ManagementError(
                 msg='No workspace manager is associated with this object.',
@@ -122,11 +154,87 @@ class ExportService(object):
                     partitionSpec=partition_spec,
                     sortOrderSpec=sort_order_spec,
                     properties=self.properties,
+                    incremental=self.incremental or None,
+                    refreshInterval=self.refresh_interval
+                    if self.refresh_interval is not None else None,
                 ).items() if v is not None
             },
         )
 
-        return ExportStatus(out.json()['egressID'], self.workspace_group)
+        self.export_id = str(out.json()['egressID'])
+
+        return ExportStatus(self.export_id, self.workspace_group)
+
+    def suspend(self) -> 'ExportStatus':
+        """Suspend the export process."""
+        if self._manager is None:
+            raise ManagementError(
+                msg='No workspace manager is associated with this object.',
+            )
+
+        if self.export_id is None:
+            raise ManagementError(
+                msg='Export ID is not set. You must start the export first.',
+            )
+
+        self._manager._post(
+            f'workspaceGroups/{self.workspace_group.id}/egress/suspendTableEgress',
+            json=dict(egressID=self.export_id),
+        )
+
+        return ExportStatus(self.export_id, self.workspace_group)
+
+    def resume(self) -> 'ExportStatus':
+        """Resume the export process."""
+        if self._manager is None:
+            raise ManagementError(
+                msg='No workspace manager is associated with this object.',
+            )
+
+        if self.export_id is None:
+            raise ManagementError(
+                msg='Export ID is not set. You must start the export first.',
+            )
+
+        self._manager._post(
+            f'workspaceGroups/{self.workspace_group.id}/egress/resumeTableEgress',
+            json=dict(egressID=self.export_id),
+        )
+
+        return ExportStatus(self.export_id, self.workspace_group)
+
+    def drop(self) -> None:
+        """Drop the export process."""
+        if self._manager is None:
+            raise ManagementError(
+                msg='No workspace manager is associated with this object.',
+            )
+
+        if self.export_id is None:
+            raise ManagementError(
+                msg='Export ID is not set. You must start the export first.',
+            )
+
+        self._manager._post(
+            f'workspaceGroups/{self.workspace_group.id}/egress/dropTableEgress',
+            json=dict(egressID=self.export_id),
+        )
+
+        return None
+
+    def status(self) -> ExportStatus:
+        """Get the status of the export process."""
+        if self._manager is None:
+            raise ManagementError(
+                msg='No workspace manager is associated with this object.',
+            )
+
+        if self.export_id is None:
+            raise ManagementError(
+                msg='Export ID is not set. You must start the export first.',
+            )
+
+        return ExportStatus(self.export_id, self.workspace_group)
 
 
 class ExportStatus(object):
@@ -167,3 +275,21 @@ class ExportStatus(object):
 
     def __repr__(self) -> str:
         return self.status
+
+
+def _get_exports(
+    workspace_group: WorkspaceGroup,
+    scope: str = 'all',
+) -> List[ExportStatus]:
+    """Get all exports in the workspace group."""
+    if workspace_group._manager is None:
+        raise ManagementError(
+            msg='No workspace manager is associated with this object.',
+        )
+
+    out = workspace_group._manager._get(
+        f'workspaceGroups/{workspace_group.id}/egress/tableEgressStatus',
+        json=dict(scope=scope),
+    )
+
+    return out.json()

singlestoredb/tests/test_management.py

@@ -450,55 +450,59 @@ class TestStage(unittest.TestCase):
     def test_open(self):
         st = self.wg.stage
 
+        open_test_sql = f'open_test_{id(self)}.sql'
+
         # See if error is raised for non-existent file
         with self.assertRaises(s2.ManagementError):
-            st.open('open_test.sql', 'r')
+            st.open(open_test_sql, 'r')
 
         # Load test file
-        st.upload_file(TEST_DIR / 'test.sql', 'open_test.sql')
+        st.upload_file(TEST_DIR / 'test.sql', open_test_sql)
 
         # Read file using `open`
-        with st.open('open_test.sql', 'r') as rfile:
+        with st.open(open_test_sql, 'r') as rfile:
             assert rfile.read() == open(TEST_DIR / 'test.sql').read()
 
         # Read file using `open` with 'rt' mode
-        with st.open('open_test.sql', 'rt') as rfile:
+        with st.open(open_test_sql, 'rt') as rfile:
             assert rfile.read() == open(TEST_DIR / 'test.sql').read()
 
         # Read file using `open` with 'rb' mode
-        with st.open('open_test.sql', 'rb') as rfile:
+        with st.open(open_test_sql, 'rb') as rfile:
             assert rfile.read() == open(TEST_DIR / 'test.sql', 'rb').read()
 
         # Read file using `open` with 'rb' mode
         with self.assertRaises(ValueError):
-            with st.open('open_test.sql', 'b') as rfile:
+            with st.open(open_test_sql, 'b') as rfile:
                 pass
 
         # Attempt overwrite file using `open` with mode 'x'
         with self.assertRaises(OSError):
-            with st.open('open_test.sql', 'x') as wfile:
+            with st.open(open_test_sql, 'x') as wfile:
                 pass
 
         # Attempt overwrite file using `open` with mode 'w'
-        with st.open('open_test.sql', 'w') as wfile:
+        with st.open(open_test_sql, 'w') as wfile:
             wfile.write(open(TEST_DIR / 'test2.sql').read())
 
-        txt = st.download_file('open_test.sql', encoding='utf-8')
+        txt = st.download_file(open_test_sql, encoding='utf-8')
 
         assert txt == open(TEST_DIR / 'test2.sql').read()
 
+        open_raw_test_sql = f'open_raw_test_{id(self)}.sql'
+
         # Test writer without context manager
-        wfile = st.open('open_raw_test.sql', 'w')
+        wfile = st.open(open_raw_test_sql, 'w')
         for line in open(TEST_DIR / 'test.sql'):
             wfile.write(line)
         wfile.close()
 
-        txt = st.download_file('open_raw_test.sql', encoding='utf-8')
+        txt = st.download_file(open_raw_test_sql, encoding='utf-8')
 
         assert txt == open(TEST_DIR / 'test.sql').read()
 
         # Test reader without context manager
-        rfile = st.open('open_raw_test.sql', 'r')
+        rfile = st.open(open_raw_test_sql, 'r')
         txt = ''
         for line in rfile:
             txt += line
@@ -509,15 +513,18 @@ class TestStage(unittest.TestCase):
     def test_obj_open(self):
         st = self.wg.stage
 
+        obj_open_test_sql = f'obj_open_test_{id(self)}.sql'
+        obj_open_dir = f'obj_open_dir_{id(self)}'
+
         # Load test file
-        f = st.upload_file(TEST_DIR / 'test.sql', 'obj_open_test.sql')
+        f = st.upload_file(TEST_DIR / 'test.sql', obj_open_test_sql)
 
         # Read file using `open`
         with f.open() as rfile:
             assert rfile.read() == open(TEST_DIR / 'test.sql').read()
 
         # Make sure directories error out
-        d = st.mkdir('obj_open_dir')
+        d = st.mkdir(obj_open_dir)
         with self.assertRaises(IsADirectoryError):
             d.open()
 
@@ -1143,58 +1150,62 @@ class TestFileSpaces(unittest.TestCase):
 
     def test_open(self):
         for space in [self.personal_space, self.shared_space]:
+            open_test_ipynb = f'open_test_ipynb_{id(self)}.ipynb'
+
             # See if error is raised for non-existent file
             with self.assertRaises(s2.ManagementError):
-                space.open('open_test.ipynb', 'r')
+                space.open(open_test_ipynb, 'r')
 
             # Load test file
-            space.upload_file(TEST_DIR / 'test.ipynb', 'open_test.ipynb')
+            space.upload_file(TEST_DIR / 'test.ipynb', open_test_ipynb)
 
             # Read file using `open`
-            with space.open('open_test.ipynb', 'r') as rfile:
+            with space.open(open_test_ipynb, 'r') as rfile:
                 assert rfile.read() == open(TEST_DIR / 'test.ipynb').read()
 
             # Read file using `open` with 'rt' mode
-            with space.open('open_test.ipynb', 'rt') as rfile:
+            with space.open(open_test_ipynb, 'rt') as rfile:
                 assert rfile.read() == open(TEST_DIR / 'test.ipynb').read()
 
             # Read file using `open` with 'rb' mode
-            with space.open('open_test.ipynb', 'rb') as rfile:
+            with space.open(open_test_ipynb, 'rb') as rfile:
                 assert rfile.read() == open(TEST_DIR / 'test.ipynb', 'rb').read()
 
             # Read file using `open` with 'rb' mode
             with self.assertRaises(ValueError):
-                with space.open('open_test.ipynb', 'b') as rfile:
+                with space.open(open_test_ipynb, 'b') as rfile:
                     pass
 
             # Attempt overwrite file using `open` with mode 'x'
             with self.assertRaises(OSError):
-                with space.open('open_test.ipynb', 'x') as wfile:
+                with space.open(open_test_ipynb, 'x') as wfile:
                     pass
 
             # Attempt overwrite file using `open` with mode 'w'
-            with space.open('open_test.ipynb', 'w') as wfile:
+            with space.open(open_test_ipynb, 'w') as wfile:
                 wfile.write(open(TEST_DIR / 'test2.ipynb').read())
 
-            txt = space.download_file('open_test.ipynb', encoding='utf-8')
+            txt = space.download_file(open_test_ipynb, encoding='utf-8')
 
             assert txt == open(TEST_DIR / 'test2.ipynb').read()
 
+            open_raw_test_ipynb = f'open_raw_test_{id(self)}.ipynb'
+
             # Test writer without context manager
-            wfile = space.open('open_raw_test.ipynb', 'w')
+            wfile = space.open(open_raw_test_ipynb, 'w')
             for line in open(TEST_DIR / 'test.ipynb'):
                 wfile.write(line)
             wfile.close()
 
             txt = space.download_file(
-                'open_raw_test.ipynb',
+                open_raw_test_ipynb,
                 encoding='utf-8',
             )
 
             assert txt == open(TEST_DIR / 'test.ipynb').read()
 
             # Test reader without context manager
-            rfile = space.open('open_raw_test.ipynb', 'r')
+            rfile = space.open(open_raw_test_ipynb, 'r')
             txt = ''
             for line in rfile:
                 txt += line
@@ -1203,15 +1214,18 @@ class TestFileSpaces(unittest.TestCase):
             assert txt == open(TEST_DIR / 'test.ipynb').read()
 
             # Cleanup
-            space.remove('open_test.ipynb')
-            space.remove('open_raw_test.ipynb')
+            space.remove(open_test_ipynb)
+            space.remove(open_raw_test_ipynb)
 
     def test_obj_open(self):
         for space in [self.personal_space, self.shared_space]:
+            obj_open_test_ipynb = f'obj_open_test_{id(self)}.ipynb'
+            obj_open_dir = f'obj_open_dir_{id(self)}'
+
             # Load test file
             f = space.upload_file(
                 TEST_DIR / 'test.ipynb',
-                'obj_open_test.ipynb',
+                obj_open_test_ipynb,
             )
 
             # Read file using `open`
@@ -1220,7 +1234,7 @@ class TestFileSpaces(unittest.TestCase):
 
             # Make sure directories error out
             with self.assertRaises(s2.ManagementError):
-                space.mkdir('obj_open_dir')
+                space.mkdir(obj_open_dir)
 
             # Write file using `open`
             with f.open('w', encoding='utf-8') as wfile:
@@ -1248,7 +1262,7 @@ class TestFileSpaces(unittest.TestCase):
             assert txt == open(TEST_DIR / 'test.ipynb').read()
 
             # Cleanup
-            space.remove('obj_open_test.ipynb')
+            space.remove(obj_open_test_ipynb)
 
     def test_os_directories(self):
         for space in [self.personal_space, self.shared_space]:

singlestoredb/types.py

@@ -173,17 +173,25 @@ class ColumnType(object):
         'DECIMAL', 'DEC', 'FIXED', 'NUMERIC', 0, decimal.Decimal,
     )
     DEC = FIXED = NUMERIC = DECIMAL
-    TINY = TINYINT = BOOL = BOOLEAN = NumberDBAPIType(
-        'TINY', 'TINYINT', 'BOOL', 'BOOLEAN', 1,
+    TINY = TINYINT = BOOL = BOOLEAN = UNSIGNED_TINY = UNSIGNED_TINYINT = NumberDBAPIType(
+        'TINY', 'TINYINT', 'BOOL', 'BOOLEAN', 'UNSIGNED TINY', 'UNSIGNED TINYINT', 1,
+    )
+    SHORT = SMALLINT = UNSIGNED_SHORT = UNSIGNED_SMALLINT = NumberDBAPIType(
+        'SMALLINT', 'SHORT', 'UNSIGNED SHORT', 'UNSIGNED SMALLINT', 2,
+    )
+    LONG = INT = UNSIGNED_LONG = UNSIGNED_INT = NumberDBAPIType(
+        'LONG', 'INT', 'UNSIGNED LONG', 'UNSIGNED INT', 3,
     )
-    SHORT = SMALLINT = NumberDBAPIType('SMALLINT', 'SHORT', 2)
-    LONG = INT = NumberDBAPIType('LONG', 'INT', 3)
     FLOAT = NumberDBAPIType('FLOAT', 4)
     DOUBLE = REAL = NumberDBAPIType('DOUBLE', 5, float)
     NULL = DBAPIType('NULL', 6)
     TIMESTAMP = DatetimeDBAPIType('TIMESTAMP', 7)
-    LONGLONG = BIGINT = NumberDBAPIType('BIGINT', 'LONGLONG', 8, int)
-    MEDIUMINT = INT24 = NumberDBAPIType('MEDIUMINT', 'INT24', 9)
+    LONGLONG = BIGINT = UNSIGNED_LONGLONG = UNSIGNED_BIGINT = NumberDBAPIType(
+        'BIGINT', 'LONGLONG', 'UNSIGNED LONGLONG', 'UNSIGNED BIGINT', 8, int,
+    )
+    MEDIUMINT = INT24 = UNSIGNED_MEDIUMINT = UNSIGNED_INT24 = NumberDBAPIType(
+        'MEDIUMINT', 'INT24', 'UNSIGNED MEDIUMINT', 'UNSIGNED INT24', 9,
+    )
     DATE = DBAPIType('DATE', 10, datetime.date)
     TIME = DBAPIType('TIME', 11, datetime.time)
     DATETIME = DatetimeDBAPIType('DATETIME', 12, datetime.datetime)

{singlestoredb-1.14.0.dist-info → singlestoredb-1.14.1.dist-info}/METADATA

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

{singlestoredb-1.14.0.dist-info → singlestoredb-1.14.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
-_singlestoredb_accel.pyd,sha256=0WgZ4A3-2pU8ZtZ5AUnpQOuHTeORtweaAr1qNDSJ2XM,62464
-singlestoredb/__init__.py,sha256=GNnvb4MNLXKVSKSIAj7TseHmVPIMY7O2Qy37yI3OwL0,2162
+_singlestoredb_accel.pyd,sha256=44TC8rZGqJ2E373qMsrMCXTy3HvMz65Vh-x52kQtmDY,62464
+singlestoredb/__init__.py,sha256=S78W6cTcUSwZR1BTOayQSnIGbOgnKs4Tmw_VSxYqgco,2162
 singlestoredb/auth.py,sha256=RmYiH0Wlc2RXc4pTlRMysxtBI445ggCIwojWKC_eDLE,7844
 singlestoredb/config.py,sha256=t3aiWi1i3kT5VhEgXca0gwT6591YkZUed-wzvVEBMs0,13424
 singlestoredb/connection.py,sha256=I2AP_0l7hNARfXiSuVW953CsGYn_rKbTg_NyWEiGHbY,47542
@@ -7,7 +7,7 @@ singlestoredb/converters.py,sha256=6gN3_RzSbw0Aimd5cGgBNPNq1yiHb1a_NK8qC9DmOQ0,2
 singlestoredb/exceptions.py,sha256=WCCJrNSsU-hD-621Jpd6bwmvGftQ7byXkk-XKXlaxpg,3354
 singlestoredb/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 singlestoredb/pytest.py,sha256=TH364xRCN7_QaN0oRQDHixrEcDx_ZBgu3bmY0tvKrYU,9357
-singlestoredb/types.py,sha256=Lv0BEQl6aSZBiAe0OSI07FEJhcHZ9HX45iT9NU_mxHQ,10334
+singlestoredb/types.py,sha256=g6iJnOSCuRUkuUJOYSdRPt3QTjC9h2Dq4fqFFktXxXg,10770
 singlestoredb/vectorstore.py,sha256=4YvXml3PpOEOtUGO7gylucKG2Rny8Bx6L29kmhsFiCY,8600
 singlestoredb/ai/__init__.py,sha256=5vlx0XpzxalMKySnVF7y40gfuCgaz7COUKqN4KfNKF8,116
 singlestoredb/ai/chat.py,sha256=oDig8C8QdPEHL-JmmpdFvt_Ct7-K_D0pG_UJ00WCZ7Y,828
@@ -41,7 +41,7 @@ singlestoredb/fusion/handler.py,sha256=ohnU0BIoJ9AHrVLlCHI-3E4Icqoocxqip8T-XyYxB
 singlestoredb/fusion/registry.py,sha256=_eT1gd38VPlFKs5f9Pu6lqQyoDQ_ixW5O56QwYLQ89Y,6361
 singlestoredb/fusion/result.py,sha256=KAwhXxXVgfkAWekCFY8-Y03ANKDiTflYRXyEc_1Id0k,12189
 singlestoredb/fusion/handlers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-singlestoredb/fusion/handlers/export.py,sha256=OPA1yY1GRRTh1YiGawfAx_LfsxA_qs8Ci0LDTfVEq9A,9278
+singlestoredb/fusion/handlers/export.py,sha256=MqPINMHGl-7SkKzdHcgg343uoFQDXaKSHdoFmefA-KM,15834
 singlestoredb/fusion/handlers/files.py,sha256=pCx1sqnjPtQrp39rv_V4RX9CVtj6uSiL6HPUyiABYpI,19681
 singlestoredb/fusion/handlers/job.py,sha256=3enfxHwERH7T4u0FEwOPN0IL0GtepaCYgEsisiy3Df4,21753
 singlestoredb/fusion/handlers/models.py,sha256=XWaPJQc3GQIOAcjNcxBSGUBJ3xu2qkzQ4ILa40TFQmY,6486
@@ -49,14 +49,14 @@ singlestoredb/fusion/handlers/stage.py,sha256=PP-SSP204lwpmnycSXXSmFPzoN535JVuwg
 singlestoredb/fusion/handlers/utils.py,sha256=nV2lSzKhv7CzM7I_uIh5kmDV0Ec6VeeKoHczx5pVNcw,11009
 singlestoredb/fusion/handlers/workspace.py,sha256=NxoEY5xd5lCQmXiim4nhAYCL0agHo1H_rGPpqa31hiw,28397
 singlestoredb/http/__init__.py,sha256=4cEDvLloGc3LSpU-PnIwacyu0n5oIIIE6xk2SPyWD_w,939
-singlestoredb/http/connection.py,sha256=pYewBJsoMLVNw5HMtSECQWHwPQ1qf6onwz4BmQ01_TU,41156
+singlestoredb/http/connection.py,sha256=X-zRf7BfsaKRg8GXcaa5Ic42b9uqEfqqxiI47ZijpDE,41221
 singlestoredb/magics/__init__.py,sha256=fqCBQ0s8o1CYE4Xo_XiSbkLDzLgMNDgpSkOx66-uDZw,1244
 singlestoredb/magics/run_personal.py,sha256=M11xHi9lWquh_pLSpFI89LGE7PhOPQOGqlSPDl48itE,1900
 singlestoredb/magics/run_shared.py,sha256=rnKpW4d8CJvD6ehK8jG8FlxuqZvjZl4KocPTsk-23O8,1805
 singlestoredb/management/__init__.py,sha256=A66ZnFyX--PsAZ2tvtYUfIUBvVGDBFQsnVc6nGTlX60,277
 singlestoredb/management/billing_usage.py,sha256=0UHFSPCrN0nyeGFFM-HXS3NP8pYmYo2BCCahDEPXvzg,3883
 singlestoredb/management/cluster.py,sha256=auBzNYIXvnI6rq3DNpPgJhwWoT6JsyZRikjpON23Pxg,14867
-singlestoredb/management/export.py,sha256=2dCvnTnkxVI-arX3_375DtWzHkI1YNK5zaFYdHKE4cs,5277
+singlestoredb/management/export.py,sha256=T2uU2EaeFYitvK_7475U_wYJS-rnGLE1RsTsP6XbsQs,9376
 singlestoredb/management/files.py,sha256=Z9GpS2EHf9atE8kJdz1vJtsiT80O6TV00MPhqyXfAAw,31579
 singlestoredb/management/inference_api.py,sha256=9d9-7edoZ6JI3SPvStcVDOSHOY6l38V1MFpyskdLAZY,2684
 singlestoredb/management/job.py,sha256=Npfe1JLYJlggGBrXLniPKwKUKF1i3alvSY1SFtvauSs,25498
@@ -128,7 +128,7 @@ singlestoredb/tests/test_ext_func.py,sha256=LhuPz8o3UF7x2LNod5oZ1tlxeLvGDEUE5Fnz
 singlestoredb/tests/test_ext_func_data.py,sha256=9kn8BWmCjkbnP6hSbFhmhcdW4OmVT-GSvBTIzFBLEys,48796
 singlestoredb/tests/test_fusion.py,sha256=S0Jk2NrcOitqM98r5fosHGbZ1sCZ2uxar5t48v-uOD0,52045
 singlestoredb/tests/test_http.py,sha256=7hwXe61hlUes3nji0MTTZweo94tJAlJ-vA5ct9geXFQ,8868
-singlestoredb/tests/test_management.py,sha256=bq75vQw_Uc6qrPk-lxoSbsz6p1AkF9hzQqaXRL6vgGY,46499
+singlestoredb/tests/test_management.py,sha256=JGmfI408wp6lCzeteK7qSuiFOMfAgqkpE0jy-DP7uL0,46935
 singlestoredb/tests/test_plugin.py,sha256=P1nXLnTafaHkHN-6bVbGryxTu7OWJPU9SYFZ_WQUwq8,845
 singlestoredb/tests/test_results.py,sha256=Zg1ynZFRZqalAMfNLOU5C6BDXaox6JxrKm_XZwVNFcg,6753
 singlestoredb/tests/test_types.py,sha256=YeVE6KPqlqzJke-4hbRmc8ko1E7RLHu5S8qLg04Bl5Y,4632
@@ -149,9 +149,9 @@ singlestoredb/utils/results.py,sha256=wR70LhCqlobniZf52r67zYLBOKjWHQm68NAskdRQND
 singlestoredb/utils/xdict.py,sha256=-wi1lSPTnY99fhVMBhPKJ8cCsQhNG4GMUfkEBDKYgCw,13321
 sqlx/__init__.py,sha256=4Sdn8HN-Hf8v0_wCt60DCckCg8BvgM3-9r4YVfZycRE,89
 sqlx/magic.py,sha256=6VBlotgjautjev599tHaTYOfcfOA9m6gV_-P1_Qc4lI,3622
-singlestoredb-1.14.0.dist-info/LICENSE,sha256=Bojenzui8aPNjlF3w4ojguDP7sTf8vFV_9Gc2UAG1sg,11542
-singlestoredb-1.14.0.dist-info/METADATA,sha256=oAx3-B1zWSeqAjatJUyu_u1GzqOmZ9aYNKU0v0Mz8Ic,5895
-singlestoredb-1.14.0.dist-info/WHEEL,sha256=c4k7z5HB0t-y0nBCv6KyJ6KCjn8SEGPddD0lhaPtU3E,96
-singlestoredb-1.14.0.dist-info/entry_points.txt,sha256=bSLaTWB5zGjpVYPAaI46MkkDup0su-eb3uAhCNYuRV0,48
-singlestoredb-1.14.0.dist-info/top_level.txt,sha256=lA65Vf4qAMfg_s1oG3LEO90h4t1Z-SPDbRqkevI3bSY,40
-singlestoredb-1.14.0.dist-info/RECORD,,
+singlestoredb-1.14.1.dist-info/LICENSE,sha256=Bojenzui8aPNjlF3w4ojguDP7sTf8vFV_9Gc2UAG1sg,11542
+singlestoredb-1.14.1.dist-info/METADATA,sha256=7RYWH5ZbFrph0qvRKBQo-ZnhQhL9xuVb2rGrdezCQGU,5895
+singlestoredb-1.14.1.dist-info/WHEEL,sha256=c4k7z5HB0t-y0nBCv6KyJ6KCjn8SEGPddD0lhaPtU3E,96
+singlestoredb-1.14.1.dist-info/entry_points.txt,sha256=bSLaTWB5zGjpVYPAaI46MkkDup0su-eb3uAhCNYuRV0,48
+singlestoredb-1.14.1.dist-info/top_level.txt,sha256=lA65Vf4qAMfg_s1oG3LEO90h4t1Z-SPDbRqkevI3bSY,40
+singlestoredb-1.14.1.dist-info/RECORD,,