import logging
import os
import sys
from io import TextIOWrapper, BufferedRandom, BufferedRWPair, BufferedWriter, IOBase
from pwd import getpwnam
from typing import Union, Tuple, Dict
from temci.utils.settings import Settings
[docs]def get_bench_user() -> str:
user = Settings()["env"]["USER"]
return os.getenv("USER", "") if user == "" else user
[docs]def bench_as_different_user() -> bool:
return get_bench_user() != os.getenv("USER", get_bench_user())
[docs]def get_bench_uid_and_gid() -> Tuple[int, int]:
pwnam = getpwnam(get_bench_user())
return pwnam.pw_uid, pwnam.pw_gid
_logged_chown_error = False
[docs]def chown(path: Union[str, TextIOWrapper, BufferedRandom, BufferedRWPair, BufferedWriter, IOBase]):
if not bench_as_different_user():
return
if isinstance(path, IOBase) and path.isatty():
return
if not isinstance(path, str):
return chown(path.name)
ids = ()
try:
ids = get_bench_uid_and_gid()
except:
global _logged_chown_error
if not _logged_chown_error:
logging.warn("Could not get user id for user {} therefore no chowning possible".format(get_bench_user()))
_logged_chown_error = True
return
try:
os.chown(path, *ids)
except FileNotFoundError:
pass
[docs]def get_env_setting() -> Dict[str, str]:
env = Settings()["env"].copy()
if env["USER"] == "":
env["USER"] = get_bench_user()
if env["PATH"] == "":
env["PATH"] = os.getenv("PATH", "")
return env
if __name__ == '__main__':
chown(sys.stdin)
