Small cleanup based on python-babel#45

EnTeQuAk · May 7, 2019 · dbc1137 · dbc1137
1 parent c60a307
commit dbc1137
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/django_babel/extract.py b/django_babel/extract.py
@@ -1,10 +1,10 @@
 # -*- coding: utf-8 -*-
+from django.template.base import Lexer
 try:
-    from django.template.base import Lexer, TOKEN_TEXT, TOKEN_VAR, TOKEN_BLOCK
-except ImportError:
-    # Django 2.1+
-    from django.template.base import Lexer, TokenType
-
+    from django.template.base import TokenType
+except ImportError:  # django < 2.1
+    from django.template.base import TOKEN_TEXT, TOKEN_VAR, TOKEN_BLOCK
+else:
     TOKEN_TEXT = TokenType.TEXT
     TOKEN_VAR = TokenType.VAR
     TOKEN_BLOCK = TokenType.BLOCK