108 lines
2.9 KiB
Python
108 lines
2.9 KiB
Python
#!/usr/bin/env python3
|
|
|
|
|
|
import logging
|
|
from datetime import datetime
|
|
from getpass import getuser
|
|
from os import path, sep, chdir, devnull
|
|
from sys import platform
|
|
|
|
|
|
def execquite(code: int) -> None:
|
|
"""Exit procedure.
|
|
|
|
Args:
|
|
code (int): exitcode
|
|
"""
|
|
if args['show'] != 'qn':
|
|
input("Press [ENTER] to continue...")
|
|
global time_start
|
|
time_execute = datetime.now() - time_start
|
|
logging.info(msg='execution time is ' + str(time_execute) + '. Exit.')
|
|
exit(code)
|
|
|
|
|
|
def getconfig(config: str) -> dict:
|
|
"""Simple config reader.
|
|
|
|
Args:
|
|
config (str): custom configuration file path
|
|
|
|
Returns:
|
|
dict: dictionary as "key":"value"
|
|
"""
|
|
dictionary = {}
|
|
dictionary['log_root'] = None
|
|
if path.exists(config):
|
|
with open(config) as file:
|
|
raw = file.read()
|
|
for line in raw.splitlines():
|
|
if not line.lstrip().startswith('#') and "=" in line:
|
|
if "log_root=" in line:
|
|
dictionary['log_root'] = line.split('=')[1].strip()
|
|
return dictionary
|
|
|
|
|
|
def checkroot() -> bool:
|
|
"""Crossplatform privileged rights checker.
|
|
|
|
Returns:
|
|
bool: True - if privileged rights, False - if not privileged rights
|
|
"""
|
|
if platform.startswith('linux') or platform.startswith('darwin'):
|
|
from os import geteuid
|
|
if geteuid() != 0:
|
|
return False
|
|
else:
|
|
return True
|
|
elif platform.startswith('win32'):
|
|
import ctypes
|
|
return ctypes.windll.shell32.IsUserAnAdmin()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
from argparse import ArgumentParser
|
|
|
|
global time_start
|
|
time_start = datetime.now()
|
|
|
|
args = ArgumentParser(
|
|
prog='script',
|
|
description='Checking privileged rights',
|
|
epilog='Dependencies: '
|
|
'- Python 3 (tested version 3.9.5)'
|
|
)
|
|
args.add_argument('-s', '--show', type=str, default=None, required=False,
|
|
help='qn - execution without pauses')
|
|
args.add_argument('-c', '--conf', type=str, default=None, required=False,
|
|
help='path to configuration file')
|
|
args = vars(args.parse_args())
|
|
|
|
conf = path.abspath(__file__).replace('.py', '.conf')
|
|
if args['conf']:
|
|
conf = args['conf']
|
|
conf = getconfig(conf)
|
|
log_path = devnull
|
|
if conf['log_root']:
|
|
log_path = conf['log_root'] + sep + path.splitext(path.basename(__file__))[0] + '.log'
|
|
logging.basicConfig(
|
|
format='%(asctime)s %(levelname)s: %(message)s',
|
|
datefmt='%Y-%m-%d_%H.%M.%S',
|
|
handlers=[
|
|
logging.FileHandler(
|
|
filename=log_path,
|
|
mode='a'
|
|
),
|
|
logging.StreamHandler()
|
|
],
|
|
level='INFO'
|
|
)
|
|
|
|
chdir(path.split(path.abspath(__file__))[0])
|
|
if checkroot():
|
|
print('Running as', getuser())
|
|
execquite(0)
|
|
else:
|
|
logging.warning(msg='Restart this as root!')
|
|
execquite(1)
|