proto in mixin

fizyk · Jan 8, 2021 · e08982c · e08982c
1 parent 6b66abb
commit e08982c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/pyramid_basemodel/mixin.py b/src/pyramid_basemodel/mixin.py
@@ -9,9 +9,14 @@
 ]
 
 import logging
+import sys
 
 from datetime import datetime
-from typing import Callable, Protocol, List, Union, Tuple
+from typing import Callable, List, Union, Tuple
+if sys.version_info[0] == 3 and sys.version_info[1] <= 7:
+    from typing_extensions import Protocol
+else:
+    from typing import Protocol
 
 from sqlalchemy import Column
 from sqlalchemy import Unicode