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='')),