Merge pull request #24 from codectl/feat/#23

ensure no error when OS is not supported by pwd

.github/workflows/ci.yaml

@@ -29,7 +29,7 @@ jobs:
     continue-on-error: ${{ matrix.experimental }}
     strategy:
       matrix:
-        os: [ ubuntu-latest, macos-latest ]
+        os: [ ubuntu-latest, macos-latest, windows-latest ]
         python-version: [ "3.8", "3.9", "3.10", "3.11" ]
         experimental: [ false ]
 

src/impersonation/core.py

@@ -2,6 +2,7 @@
 import os
 import re
 import sys
+import warnings
 from multiprocessing import Pipe, get_context
 
 from impersonation import utils
@@ -55,8 +56,12 @@ def decorator(fn):
 
         @functools.wraps(fn)
         def wrapper(*args, **kwargs):
-            # call wrapped function if no user impersonation
-            if not username or str(username) == utils.system_username():
+            # call wrapped function if user cannot be impersonated
+            try:
+                if not username or str(username) == utils.pw_username():
+                    return fn(*args, **kwargs)
+            except OSError:
+                warnings.warn("OS not supported: decorated fn cannot be impersonated.")
                 return fn(*args, **kwargs)
 
             cls_name = fn.__qualname__.split(".")[0]

src/impersonation/utils.py

@@ -1,11 +1,25 @@
 import os
-import pwd
 
 
-def pw_pair(username):
-    pwnam = pwd.getpwnam(username)
-    return pwnam.pw_uid, pwnam.pw_gid
+__all__ = (
+    "pw_username",
+    "pw_pair",
+)
+
+
+def pwd():
+    try:
+        import pwd
+
+        return pwd
+    except ImportError:
+        raise OSError
 
 
-def system_username():
-    return pwd.getpwuid(os.getuid()).pw_name
+def pw_username():
+    return pwd().getpwuid(os.getuid()).pw_name
+
+
+def pw_pair(username):
+    pwnam = pwd().getpwnam(username)
+    return pwnam.pw_uid, pwnam.pw_gid