From a071ad17652d2e43eeef889765f032d789dc16bb Mon Sep 17 00:00:00 2001
From: Qi Zhao <zhaoqi99@outlook.com>
Date: Tue, 5 Nov 2024 17:57:19 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=F0=9F=90=9B=20add=20ENCRYPT=5FSUFFIX=20?=
 =?UTF-8?q?for=20`inspect.getmodulename`?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pyencrypt/loader.py | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/pyencrypt/loader.py b/pyencrypt/loader.py
index cefe5b1..56f78a7 100644
--- a/pyencrypt/loader.py
+++ b/pyencrypt/loader.py
@@ -2,7 +2,7 @@
 import sys
 import traceback
 import types
-from importlib import abc
+from importlib import abc, machinery
 from importlib._bootstrap_external import _NamespacePath
 from importlib.machinery import ModuleSpec
 from importlib.util import spec_from_loader
@@ -21,6 +21,9 @@ def __dir__(self) -> Iterable[str]:
         return []
 
 
+ENCRYPT_SUFFIX = ".pye"
+
+
 class EncryptFileLoader(abc.SourceLoader, Base):
     POSSIBLE_PATH = [
         Path(os.path.expanduser("~")) / ".licenses" / "license.lic",
@@ -79,12 +82,17 @@ def find_spec(
     ) -> ModuleSpec:
         if path:
             if isinstance(path, _NamespacePath):
-                file_path = Path(path._path[0]) / f'{fullname.rsplit(".", 1)[-1]}.pye'
+                file_path = (
+                    Path(path._path[0])
+                    / f'{fullname.rsplit(".", 1)[-1]}{ENCRYPT_SUFFIX}'
+                )
             else:
-                file_path = Path(path[0]) / f'{fullname.rsplit(".", 1)[-1]}.pye'
+                file_path = (
+                    Path(path[0]) / f'{fullname.rsplit(".", 1)[-1]}{ENCRYPT_SUFFIX}'
+                )
         else:
             for p in sys.path:
-                file_path = Path(p) / f"{fullname}.pye"
+                file_path = Path(p) / f"{fullname}{ENCRYPT_SUFFIX}"
                 if file_path.exists():
                     break
         file_path = file_path.absolute().as_posix()
@@ -99,4 +107,5 @@ def invalidate_caches(cls):
 
 
 # TODO: generate randomly AES Class
+machinery.EXTENSION_SUFFIXES.append(ENCRYPT_SUFFIX)
 sys.meta_path.insert(0, EncryptFileFinder)