fix bug introduced by psf/black#4270 where black.assert_equivalent …

…no longer raises an AssertionError
spyoungtech · May 11, 2024 · 1753204 · 1753204
1 parent 6bbcae5
commit 1753204
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/.unasync-rewrite.py b/.unasync-rewrite.py
@@ -9,6 +9,11 @@
 
 changes = 0
 
+if hasattr(black, 'ASTSafetyError'):
+    exceptions = (AssertionError, black.ASTSafetyError)
+else:
+    exceptions = (AssertionError,)
+
 
 def _copyfunc(src, dst, *, follow_symlinks=True):
     global changes
@@ -22,7 +27,7 @@ def _copyfunc(src, dst, *, follow_symlinks=True):
                 src=contents,
                 dst=dst_contents,
             )
-        except AssertionError:
+        except exceptions:
             changes += 1
             print('MODIFIED', dst)
             shutil.copy2(src, dst, follow_symlinks=follow_symlinks)