Modify error handlings

thombashi · Mar 21, 2020 · 5af0b1e · 5af0b1e
1 parent 00d355a
commit 5af0b1e
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/simplesqlite/_sanitizer.py b/simplesqlite/_sanitizer.py
@@ -92,7 +92,7 @@ def _validate_table_name(self, table_name: str) -> None:
             validate_sqlite_table_name(table_name)
         except ValidationError as e:
             if (
-                e.reason == ErrorReason.RESERVED_NAME and not e.reusable_name
+                e.reason == ErrorReason.RESERVED_NAME and e.reusable_name is False
             ) or e.reason == ErrorReason.INVALID_CHARACTER:
                 raise InvalidTableNameError(e)
             elif e.reason == ErrorReason.RESERVED_NAME:

diff --git a/simplesqlite/query.py b/simplesqlite/query.py
@@ -108,7 +108,7 @@ def to_query(self) -> str:
         try:
             validate_sqlite_attr_name(name)
         except ValidationError as e:
-            if e.reason == ErrorReason.RESERVED_NAME and not e.reusable_name:
+            if e.reason == ErrorReason.RESERVED_NAME and e.reusable_name is False:
                 need_quote = True
             elif e.reason in (
                 ErrorReason.RESERVED_NAME,

diff --git a/test/test_validator.py b/test/test_validator.py
@@ -229,7 +229,7 @@ def test_exception_reserved_invalid_name(self, value):
             validate_sqlite_table_name(value)
         except ValidationError as e:
             assert e.reason == ErrorReason.RESERVED_NAME
-            assert not e.reusable_name
+            assert e.reusable_name is False
 
 
 class Test_validate_sqlite_attr_name:
@@ -306,4 +306,4 @@ def test_exception_reserved_invalid_name(self, value):
             validate_sqlite_attr_name(value)
         except ValidationError as e:
             assert e.reason == ErrorReason.RESERVED_NAME
-            assert not e.reusable_name
+            assert e.reusable_name is False