misc: add get_current_user_info function
This commit is contained in:
parent
1b296026c6
commit
8fa54b3eaf
|
@ -1,17 +1,19 @@
|
||||||
import hashlib, platform, random, signal, socket, statistics, string, time, timeit
|
import grp, hashlib, os, platform, random, signal, socket, statistics, string, time, timeit
|
||||||
|
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
from getpass import getpass
|
from getpass import getpass, getuser
|
||||||
from importlib import util
|
from importlib import util
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
from . import izzylog
|
from . import izzylog
|
||||||
|
from .dotdict import DotDict
|
||||||
|
|
||||||
|
|
||||||
__all__ = [
|
__all__ = [
|
||||||
'ap_date',
|
'ap_date',
|
||||||
'boolean',
|
'boolean',
|
||||||
'catch_kb_interrupt',
|
'catch_kb_interrupt',
|
||||||
|
'get_current_user_info',
|
||||||
'get_ip',
|
'get_ip',
|
||||||
'hasher',
|
'hasher',
|
||||||
'import_from_path',
|
'import_from_path',
|
||||||
|
@ -101,6 +103,28 @@ def catch_kb_interrupt(function, *args, **kwargs):
|
||||||
izzylog.verbose('Bye! UvU')
|
izzylog.verbose('Bye! UvU')
|
||||||
|
|
||||||
|
|
||||||
|
def get_current_user_info():
|
||||||
|
data = DotDict({
|
||||||
|
'name': getuser(),
|
||||||
|
'group': None,
|
||||||
|
'groups': None,
|
||||||
|
'uid': os.getuid(),
|
||||||
|
'gid': os.getgid(),
|
||||||
|
'gids': os.getgroups()
|
||||||
|
})
|
||||||
|
|
||||||
|
try:
|
||||||
|
data.gids.remove(data.gid)
|
||||||
|
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
|
||||||
|
data.group = grp.getgrgid(data.gid).gr_name
|
||||||
|
data.groups = [grp.getgrgid(v).gr_name for v in data.gids]
|
||||||
|
|
||||||
|
return data
|
||||||
|
|
||||||
|
|
||||||
def get_ip():
|
def get_ip():
|
||||||
'''
|
'''
|
||||||
Get the IP address of the machine
|
Get the IP address of the machine
|
||||||
|
|
Loading…
Reference in a new issue