crate 0.30.0__tar.gz → 0.30.1__tar.gz

{crate-0.30.0 → crate-0.30.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: crate
-Version: 0.30.0
+Version: 0.30.1
 Summary: CrateDB Python Client
 Home-page: https://github.com/crate/crate-python
 Author: Crate.io

{crate-0.30.0 → crate-0.30.1}/src/crate/client/__init__.py

@@ -29,7 +29,7 @@ __all__ = [
 
 # version string read from setup.py using a regex. Take care not to break the
 # regex!
-__version__ = "0.30.0"
+__version__ = "0.30.1"
 
 apilevel = "2.0"
 threadsafety = 2

{crate-0.30.0 → crate-0.30.1}/src/crate/client/sqlalchemy/dialect.py

@@ -227,11 +227,11 @@ class CrateDialect(default.DefaultDialect):
     def dbapi(cls):
         return cls.import_dbapi()
 
-    def has_schema(self, connection, schema):
-        return schema in self.get_schema_names(connection)
+    def has_schema(self, connection, schema, **kw):
+        return schema in self.get_schema_names(connection, **kw)
 
-    def has_table(self, connection, table_name, schema=None):
-        return table_name in self.get_table_names(connection, schema=schema)
+    def has_table(self, connection, table_name, schema=None, **kw):
+        return table_name in self.get_table_names(connection, schema=schema, **kw)
 
     @reflection.cache
     def get_schema_names(self, connection, **kw):

{crate-0.30.0 → crate-0.30.1}/src/crate/client/sqlalchemy/tests/dialect_test.py

@@ -20,12 +20,14 @@
 # software solely pursuant to the terms of the relevant commercial agreement.
 
 from datetime import datetime
-from unittest import TestCase
+from unittest import TestCase, skipIf
 from unittest.mock import MagicMock, patch
 
 import sqlalchemy as sa
 
 from crate.client.cursor import Cursor
+from crate.client.sqlalchemy import SA_VERSION
+from crate.client.sqlalchemy.sa_version import SA_1_4, SA_2_0
 from crate.client.sqlalchemy.types import Object
 from sqlalchemy import inspect
 from sqlalchemy.orm import Session
@@ -33,7 +35,7 @@ try:
     from sqlalchemy.orm import declarative_base
 except ImportError:
     from sqlalchemy.ext.declarative import declarative_base
-from sqlalchemy.testing import eq_, in_
+from sqlalchemy.testing import eq_, in_, is_true
 
 FakeCursor = MagicMock(name='FakeCursor', spec=Cursor)
 
@@ -70,6 +72,13 @@ class SqlAlchemyDialectTest(TestCase):
 
         self.session = Session(bind=self.engine)
 
+    def init_mock(self, return_value=None):
+        self.fake_cursor.rowcount = 1
+        self.fake_cursor.description = (
+            ('foo', None, None, None, None, None, None),
+        )
+        self.fake_cursor.fetchall = MagicMock(return_value=return_value)
+
     def test_primary_keys_2_3_0(self):
         insp = inspect(self.session.bind)
         self.engine.dialect.server_version_info = (2, 3, 0)
@@ -126,3 +135,22 @@ class SqlAlchemyDialectTest(TestCase):
             ['v1', 'v2'])
         eq_(self.executed_statement, "SELECT table_name FROM information_schema.views "
                                      "ORDER BY table_name ASC, table_schema ASC")
+
+    @skipIf(SA_VERSION < SA_1_4, "Inspector.has_table only available on SQLAlchemy>=1.4")
+    def test_has_table(self):
+        self.init_mock(return_value=[["foo"], ["bar"]])
+        insp = inspect(self.session.bind)
+        is_true(insp.has_table("bar"))
+        eq_(self.executed_statement,
+            "SELECT table_name FROM information_schema.tables "
+            "WHERE table_schema = ? AND table_type = 'BASE TABLE' "
+            "ORDER BY table_name ASC, table_schema ASC")
+
+    @skipIf(SA_VERSION < SA_2_0, "Inspector.has_schema only available on SQLAlchemy>=2.0")
+    def test_has_schema(self):
+        self.init_mock(
+            return_value=[["blob"], ["doc"], ["information_schema"], ["pg_catalog"], ["sys"]])
+        insp = inspect(self.session.bind)
+        is_true(insp.has_schema("doc"))
+        eq_(self.executed_statement,
+            "select schema_name from information_schema.schemata order by schema_name asc")

{crate-0.30.0 → crate-0.30.1}/src/crate.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: crate
-Version: 0.30.0
+Version: 0.30.1
 Summary: CrateDB Python Client
 Home-page: https://github.com/crate/crate-python
 Author: Crate.io