From 4eacee8cad96f4756a71313b4d67980d84f00357 Mon Sep 17 00:00:00 2001
From: InsanePrawn <insane.prawny@gmail.com>
Date: Mon, 15 Aug 2022 22:37:19 +0200
Subject: [PATCH] utils: add get_{user,group}_name() and get_{uid,gid}()

---
 utils.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/utils.py b/utils.py
index 3237006..7ddece8 100644
--- a/utils.py
+++ b/utils.py
@@ -1,5 +1,7 @@
 import atexit
+import grp
 import logging
+import pwd
 import subprocess
 from shutil import which
 from typing import Optional, Union, Sequence
@@ -72,3 +74,27 @@ def log_or_exception(raise_exception: bool, msg: str, exc_class=Exception, log_l
         raise exc_class(msg)
     else:
         logging.log(log_level, msg)
+
+
+def get_user_name(uid: Union[str, int]) -> str:
+    if isinstance(uid, int) or uid.isnumeric():
+        return pwd.getpwuid(int(uid)).pw_name
+    return uid
+
+
+def get_group_name(gid: Union[str, int]) -> str:
+    if isinstance(gid, int) or gid.isnumeric():
+        return grp.getgrgid(int(gid)).gr_name
+    return gid
+
+
+def get_uid(user: Union[int, str]) -> int:
+    if isinstance(user, int) or user.isnumeric():
+        return int(user)
+    return pwd.getpwnam(user).pw_uid
+
+
+def get_gid(group: Union[int, str]) -> int:
+    if isinstance(group, int) or group.isnumeric():
+        return int(group)
+    return grp.getgrnam(group).gr_gid