generated from pavel.muhortov/template-python
clean commit
This commit is contained in:
parent
b26d21152e
commit
2a576407c1
43
README.md
43
README.md
|
@ -1,42 +1,3 @@
|
|||
# template-python
|
||||
# my_route
|
||||
|
||||
Template repository for projects on python
|
||||
|
||||
* [`script.py`](https://git.hmp.today/pavel.muhortov/template-python#script-py)
|
||||
|
||||
____
|
||||
|
||||
## `script.py`
|
||||
|
||||
**Description:**
|
||||
> returning current username if privileged rights are exist
|
||||
> or
|
||||
> returning error, if privileged rights are not exist
|
||||
|
||||
**Dependencies:**
|
||||
>
|
||||
> * Python 3 (tested version 3.9.5)
|
||||
|
||||
| PARAMETERS | DESCRIPTION | DEFAULT|
|
||||
|-------------|-------------|--------|
|
||||
|**[-s,--show]**|"" - execution with pauses.<br/>"qn" - execution without pauses.||
|
||||
|**[-c,--conf]**|path to configuration file|`./script.conf`|
|
||||
|
||||
Example usage in terminal with Python on Linux:
|
||||
|
||||
```shell
|
||||
python3 ./script.py
|
||||
```
|
||||
|
||||
Example usage in terminal with make the script executable on Linux:
|
||||
|
||||
```shell
|
||||
chmod u+x ./script.py
|
||||
script.py -s qn -c ./script.conf
|
||||
```
|
||||
|
||||
Example usage in terminal with Python on Windows:
|
||||
|
||||
```shell
|
||||
python .\script.py
|
||||
```
|
||||
Route manager by CIDR lists
|
||||
|
|
107
script.py
107
script.py
|
@ -1,107 +0,0 @@
|
|||
#!/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)
|
Loading…
Reference in New Issue
Block a user