Add re import when using regex pattern

horejsek · Nov 27, 2024 · 52458fe · 52458fe
1 parent b0e2bcc
commit 52458fe
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/CHANGELOG.txt b/CHANGELOG.txt
@@ -4,6 +4,7 @@
 * Fixed detecting when infinity is reached with multipleOf
 * Fixed that min/max items/lenght/properties can be float
 * Fixed that everything with empty `not` is invalid
+* Fixed missing re import when using regex pattern
 * Improved regexp for email format to comfort test suite
 * Improved regexp for date format to comfort test suite
 * Improved regexp for ipv4 format to comfort test suite

diff --git a/fastjsonschema/draft04.py b/fastjsonschema/draft04.py
@@ -271,6 +271,7 @@ def generate_format(self):
                 self._generate_format(format_, format_ + '_re_pattern', format_regex)
             # Format regex is used only in meta schemas.
             elif format_ == 'regex':
+                self._extra_imports_lines = ['import re'] 
                 with self.l('try:', optimize=False):
                     self.l('re.compile({variable})')
                 with self.l('except Exception:'):