Accept file path or file object

Fatal1ty · Oct 18, 2024 · 3cd2dab · 3cd2dab
1 parent 77ac313
commit 3cd2dab
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/aioapns/connection.py b/aioapns/connection.py
@@ -525,7 +525,7 @@ async def _create_connection(
 class APNsKeyConnectionPool(APNsBaseConnectionPool):
     def __init__(
         self,
-        key_file: str,
+        key_file: str,  # Key path Str or File object
         key_id: str,
         team_id: str,
         topic: str,
@@ -550,6 +550,8 @@ def __init__(
         self.key_id = key_id
         self.team_id = team_id
 
+        if hasattr(key_file, "read"):
+            self.key = key_file.read()
         with open(key_file) as f:
             self.key = f.read()