Merge pull request #2 from selobu/2023oct0301

refactor: ♻️ included walrus operator to simplify code
selobu · Oct 3, 2023 · 4150b99 · 4150b99
2 parents 9111bb4 + 22df1ea
commit 4150b99
Show file tree

Hide file tree

Showing 11 changed files with 7 additions and 5 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1 +1,2 @@
-/.pytest_cache
+/.pytest_cache
+**/__pycache__/
diff --git a/pyd_sqa/__pycache__/__init__.cpython-310.pyc b/pyd_sqa/__pycache__/__init__.cpython-310.pyc
diff --git a/pyd_sqa/__pycache__/__init__.cpython-39.pyc b/pyd_sqa/__pycache__/__init__.cpython-39.pyc
diff --git a/pyd_sqa/__pycache__/main.cpython-310.pyc b/pyd_sqa/__pycache__/main.cpython-310.pyc
diff --git a/pyd_sqa/__pycache__/main.cpython-39.pyc b/pyd_sqa/__pycache__/main.cpython-39.pyc
diff --git a/pyd_sqa/main.py b/pyd_sqa/main.py
@@ -15,10 +15,11 @@ def pyd_sqa(
     mapper = inspect(db_model)
     fields = {}
     for attr in mapper.attrs:
-        if not isinstance(attr, ColumnProperty) or not attr.columns:
-            continue
-        name = attr.key
-        if name in exclude:
+        if (
+            not isinstance(attr, ColumnProperty)
+            or not attr.columns
+            or (name := attr.key) in exclude
+        ):
             continue
         column = attr.columns[0]
         python_type = None

diff --git a/tests/__pycache__/__init__.cpython-310.pyc b/tests/__pycache__/__init__.cpython-310.pyc
diff --git a/tests/__pycache__/__init__.cpython-39.pyc b/tests/__pycache__/__init__.cpython-39.pyc
diff --git a/tests/__pycache__/test_pyd_sqa.cpython-39-pytest-5.4.3.pyc b/tests/__pycache__/test_pyd_sqa.cpython-39-pytest-5.4.3.pyc
diff --git a/tests/__pycache__/test_pydantic_sqlalchemy.cpython-310-pytest-7.4.0.pyc b/tests/__pycache__/test_pydantic_sqlalchemy.cpython-310-pytest-7.4.0.pyc
diff --git a/tests/__pycache__/test_pydantic_sqlalchemy.cpython-39-pytest-5.4.3.pyc b/tests/__pycache__/test_pydantic_sqlalchemy.cpython-39-pytest-5.4.3.pyc