From 555b1661c15c76878b112b0475d2fac841fca16e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Alejandro=20Avil=C3=A9s?= <ome@unconventional.dev> Date: Tue, 19 Sep 2023 17:24:58 +0200 Subject: [PATCH] Add support for .pyi files --- CHANGELOG.md | 1 + src/unbeheader/__init__.py | 3 +++ 2 files changed, 4 insertions(+) diff --git a/CHANGELOG.md b/CHANGELOG.md index 5f5fa4a..deb6f25 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -4,6 +4,7 @@ - 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='')),