diff --git a/CHANGELOG.md b/CHANGELOG.md index 0063a54..d09db00 100644 --- 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 index b1f586a..1e4c39c 100644 --- 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='')),