Add support for .pyi files

unconventionaldotdev · Sep 19, 2023 · 9fe17bd · 9fe17bd
1 parent cfa80de
commit 9fe17bd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -5,6 +5,7 @@
 - Renamed CLI `--ci` flag to `--check`.
 - Removed Unicode interpuct from CLI output.
 - Removed colors from CLI output when CI envvar is set.
+- Added support for `.pyi` files.
 
 ## v1.1.1
 

diff --git a/src/unbeheader/__init__.py b/src/unbeheader/__init__.py
@@ -10,6 +10,9 @@
     'py': SupportedFileType(
         re.compile(r'((^#|[\r\n]#).*)*'),
         CommentSkeleton(comment_start='#', comment_middle='#', comment_end='')),
+    'pyi': SupportedFileType(
+        re.compile(r'((^#|[\r\n]#).*)*'),
+        CommentSkeleton(comment_start='#', comment_middle='#', comment_end='')),
     'wsgi': SupportedFileType(
         re.compile(r'((^#|[\r\n]#).*)*'),
         CommentSkeleton(comment_start='#', comment_middle='#', comment_end='')),