p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

Fix DUO116 warnings in monkey/infection_monkey/utils/linux/users.py

This commit is contained in:
Shreya 2021-04-22 19:37:25 +05:30 committed by Mike Salvatore
parent c0fdc9561f
commit 6b467fd20b
1 changed files with 6 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
monkey/infection_monkey/utils/linux/users.py
View File

@ -1,5 +1,6 @@
import datetime
import logging
import shlex
import subprocess
from infection_monkey.utils.auto_new_user import AutoNewUser
@ -42,18 +43,16 @@ class AutoNewLinuxUser(AutoNewUser):
logger.debug(
"Trying to add {} with commands {}".format(self.username, str(commands_to_add_user))
)
_ = subprocess.check_output(
" ".join(commands_to_add_user), stderr=subprocess.STDOUT, shell=True
)
_ = subprocess.check_output(commands_to_add_user, stderr=subprocess.STDOUT)
def __enter__(self):
return self # No initialization/logging on needed in Linux
def run_as(self, command):
command_as_new_user = "sudo -u {username} {command}".format(
username=self.username, command=command
command_as_new_user = shlex.split(
"sudo -u {username} {command}".format(username=self.username, command=command)
)
return subprocess.call(command_as_new_user, shell=True)
return subprocess.call(command_as_new_user)
def __exit__(self, exc_type, exc_val, exc_tb):
# delete the user.
@ -63,6 +62,4 @@ class AutoNewLinuxUser(AutoNewUser):
self.username, str(commands_to_delete_user)
)
)
_ = subprocess.check_output(
" ".join(commands_to_delete_user), stderr=subprocess.STDOUT, shell=True
)
_ = subprocess.check_output(commands_to_delete_user, stderr=subprocess.STDOUT)