Jerry/pacroller
1
0
Fork 0
mirror of https://github.com/isjerryxiao/pacroller.git synced 2024-12-23 06:39:24 +08:00
Code Issues Projects Releases Packages Wiki Activity

add root check

Browse source
This commit is contained in:
JerryXiao 2021-01-16 00:12:17 +08:00
parent 57e96c2650
commit 7083d4b41b
Signed by: Jerry
GPG key ID: 9D9CE43650FF2BAA
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/pacroller/main.py
View file

@ -5,7 +5,7 @@ import subprocess
import logging
from re import match
import json
from os import environ
from os import environ, getuid
import traceback
import pyalpm
import pycman
@ -169,6 +169,8 @@ def main() -> None:
logging.basicConfig(level=logging.INFO, format='%(levelname)s - %(message)s')
if args.action == 'run':
if getuid() != 0:
parser.error('you need to be root')
if prev_err := has_previous_error():
logger.error(f'Cannot continue, a previous error {prev_err} is still present. Please resolve this issue and run fail-reset.')
else:
@ -199,6 +201,8 @@ def main() -> None:
break
print()
elif args.action == 'fail-reset':
if getuid() != 0:
parser.error('you need to be root')
if prev_err := has_previous_error():
write_db(None)
logger.info(f'reset previous error {prev_err}')