Coverage for pyquickhelper/loghelper/os_helper.py: 60%

1""" 

2@file 

3@brief Helpers for module :epkg:`*py:os`. 

4""" 

5import os 

6 

7 

8def get_machine(): 

9 """ 

10 Returns the machine name. 

11 

12 @return machine name 

13 

14 The method assumes environment variable ``COMPUTERNAME``, 

15 ``HOSTNAME`` or ``NAME`` is available. 

16 Otherwise, you should use module ``platform``. 

17 """ 

18 name = os.environ.get( 

19 "COMPUTERNAME", 

20 os.environ.get( 

21 "HOSTNAME", 

22 os.environ.get( 

23 "NAME", 

24 os.environ.get( 

25 "TRAVIS_OS_NAME", 

26 os.environ.get( 

27 "AGENT_MACHINENAME", 

28 None))))) 

29 if name is None: 

30 raise ValueError("Unable to find machine name in {0}".format( 

31 ",".join(sorted(os.environ.keys())))) 

32 return name 

33 

34 

35def get_user(): 

36 """ 

37 Returns the user name. 

38 

39 @return user name 

40 

41 The method assumes environment variable ``USERNAME`` or ``USER`` 

42 is available. 

43 """ 

44 name = os.environ.get("USERNAME", os.environ.get( 

45 "USER", os.environ.get("CIRCLE_USERNAME", None))) 

46 if name is None: 

47 raise ValueError( # pragma: no cover 

48 f"Unable to find user name in {','.join(sorted(os.environ.keys()))}") 

49 return name