singlestoredb 1.5.0__py3-none-any.whl → 1.6.1__py3-none-any.whl

singlestoredb/management/organization.py

@@ -7,6 +7,7 @@ from typing import Optional
 from typing import Union
 
 from ..exceptions import ManagementError
+from .job import JobsManager
 from .manager import Manager
 from .utils import vars_to_str
 
@@ -190,3 +191,19 @@ class Organization(object):
         )
         out._manager = manager
         return out
+
+    @property
+    def jobs(self) -> JobsManager:
+        """
+        Retrieve a SingleStoreDB scheduled job manager.
+
+        Parameters
+        ----------
+        manager : WorkspaceManager, optional
+            The WorkspaceManager the JobsManager belongs to
+
+        Returns
+        -------
+        :class:`JobsManager`
+        """
+        return JobsManager(self._manager)

singlestoredb/management/utils.py

@@ -215,6 +215,26 @@ def get_token() -> Optional[str]:
     return None
 
 
+def get_cluster_id() -> Optional[str]:
+    """Return the cluster id for the current token or environment."""
+    return os.environ.get('SINGLESTOREDB_CLUSTER') or None
+
+
+def get_workspace_id() -> Optional[str]:
+    """Return the workspace id for the current token or environment."""
+    return os.environ.get('SINGLESTOREDB_WORKSPACE') or None
+
+
+def get_virtual_workspace_id() -> Optional[str]:
+    """Return the virtual workspace id for the current token or environment."""
+    return os.environ.get('SINGLESTOREDB_VIRTUAL_WORKSPACE') or None
+
+
+def get_database_name() -> Optional[str]:
+    """Return the default database name for the current token or environment."""
+    return os.environ.get('SINGLESTOREDB_DEFAULT_DATABASE') or None
+
+
 def enable_http_tracing() -> None:
     """Enable tracing of HTTP requests."""
     import logging
@@ -246,7 +266,33 @@ def to_datetime(
     out = converters.datetime_fromisoformat(obj)
     if isinstance(out, str):
         return None
-    if isinstance(out, datetime.date):
+    if isinstance(out, datetime.date) and not isinstance(out, datetime.datetime):
+        return datetime.datetime(out.year, out.month, out.day)
+    return out
+
+
+def to_datetime_strict(
+    obj: Optional[Union[str, datetime.datetime]],
+) -> datetime.datetime:
+    """Convert string to datetime."""
+    if not obj:
+        raise TypeError('not possible to convert None to datetime')
+    if isinstance(obj, datetime.datetime):
+        return obj
+    if obj == '0001-01-01T00:00:00Z':
+        raise ValueError('not possible to convert 0001-01-01T00:00:00Z to datetime')
+    obj = obj.replace('Z', '')
+    # Fix datetimes with truncated zeros
+    if '.' in obj:
+        obj, micros = obj.split('.', 1)
+        micros = micros + '0' * (6 - len(micros))
+        obj = obj + '.' + micros
+    out = converters.datetime_fromisoformat(obj)
+    if not out:
+        raise TypeError('not possible to convert None to datetime')
+    if isinstance(out, str):
+        raise ValueError('value cannot be str')
+    if isinstance(out, datetime.date) and not isinstance(out, datetime.datetime):
         return datetime.datetime(out.year, out.month, out.day)
     return out
 

singlestoredb/mysql/connection.py

@@ -716,6 +716,7 @@ class Connection(BaseConnection):
         Error : If the connection is already closed.
 
         """
+        self._result = None
         if self.host == 'singlestore.com':
             return
         if self._closed:
@@ -1909,12 +1910,12 @@ class MySQLResultSV(MySQLResult):
                 encoding_errors=connection.encoding_errors,
             ).items() if v is not UNSET
         }
-        self._read_rowdata_packet = functools.partial(
-            _singlestoredb_accel.read_rowdata_packet, self, False,
-        )
-        self._read_rowdata_packet_unbuffered = functools.partial(
-            _singlestoredb_accel.read_rowdata_packet, self, True,
-        )
+
+    def _read_rowdata_packet(self, *args, **kwargs):
+        return _singlestoredb_accel.read_rowdata_packet(self, False, *args, **kwargs)
+
+    def _read_rowdata_packet_unbuffered(self, *args, **kwargs):
+        return _singlestoredb_accel.read_rowdata_packet(self, True, *args, **kwargs)
 
 
 class LoadLocalFile:

singlestoredb/tests/test_fusion.py

@@ -91,7 +91,7 @@ class TestFusion(unittest.TestCase):
 
 
 @pytest.mark.management
-class TestManagementAPIFusion(unittest.TestCase):
+class TestWorkspaceFusion(unittest.TestCase):
 
     id: str = secrets.token_hex(8)
     dbname: str = ''
@@ -463,3 +463,252 @@ class TestManagementAPIFusion(unittest.TestCase):
                 mgr.workspace_groups[wg_name].terminate(force=True)
             except Exception:
                 pass
+
+
+@pytest.mark.management
+class TestJobsFusion(unittest.TestCase):
+
+    id: str = secrets.token_hex(8)
+    notebook_name: str = 'Scheduling Test.ipynb'
+    dbname: str = ''
+    dbexisted: bool = False
+    manager: None
+    workspace_group: None
+    workspace: None
+    job_ids = []
+
+    @classmethod
+    def setUpClass(cls):
+        sql_file = os.path.join(os.path.dirname(__file__), 'test.sql')
+        cls.dbname, cls.dbexisted = utils.load_sql(sql_file)
+        cls.manager = s2.manage_workspaces()
+        us_regions = [x for x in cls.manager.regions if x.name.startswith('US')]
+        cls.workspace_group = cls.manager.create_workspace_group(
+            f'Jobs Fusion Testing {cls.id}',
+            region=random.choice(us_regions),
+            firewall_ranges=[],
+        )
+        cls.workspace = cls.workspace_group.create_workspace(
+                f'jobs-test-{cls.id}',
+                wait_on_active=True,
+        )
+        os.environ['SINGLESTOREDB_DEFAULT_DATABASE'] = cls.dbname
+        os.environ['SINGLESTOREDB_WORKSPACE'] = cls.workspace.id
+
+    @classmethod
+    def tearDownClass(cls):
+        for job_id in cls.job_ids:
+            cls.manager.organizations.current.jobs.delete(job_id)
+        if cls.workspace_group is not None:
+            cls.workspace_group.terminate(force=True)
+        cls.manager = None
+        cls.workspace_group = None
+        cls.workspace = None
+        if os.environ.get('SINGLESTOREDB_WORKSPACE', None) is not None:
+            del os.environ['SINGLESTOREDB_WORKSPACE']
+        if os.environ.get('SINGLESTOREDB_DEFAULT_DATABASE', None) is not None:
+            del os.environ['SINGLESTOREDB_DEFAULT_DATABASE']
+
+    def setUp(self):
+        self.enabled = os.environ.get('SINGLESTOREDB_FUSION_ENABLED')
+        os.environ['SINGLESTOREDB_FUSION_ENABLED'] = '1'
+        self.conn = s2.connect(database=type(self).dbname, local_infile=True)
+        self.cur = self.conn.cursor()
+
+    def tearDown(self):
+        if self.enabled:
+            os.environ['SINGLESTOREDB_FUSION_ENABLED'] = self.enabled
+        else:
+            del os.environ['SINGLESTOREDB_FUSION_ENABLED']
+
+        try:
+            if self.cur is not None:
+                self.cur.close()
+        except Exception:
+            # traceback.print_exc()
+            pass
+
+        try:
+            if self.conn is not None:
+                self.conn.close()
+        except Exception:
+            # traceback.print_exc()
+            pass
+
+    def test_schedule_drop_job(self):
+        # schedule recurring job
+        self.cur.execute(
+                f'schedule job using notebook "{self.notebook_name}" '
+                'with mode "recurring" '
+                'execute every 5 minutes '
+                'with name "recurring-job" '
+                'create snapshot '
+                'resume target '
+                'with runtime "notebooks-cpu-small" '
+                'with parameters '
+                '{"strParam": "string", "intParam": 1, '
+                '"floatParam": 1.0, "boolParam": true}',
+        )
+        out = list(self.cur)
+        job_id = out[0][0]
+        self.job_ids.append(job_id)
+        desc = self.cur.description
+        assert len(desc) == 1
+        assert desc[0][0] == 'JobID'
+        assert len(out) == 1
+        assert out[0][0] == job_id
+
+        # drop job
+        self.cur.execute(f'drop jobs {job_id}')
+        out = list(self.cur)
+        desc = self.cur.description
+        assert len(desc) == 2
+        assert [x[0] for x in desc] == [
+            'JobID', 'Success',
+        ]
+        assert len(out) == 1
+        res = out[0]
+        assert res[0] == job_id
+        assert res[1] == 1
+
+    def test_run_wait_drop_job(self):
+        # run job
+        self.cur.execute(
+            f'run job using notebook "{self.notebook_name}" '
+            'with runtime "notebooks-cpu-small" '
+            'with parameters '
+            '{"strParam": "string", "intParam": 1, '
+            '"floatParam": 1.0, "boolParam": true}',
+        )
+        out = list(self.cur)
+        job_id = out[0][0]
+        self.job_ids.append(job_id)
+        desc = self.cur.description
+        assert len(desc) == 1
+        assert desc[0][0] == 'JobID'
+        assert len(out) == 1
+        assert out[0][0] == job_id
+
+        # wait on job
+        self.cur.execute(f'wait on jobs {job_id}')
+        out = list(self.cur)
+        desc = self.cur.description
+        assert len(desc) == 1
+        assert desc[0][0] == 'Success'
+        assert out[0][0] == 1
+
+        # drop job
+        self.cur.execute(f'drop jobs {job_id}')
+        out = list(self.cur)
+        desc = self.cur.description
+        assert len(desc) == 2
+        assert [x[0] for x in desc] == [
+            'JobID', 'Success',
+        ]
+        assert len(out) == 1
+        res = out[0]
+        assert res[0] == job_id
+        assert res[1] == 1
+
+    def test_show_jobs_and_executions(self):
+        # schedule recurring job
+        self.cur.execute(
+                f'schedule job using notebook "{self.notebook_name}" '
+                'with mode "recurring" '
+                'execute every 5 minutes '
+                'with name "show-job" '
+                'with runtime "notebooks-cpu-small" '
+                'with parameters '
+                '{"strParam": "string", "intParam": 1, '
+                '"floatParam": 1.0, "boolParam": true}',
+        )
+        out = list(self.cur)
+        job_id = out[0][0]
+        self.job_ids.append(job_id)
+        desc = self.cur.description
+        assert len(desc) == 1
+        assert desc[0][0] == 'JobID'
+        assert len(out) == 1
+        assert out[0][0] == job_id
+
+        # show jobs with name like "show-job"
+        self.cur.execute(f'show jobs {job_id} like "show-job"')
+        out = list(self.cur)
+        desc = self.cur.description
+        assert len(desc) == 9
+        assert [x[0] for x in desc] == [
+            'JobID', 'Name', 'CreatedAt', 'EnqueuedBy',
+            'CompletedExecutions', 'NotebookPath', 'DatabaseName', 'TargetID',
+            'TargetType',
+        ]
+        assert len(out) == 1
+        job = out[0]
+        assert job[0] == job_id
+        assert job[1] == 'show-job'
+        assert job[5] == self.notebook_name
+        assert job[6] == self.dbname
+        assert job[7] == self.workspace.id
+        assert job[8] == 'Workspace'
+
+        # show jobs with name like "show-job" extended
+        self.cur.execute(f'show jobs {job_id} like "show-job" extended')
+        out = list(self.cur)
+        desc = self.cur.description
+        assert len(desc) == 17
+        assert [x[0] for x in desc] == [
+            'JobID', 'Name', 'CreatedAt', 'EnqueuedBy',
+            'CompletedExecutions', 'NotebookPath', 'DatabaseName', 'TargetID',
+            'TargetType', 'Description', 'TerminatedAt', 'CreateSnapshot',
+            'MaxDurationInMins', 'ExecutionIntervalInMins', 'Mode', 'StartAt',
+            'ResumeTarget',
+        ]
+        assert len(out) == 1
+        job = out[0]
+        assert job[0] == job_id
+        assert job[1] == 'show-job'
+        assert job[5] == self.notebook_name
+        assert job[6] == self.dbname
+        assert job[7] == self.workspace.id
+        assert job[8] == 'Workspace'
+        assert not job[11]
+        assert job[13] == 5
+        assert job[14] == 'Recurring'
+        assert not job[16]
+
+        # show executions for job with id job_id from 1 to 5
+        self.cur.execute(f'show job executions for {job_id} from 1 to 5')
+        out = list(self.cur)
+        desc = self.cur.description
+        assert len(desc) == 7
+        assert [x[0] for x in desc] == [
+            'ExecutionID', 'ExecutionNumber', 'JobID',
+            'Status', 'ScheduledStartTime', 'StartedAt', 'FinishedAt',
+        ]
+        exec_job_ids = [x[2] for x in out]
+        assert [x for x in exec_job_ids] == [job_id]
+
+        # show executions for job with id job_id from 1 to 5 extended
+        self.cur.execute(f'show job executions for {job_id} from 1 to 5 extended')
+        out = list(self.cur)
+        desc = self.cur.description
+        assert len(desc) == 8
+        assert [x[0] for x in desc] == [
+            'ExecutionID', 'ExecutionNumber', 'JobID',
+            'Status', 'ScheduledStartTime', 'StartedAt', 'FinishedAt',
+            'SnapshotNotebookPath',
+        ]
+        exec_job_ids = [x[2] for x in out]
+        assert [x for x in exec_job_ids] == [job_id]
+
+        # drop job
+        self.cur.execute(f'drop jobs {job_id}')
+        out = list(self.cur)
+        desc = self.cur.description
+        assert len(desc) == 2
+        assert [x[0] for x in desc] == [
+            'JobID', 'Success',
+        ]
+        assert len(out) == 1
+        res = out[0]
+        assert res[0] == job_id
+        assert res[1] == 1

singlestoredb/tests/test_management.py

@@ -11,6 +11,8 @@ import unittest
 import pytest
 
 import singlestoredb as s2
+from singlestoredb.management.job import Status
+from singlestoredb.management.job import TargetType
 
 
 TEST_DIR = pathlib.Path(os.path.dirname(__file__))
@@ -876,3 +878,153 @@ class TestSecrets(unittest.TestCase):
 
         assert secret.name == 'secret_name'
         assert secret.value == 'secret_value'
+
+
+@pytest.mark.management
+class TestJob(unittest.TestCase):
+
+    manager = None
+    workspace_group = None
+    workspace = None
+    password = None
+    job_ids = []
+
+    @classmethod
+    def setUpClass(cls):
+        cls.manager = s2.manage_workspaces()
+
+        us_regions = [x for x in cls.manager.regions if 'US' in x.name]
+        cls.password = secrets.token_urlsafe(20)
+
+        name = clean_name(secrets.token_urlsafe(20)[:20])
+
+        cls.workspace_group = cls.manager.create_workspace_group(
+            f'wg-test-{name}',
+            region=random.choice(us_regions).id,
+            admin_password=cls.password,
+            firewall_ranges=['0.0.0.0/0'],
+        )
+
+        try:
+            cls.workspace = cls.workspace_group.create_workspace(
+                f'ws-test-{name}-x',
+                wait_on_active=True,
+            )
+        except Exception:
+            cls.workspace_group.terminate(force=True)
+            raise
+
+    @classmethod
+    def tearDownClass(cls):
+        for job_id in cls.job_ids:
+            cls.manager.organizations.current.jobs.delete(job_id)
+        if cls.workspace_group is not None:
+            cls.workspace_group.terminate(force=True)
+        cls.workspace_group = None
+        cls.workspace = None
+        cls.manager = None
+        cls.password = None
+        if os.environ.get('SINGLESTOREDB_WORKSPACE', None) is not None:
+            del os.environ['SINGLESTOREDB_WORKSPACE']
+        if os.environ.get('SINGLESTOREDB_DEFAULT_DATABASE', None) is not None:
+            del os.environ['SINGLESTOREDB_DEFAULT_DATABASE']
+
+    def test_job_without_database_target(self):
+        """
+        Creates job without target database on a specific runtime
+        Waits for job to finish
+        Gets the job
+        Deletes the job
+        """
+        if os.environ.get('SINGLESTOREDB_WORKSPACE', None) is not None:
+            del os.environ['SINGLESTOREDB_WORKSPACE']
+        if os.environ.get('SINGLESTOREDB_DEFAULT_DATABASE', None) is not None:
+            del os.environ['SINGLESTOREDB_DEFAULT_DATABASE']
+
+        job_manager = self.manager.organizations.current.jobs
+        job = job_manager.run(
+            'Scheduling Test.ipynb',
+            'notebooks-cpu-small',
+            {'strParam': 'string', 'intParam': 1, 'floatParam': 1.0, 'boolParam': True},
+        )
+        self.job_ids.append(job.job_id)
+        assert job.execution_config.notebook_path == 'Scheduling Test.ipynb'
+        assert job.schedule.mode == job_manager.modes().ONCE
+        assert not job.execution_config.create_snapshot
+        assert job.completed_executions_count == 0
+        assert job.name is None
+        assert job.description is None
+        assert job.job_metadata == []
+        assert job.terminated_at is None
+        assert job.target_config is None
+        job.wait()
+        job = job_manager.get(job.job_id)
+        assert job.execution_config.notebook_path == 'Scheduling Test.ipynb'
+        assert job.schedule.mode == job_manager.modes().ONCE
+        assert not job.execution_config.create_snapshot
+        assert job.completed_executions_count == 1
+        assert job.name is None
+        assert job.description is None
+        assert job.job_metadata != []
+        assert len(job.job_metadata) == 1
+        assert job.job_metadata[0].count == 1
+        assert job.job_metadata[0].status == Status.COMPLETED
+        assert job.terminated_at is None
+        assert job.target_config is None
+        deleted = job.delete()
+        assert deleted
+        job = job_manager.get(job.job_id)
+        assert job.terminated_at is not None
+
+    def test_job_with_database_target(self):
+        """
+        Creates job with target database on a specific runtime
+        Waits for job to finish
+        Gets the job
+        Deletes the job
+        """
+        os.environ['SINGLESTOREDB_DEFAULT_DATABASE'] = 'information_schema'
+        os.environ['SINGLESTOREDB_WORKSPACE'] = self.workspace.id
+
+        job_manager = self.manager.organizations.current.jobs
+        job = job_manager.run(
+            'Scheduling Test.ipynb',
+            'notebooks-cpu-small',
+            {'strParam': 'string', 'intParam': 1, 'floatParam': 1.0, 'boolParam': True},
+        )
+        self.job_ids.append(job.job_id)
+        assert job.execution_config.notebook_path == 'Scheduling Test.ipynb'
+        assert job.schedule.mode == job_manager.modes().ONCE
+        assert not job.execution_config.create_snapshot
+        assert job.completed_executions_count == 0
+        assert job.name is None
+        assert job.description is None
+        assert job.job_metadata == []
+        assert job.terminated_at is None
+        assert job.target_config is not None
+        assert job.target_config.database_name == 'information_schema'
+        assert job.target_config.target_id == self.workspace.id
+        assert job.target_config.target_type == TargetType.WORKSPACE
+        assert not job.target_config.resume_target
+        job.wait()
+        job = job_manager.get(job.job_id)
+        assert job.execution_config.notebook_path == 'Scheduling Test.ipynb'
+        assert job.schedule.mode == job_manager.modes().ONCE
+        assert not job.execution_config.create_snapshot
+        assert job.completed_executions_count == 1
+        assert job.name is None
+        assert job.description is None
+        assert job.job_metadata != []
+        assert len(job.job_metadata) == 1
+        assert job.job_metadata[0].count == 1
+        assert job.job_metadata[0].status == Status.COMPLETED
+        assert job.terminated_at is None
+        assert job.target_config is not None
+        assert job.target_config.database_name == 'information_schema'
+        assert job.target_config.target_id == self.workspace.id
+        assert job.target_config.target_type == TargetType.WORKSPACE
+        assert not job.target_config.resume_target
+        deleted = job.delete()
+        assert deleted
+        job = job_manager.get(job.job_id)
+        assert job.terminated_at is not None

singlestoredb/tests/utils.py

@@ -73,6 +73,13 @@ def load_sql(sql_file: str) -> str:
     with open(sql_file, 'r') as infile:
         with s2.connect(**args) as conn:
             with conn.cursor() as cur:
+                try:
+                    cur.execute('SET GLOBAL default_partitions_per_leaf=2')
+                    cur.execute('SET GLOBAL log_file_size_partitions=1048576')
+                    cur.execute('SET GLOBAL log_file_size_ref_dbs=1048576')
+                except s2.OperationalError:
+                    pass
+
                 if not dbname:
                     dbname = 'TEST_{}'.format(uuid.uuid4()).replace('-', '_')
                     cur.execute(f'CREATE DATABASE {dbname};')

{singlestoredb-1.5.0.dist-info → singlestoredb-1.6.1.dist-info}/METADATA

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

{singlestoredb-1.5.0.dist-info → singlestoredb-1.6.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-singlestoredb/__init__.py,sha256=uEuDac8tSZiJCE2NiWh3zAtGwMalgs9DTd-xAsWHGYM,1634
+singlestoredb/__init__.py,sha256=n3bl_B1xZ3yauNuLO0b0MQNYSMrx5LfpmEJVE_qgXN0,1634
 singlestoredb/auth.py,sha256=u8D9tpKzrqa4ssaHjyZnGDX1q8XBpGtuoOkTkSv7B28,7599
 singlestoredb/config.py,sha256=0qU2lweqHIA5yuV0ZN_JS-cZCfLIxIHutgS2YKRuwXw,12067
 singlestoredb/connection.py,sha256=Gzq38vG8lXTx5v2WvHkdMg-abSTS6lguTsVMxbYHZmQ,45306
@@ -22,27 +22,29 @@ singlestoredb/functions/ext/rowdat_1.py,sha256=JgKRsVSQYczFD6cmo2xLilbNPYpyLL2tP
 singlestoredb/functions/ext/utils.py,sha256=2-B8YU_Iekv8JcpI-ochs9TIeuyatLaLAH-AyYyUUIg,5311
 singlestoredb/fusion/__init__.py,sha256=Qo7SuqGw-l-vE8-EI2jhm6hXJkYfOLUKIws9c7LFNX0,356
 singlestoredb/fusion/graphql.py,sha256=ZA3HcDq5rER-dCEavwTqnF7KM0D2LCYIY7nLQk7lSso,5207
-singlestoredb/fusion/handler.py,sha256=wuNq-nzzBngSha_kZ_Da7V7VzQzmhgXgUfh2V0HXLVI,24976
+singlestoredb/fusion/handler.py,sha256=6OmDzGUDl98bmQm1XjLIiK_6zi14qIz_XE-PmCBbX-Y,25155
 singlestoredb/fusion/registry.py,sha256=jjdRTYZ3ylhy6gAoW5xBj0tkxGFBT-2yLQ0tztTgDIY,6112
 singlestoredb/fusion/result.py,sha256=Bd3KbRpqWqQcWp_Chd4bzBy8Kfc8nXLS_Pn_GGbPO6o,11772
 singlestoredb/fusion/handlers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+singlestoredb/fusion/handlers/job.py,sha256=EfAahiQkhrNUbO8jxCPMHtUhntAZOK2xsd1D0dSTMZY,21109
 singlestoredb/fusion/handlers/stage.py,sha256=4BSwqcmA4PI3oksTdYaTmQ6bY-AzmL0mZfGnP3U0ldM,13643
 singlestoredb/fusion/handlers/utils.py,sha256=oYbf13Y3orEkJfHMNnO7B_W1anEdK-0S9vVVkF2pPFk,5109
-singlestoredb/fusion/handlers/workspace.py,sha256=faVuqiE4PTVOAgbsr18MOc7H92nVzfwga0SUTPODvSc,26455
+singlestoredb/fusion/handlers/workspace.py,sha256=4xN2TFO4yF7KZB2Fcht7IuvoDdAT6fDfDLjixiHZN8w,27506
 singlestoredb/http/__init__.py,sha256=A_2ZUCCpvRYIA6YDpPy57wL5R1eZ5SfP6I1To5nfJ2s,912
 singlestoredb/http/connection.py,sha256=T2NrL7XV30eQRdrcbP_IY8c3ZOV6Th4KQiuleW4tHUE,39450
 singlestoredb/management/__init__.py,sha256=mhWXjLhp5-t8dhl0vl7SjayKrvJlDb5_hl1YWvDgiMA,237
 singlestoredb/management/billing_usage.py,sha256=9ighjIpcopgIyJOktBYQ6pahBZmWGHOPyyCW4gu9FGs,3735
 singlestoredb/management/cluster.py,sha256=i23Smr1PBrDZ8NO_VPd_-bEYkyHvVe9CCRGUjHn_1yQ,14362
+singlestoredb/management/job.py,sha256=4-xLWzbE8odQogVVaFer80UEoTAZY1T28VZ9Ug4rbmM,24611
 singlestoredb/management/manager.py,sha256=sFP1vZGS8WpN8E0XLu1N7Mxtq6Sixalln44HlTQEyXI,8800
-singlestoredb/management/organization.py,sha256=Y0JFSxYF_UOjip53gcbBXWPCe_t92zo3d99jZNrhkx4,4995
+singlestoredb/management/organization.py,sha256=hqMaM7H-naMjNbxDl_f7G_2o5TkiGKyzPhxuzDveJAw,5402
 singlestoredb/management/region.py,sha256=HnLcWUh7r_aLECliplCDHak4a_F3B7LOSXEYMW66qD0,1611
-singlestoredb/management/utils.py,sha256=t-O6NRb4Wad1qJJ1afxtFR1IX-0bezEqplKllnm8tto,11398
+singlestoredb/management/utils.py,sha256=P4fp8a7EwaYiag_hvpILcgwXtdFNYKKO70dsKjmxn1A,13171
 singlestoredb/management/workspace.py,sha256=9oamNIaE5vLZNAlpl5SK-xu27qPqx2Ff3ZIdKckNfmc,61673
 singlestoredb/mysql/__init__.py,sha256=olUTAvkiERhDW41JXQMawkg-i0tvBEkoTkII1tt6lxU,4492
 singlestoredb/mysql/_auth.py,sha256=AugRitoUwgRIDFuJxuAH4MWIAmckY7Ji2pP6r_Ng9dY,8043
 singlestoredb/mysql/charset.py,sha256=-FlONDS_oAUF5B3mIgeHBPb_SCt4zHD33arUeBNctU0,10510
-singlestoredb/mysql/connection.py,sha256=cUVILM5q2tVkGAf-GpPK2pvPto3AOB_9PifK8iXaiKw,72216
+singlestoredb/mysql/connection.py,sha256=NpehiK8NTPSc2KH4toLqik3e63XMZSov4s9gOVcBfko,72260
 singlestoredb/mysql/converters.py,sha256=CVe8SDmjbIAhy1xpQ2N5OKWw6t5eWpw-EU3QTlA0Hh0,7500
 singlestoredb/mysql/cursors.py,sha256=Eqe7jITRvOo4P_TxIarTumg_2PG1DcCfZ4Uo9IFdDa8,26794
 singlestoredb/mysql/err.py,sha256=-m5rqXi8yhq6b8SCEJ2h0E5Rudh_15dlAU_WbJ1YrM8,2388
@@ -95,15 +97,15 @@ singlestoredb/tests/test_dbapi.py,sha256=IKq5Hcwx8WikASP8_AB5fo3TXv7ryWPCVGonoly
 singlestoredb/tests/test_exceptions.py,sha256=tfr_8X2w1UmG4nkSBzWGB0C7ehrf1GAVgj6_ODaG-TM,1131
 singlestoredb/tests/test_ext_func.py,sha256=OWd-CJ1Owhx72nikSWWEF2EQFCJk7vEXZM2Oy9EbYQo,37357
 singlestoredb/tests/test_ext_func_data.py,sha256=yTADD93nPxX6_rZXXLZaOWEI_yPvYyir9psn5PK9ctU,47695
-singlestoredb/tests/test_fusion.py,sha256=UPaxXt5YNa3GS44l4oZfmUcq89YgN7EWWIbW_oCkYao,15100
+singlestoredb/tests/test_fusion.py,sha256=QDBjU3OfBJoQqtzwlN2dQHwA3W91_wyJvun6HZ-YcmQ,23810
 singlestoredb/tests/test_http.py,sha256=RXasTqBWRn__omj0eLFTJYIbZjd0PPdIV2d4Cqz0MC8,8580
-singlestoredb/tests/test_management.py,sha256=P5I50_gt1VE5ja4CNVo0j10fmwAqE57psxCvI_RWRFI,28223
+singlestoredb/tests/test_management.py,sha256=89hKu82qiH1YTbLzKl5FOPEapNB5qo8k06d3fkoYajc,34304
 singlestoredb/tests/test_plugin.py,sha256=qpO9wmWc62VaijN1sJ97YSYIX7I7Y5C6sY-WzwrutDQ,812
 singlestoredb/tests/test_results.py,sha256=wg93sujwt-R9_eJCgSCElgAZhLDkIiAo3qPkPydOv78,6582
 singlestoredb/tests/test_types.py,sha256=jqoAaSjhbgwB3vt0KsTcl7XBWoMMIa0mPFKhEi5bBjo,4500
 singlestoredb/tests/test_udf.py,sha256=2Ml6VMTKIfstB-L31uX-zftwPsT5C64M29WZ6iuKdjI,28075
 singlestoredb/tests/test_xdict.py,sha256=fqHspoi39nbX3fIDVkkRXcd5H50xdOsSvK0bxAMQnaE,10408
-singlestoredb/tests/utils.py,sha256=76eNdYFVnsw6S3J_RaGgGQ87Rlm8pxwyYaFYXnvAEvk,4673
+singlestoredb/tests/utils.py,sha256=2A2tEdD3t8aXWUnHtAIcFlWrflsz2MlMcCbUDaAG29c,4995
 singlestoredb/tests/ext_funcs/__init__.py,sha256=qZLnDI_Ck0tguVi-K-BKXDHAcC0jui3dsm93Djj4x08,9290
 singlestoredb/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 singlestoredb/utils/config.py,sha256=m3Xn6hsbdKyLufSnbokhFJ9Vfaz9Qpkj1IEnIiH9oJQ,24503
@@ -114,9 +116,9 @@ singlestoredb/utils/events.py,sha256=9IB84T3pKQjs7aaoSSJCw7soNngnhoTDWIC52M51R9Y
 singlestoredb/utils/mogrify.py,sha256=-a56IF70U6CkfadeaZgfjRSVsAD3PuqRrzPpjZlgbwY,4050
 singlestoredb/utils/results.py,sha256=bJtaUaDiFq26IsPAKZ2FHGB7csMn94EAxLKrP4HaEEA,15277
 singlestoredb/utils/xdict.py,sha256=S9HKgrPrnu_6b7iOwa2KrW8CmU1Uqx0BWdEyogFzWbE,12896
-singlestoredb-1.5.0.dist-info/LICENSE,sha256=Mlq78idURT-9G026aMYswwwnnrLcgzTLuXeAs5hjDLM,11341
-singlestoredb-1.5.0.dist-info/METADATA,sha256=NNlSvXOAcd1pZGUOnuUM46RAZdw6LInc8KtKhDODwJw,5570
-singlestoredb-1.5.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-singlestoredb-1.5.0.dist-info/entry_points.txt,sha256=bSLaTWB5zGjpVYPAaI46MkkDup0su-eb3uAhCNYuRV0,48
-singlestoredb-1.5.0.dist-info/top_level.txt,sha256=eet8bVPNRqiGeY0PrO5ERH2UpamwlrKHEQCffz4dOh8,14
-singlestoredb-1.5.0.dist-info/RECORD,,
+singlestoredb-1.6.1.dist-info/LICENSE,sha256=Mlq78idURT-9G026aMYswwwnnrLcgzTLuXeAs5hjDLM,11341
+singlestoredb-1.6.1.dist-info/METADATA,sha256=PAvLc4S7yV3hhUOwUIZdbrGlRZHW1atEcmximSrGY2w,5570
+singlestoredb-1.6.1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+singlestoredb-1.6.1.dist-info/entry_points.txt,sha256=bSLaTWB5zGjpVYPAaI46MkkDup0su-eb3uAhCNYuRV0,48
+singlestoredb-1.6.1.dist-info/top_level.txt,sha256=eet8bVPNRqiGeY0PrO5ERH2UpamwlrKHEQCffz4dOh8,14
+singlestoredb-1.6.1.dist-info/RECORD,,