add compiled check

pydantic · samuelcolvin · May 30, 2019 · May 25, 2019 · May 25, 2019 · May 25, 2019
commit 0a4e37a375d7fb0cc6700f8bc34d96e3bbb99259
diff --git a/.travis.yml b/.travis.yml
@@ -24,8 +24,10 @@ script:
 
 # test with and without ujson and email-validator then combine coverage
 - make test && mv .coverage .coverage.extra-cython
+- python -c "import sys, pydantic; print('compiled:', pydantic.compiled); sys.exit(not pydantic.compiled)"
 - pip uninstall -y cython
 - make clean
+- python -c "import sys, pydantic; print('compiled:', pydantic.compiled); sys.exit(pydantic.compiled)"
 - make test && mv .coverage .coverage.no-cython
 - pip uninstall -y ujson email-validator
 - make test && mv .coverage .coverage.no-extra

diff --git a/pydantic/__init__.py b/pydantic/__init__.py
@@ -6,7 +6,7 @@
 from .errors import *
 from .fields import Required
 from .main import BaseConfig, BaseModel, Extra, create_model
-from .parse import parse_model
+from .parse import compiled, parse_model
 from .parse_files import Protocol
 from .schema import Schema
 from .types import *

diff --git a/pydantic/parse.py b/pydantic/parse.py
@@ -14,6 +14,17 @@
     SetStr = Set[str]
 
 
+__all__ = ['Extra', 'compiled', 'parse_model']
+
+
+try:
+    import cython
+except ImportError:
+    compiled = False
+else:
+    compiled = cython.compiled
+
+
 class Extra(str, Enum):
     allow = 'allow'
     ignore = 'ignore'