fix mypy

Author: collerek

sqlalchemy_to_ormar/__init__.py

@@ -1,5 +1,5 @@
 from .main import ormar_model_str_repr, sqlalchemy_to_ormar
 
-__version__ = "0.0.1"
+__version__ = "0.0.2"
 
 __all__ = ["sqlalchemy_to_ormar", "ormar_model_str_repr"]

sqlalchemy_to_ormar/main.py

@@ -69,7 +69,7 @@ def _update_refs_in_related(model: Type[Model]):
                     **{k.__name__: v for k, v in PARSED_MODELS.items()}
                 )
             else:
-                if target.__forward_arg__ == model.__name__:
+                if target.__forward_arg__ == model.__name__:  # type: ignore
                     model.update_forward_refs(**{model.__name__: model})
 
 
@@ -129,7 +129,7 @@ def _extract_relations(
                     )
                     target = PARSED_MODELS[target_sqlalchemy]
                 else:
-                    target = ForwardRef(target_sqlalchemy.__name__)
+                    target = ForwardRef(target_sqlalchemy.__name__)  # type: ignore
 
                 column = next(iter(attr.local_columns))
                 sql_fk = next(iter(column.foreign_keys))

tests/test_loop_in_models.py

@@ -28,7 +28,7 @@ class User(Base):
     USERNAME = Column(String(255), index=True)
     PASSWORD = Column(String(40))
     EMAIL = Column(String(255))
-    CUSTOMER_ID = Column(ForeignKey("customer.CUSTOMER_ID"), index=True)
+    CUSTOMER_ID = Column(ForeignKey("customer.CUSTOMER_ID"), index=True)  # type: ignore
     user_customer = relationship(
         "Customer", primaryjoin="User.CUSTOMER_ID == Customer.CUSTOMER_ID"
     )
@@ -42,7 +42,7 @@ class Customer(Base):
     ORGNO = Column(String(20))
     TYPE = Column(Integer())
     STATUS = Column(Integer())
-    SELLER_ID = Column(ForeignKey("user.USER_ID"), index=True)
+    SELLER_ID = Column(ForeignKey("user.USER_ID"), index=True)  # type: ignore
     PHONE = Column(String(20))
     FAX = Column(String(20))
     seller = relationship("User", primaryjoin="Customer.SELLER_ID == User.USER_ID")

tests/test_self_relation.py

@@ -28,7 +28,7 @@ class User(Base):
     USERNAME = Column(String(255), index=True)
     PASSWORD = Column(String(40))
     EMAIL = Column(String(255))
-    PARENT_ID = Column(ForeignKey("user.USER_ID"), index=True)
+    PARENT_ID = Column(ForeignKey("user.USER_ID"), index=True)  # type: ignore
     parent = relationship("User", remote_side=[USER_ID])