change postgres check to dialect name

2020-10-28 13:57:39 +01:00
parent 250f55ad68
commit d18e79a83e
2 changed files with 5 additions and 2 deletions
--- a/ormar/models/newbasemodel.py
+++ b/ormar/models/newbasemodel.py
@ -166,6 +166,10 @@ class NewBaseModel(pydantic.BaseModel, ModelTableProxy, metaclass=ModelMetaclass
    def pk_type(cls) -> Any:
        return cls.Meta.model_fields[cls.Meta.pkname].__type__

+    @classmethod
+    def db_backend_name(cls) -> str:
+        return cls.Meta.database._backend._dialect.name
+
    def remove(self, name: "Model") -> None:
        self._orm.remove_parent(self, name)