Add black back in

CHANGELOG.md

@@ -1,5 +1,10 @@
 # Release Notes
 
+## Version _next_
+
+- Fix bug where iterating over constructable schemas did not handle some exceptions
+- Add black dependency back in
+
 ## Version 1.4.1 - 2020-08-09
 
 - Remove black dependency

open_alchemy/facades/__init__.py

@@ -1,6 +1,7 @@
 """Facades for ringfenced modules."""
 # pylint: disable=useless-import-alias
 
+from . import code_formatter as code_formatter
 from . import jsonschema as jsonschema
 from . import models as models
 from . import sqlalchemy as sqlalchemy

open_alchemy/facades/code_formatter/__init__.py

@@ -0,0 +1,22 @@
+"""Apply automatic code formatting to source code."""
+
+import black
+
+
+def apply(*, source: str) -> str:
+    """
+    Apply automatic code formatting to source code.
+
+    Args:
+        source: The source code.
+
+    Returns:
+        The formatted source code.
+
+    """
+    try:
+        return black.format_file_contents(
+            src_contents=source, fast=False, mode=black.FileMode()
+        )
+    except black.NothingChanged:
+        return source

open_alchemy/models_file/__init__.py

@@ -47,4 +47,5 @@ def generate_models(self) -> str:
             model_sources.append(model_source)
 
         # Generate source code for models file
-        return _models.generate(models=model_sources)
+        raw_source = _models.generate(models=model_sources)
+        return facades.code_formatter.apply(source=raw_source)

setup.cfg

@@ -21,6 +21,7 @@ markers =
     init
     slow
     schemas
+    code_formatter
 python_functions = test_*
 mocked-sessions = examples.app.database.db.session
 

setup.py

@@ -35,6 +35,7 @@
         "jsonschema>=3",
         "Jinja2>=2",
         "sqlalchemy-stubs>=0.3",
+        "black",
     ],
     include_package_data=True,
     extras_require={

tests/open_alchemy/facades/test_code_formatter.py

@@ -0,0 +1,25 @@
+"""Tests for code formatting."""
+
+import pytest
+
+from open_alchemy import facades
+
+
+@pytest.mark.parametrize(
+    "source, expected_source",
+    [
+        pytest.param("'test'\n", '"test"\n', id="apply formatting"),
+        pytest.param('"test"\n', '"test"\n', id="no formatting"),
+    ],
+)
+@pytest.mark.facade
+@pytest.mark.code_formatter
+def test_apply(source, expected_source):
+    """
+    GIVEN source code and expected source code
+    WHEN apply is called with the source code
+    THEN the expected source code is returned.
+    """
+    returned_source = facades.code_formatter.apply(source=source)
+
+    assert returned_source == expected_source

tests/open_alchemy/integration/test_init.py

@@ -577,7 +577,7 @@ class TTable({expected_model_base}):
     query: orm.Query
 
     # Model properties
-    column: 'sqlalchemy.Column[typing.Optional[int]]'
+    column: "sqlalchemy.Column[typing.Optional[int]]"
 
     def __init__(self, column: typing.Optional[int] = None) -> None:
         """

tests/open_alchemy/models_file/test_integration.py

@@ -1,7 +1,5 @@
 """Tests for models file."""
 
-# pylint: disable=line-too-long
-
 import sys
 
 import pytest
@@ -59,7 +57,7 @@ class TModel({_EXPECTED_MODEL_BASE}):
     query: orm.Query
 
     # Model properties
-    id: 'sqlalchemy.Column[typing.Optional[int]]'
+    id: "sqlalchemy.Column[typing.Optional[int]]"
 
     def __init__(self, id: typing.Optional[int] = None) -> None:
         """
@@ -157,7 +155,7 @@ class TModel1({_EXPECTED_MODEL_BASE}):
     query: orm.Query
 
     # Model properties
-    id: 'sqlalchemy.Column[typing.Optional[int]]'
+    id: "sqlalchemy.Column[typing.Optional[int]]"
 
     def __init__(self, id: typing.Optional[int] = None) -> None:
         """
@@ -239,7 +237,7 @@ class TModel2({_EXPECTED_MODEL_BASE}):
     query: orm.Query
 
     # Model properties
-    id: 'sqlalchemy.Column[typing.Optional[str]]'
+    id: "sqlalchemy.Column[typing.Optional[str]]"
 
     def __init__(self, id: typing.Optional[str] = None) -> None:
         """
@@ -366,7 +364,7 @@ class TModel({_EXPECTED_MODEL_BASE}):
     query: orm.Query
 
     # Model properties
-    id: 'sqlalchemy.Column[typing.Optional[int]]'
+    id: "sqlalchemy.Column[typing.Optional[int]]"
 
     def __init__(self, id: typing.Optional[int] = None) -> None:
         """
@@ -480,10 +478,12 @@ class TModel({_EXPECTED_MODEL_BASE}):
     query: orm.Query
 
     # Model properties
-    name: 'sqlalchemy.Column[typing.Optional[str]]'
-    id: 'sqlalchemy.Column[typing.Optional[int]]'
+    name: "sqlalchemy.Column[typing.Optional[str]]"
+    id: "sqlalchemy.Column[typing.Optional[int]]"
 
-    def __init__(self, name: typing.Optional[str] = None, id: typing.Optional[int] = None) -> None:
+    def __init__(
+        self, name: typing.Optional[str] = None, id: typing.Optional[int] = None
+    ) -> None:
         """
         Construct.
 
@@ -495,7 +495,9 @@ def __init__(self, name: typing.Optional[str] = None, id: typing.Optional[int] =
         ...
 
     @classmethod
-    def from_dict(cls, name: typing.Optional[str] = None, id: typing.Optional[int] = None) -> "TModel":
+    def from_dict(
+        cls, name: typing.Optional[str] = None, id: typing.Optional[int] = None
+    ) -> "TModel":
         """
         Construct from a dictionary (eg. a POST payload).
 
@@ -542,7 +544,9 @@ def to_str(self) -> str:
 
 
 Model: typing.Type[TModel] = models.Model  # type: ignore
-'''  # noqa: E501
+'''
+    print(repr(source))
+    print(repr(expected_source))
     assert source == expected_source
 
 
@@ -593,7 +597,7 @@ class TModel({_EXPECTED_MODEL_BASE}):
     query: orm.Query
 
     # Model properties
-    id: 'sqlalchemy.Column[typing.Optional[int]]'
+    id: "sqlalchemy.Column[typing.Optional[int]]"
 
     def __init__(self, id: typing.Optional[int] = None) -> None:
         """