Fix mypy plugin to work from other projects

jacebrowning · Jun 10, 2019 · be60ec4 · be60ec4
1 parent b28d906
commit be60ec4
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/.coveragerc b/.coveragerc
@@ -8,6 +8,7 @@ omit =
     .venv/*
     */tests/*
     */__main__.py
+    datafiles/plugins.py
 
 [report]
 

diff --git a/datafiles/plugins.py b/datafiles/plugins.py
@@ -12,7 +12,7 @@ class DatafilesPlugin(Plugin):
     def get_class_decorator_hook(
         self, fullname: str
     ) -> Optional[Callable[[ClassDefContext], None]]:
-        if fullname == 'datafiles.decorators.datafile':
+        if fullname.endswith('.datafile'):
             return datafile_class_maker_callback
         return None
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,7 +1,7 @@
 [tool.poetry]
 
 name = "datafiles"
-version = "0.4b2"
+version = "0.4b3"
 description = "File-based ORM for dataclasses."
 
 license = "MIT"