Question

我试图使用python＆＃39; s subprocess调用带有相关参数的windows命令。命令正在执行，参数及其值看起来是正确的，但是当使用＆＃34;本地模式时，它似乎只能正常工作＆＃34; -l。我在使用远程模式时遇到invalid argument/option错误。可以指出我出错的地方吗？

有人能指出如何正确格式化subprocess.check_ouput()参数以包含执行脚本时在命令行中给出的变量吗？正如你所看到的，我尝试使用字符串格式化，无论是新旧还是尝试使它工作，因为我无法锻炼如何在过滤器（/ FI）参数值之间添加最后domain变量而不进行字符串格式化。

预期的命令行执行

tasklist /V /S 192.168.1.122 /U 'DOMAIN'\'USERNAME' /P 'PASSWORD' /FI "USERNAME eq 'DOMAIN'\*"

使用此命令行脚本示例：

hunter.py -d DOMAIN -u USERNAME -p PASSWORD -s servers.txt

这是错误：

ERROR: Invalid argument/option - '/S 192.168.1.122'.
Type "TASKLIST /?" for usage.

显然，这个论点是正确的，并且在视觉上是正确的。无论如何，这是任务列表的用法：

Description:
This tool displays a list of currently running processes on
either a local or remote machine.

Parameter List:
/S     system           Specifies the remote system to connect to.

/U     [domain\]user    Specifies the user context under which
                       the command should execute.

/P     [password]       Specifies the password for the given
                       user context. Prompts for input if omitted.

/M     [module]         Lists all tasks currently using the given
                        exe/dll name. If the module name is not
                        specified all loaded modules are displayed.

/SVC                    Displays services hosted in each process.

/APPS                   Displays Store Apps and their accociated processes.

/V                      Displays verbose task information.

/FI    filter           Displays a set of tasks that match a
                        given criteria specified by the filter.

/FO    format           Specifies the output format.
                        Valid values: "TABLE", "LIST", "CSV".

/NH                     Specifies that the "Column Header" should
                        not be displayed in the output.
                        Valid only for "TABLE" and "CSV" formats.
/?                      Displays this help message.

这是我到目前为止的python 代码;

#!/usr/bin/python

"""
Description:

Used for checking users logged into a list of servers.

Usage:
  hunter.py [-u <username>] [-p <password>] [-s <FILE>] (-d <domain>)
  hunter.py (-d <domain>) (-l)
  hunter.py -h | --help
  hunter.py --version

Options:
  -l --local
  -u --username
  -h --help     Show this screen.
  --version     Show version.
  -p --password
  -d --domain
  -s --serverfile=FILE
  """
from docopt import docopt
import subprocess
from subprocess import CalledProcessError

def tldomain(serverlist, domain, username, password):
    nlist = serverlist
    for serverl in nlist:
        try:
            print subprocess.check_output(["tasklist", "/V", "/S " + serverl, "/U" + domain, "\\" + username, "/P" + password, "/FI", "'USERNAME eq %s\\\*'"]) % domain
        except CalledProcessError as e:
            print(e.returncode)

def tllocal(domain):
        try:
            cmd = 'tasklist /V /FI "USERNAME eq {0}\\*"' .format(domain)
            subprocess.call(cmd)
        except OSError as e:
            print e

def getservers(servers):
        slist = open(servers).readlines()
        return [s.replace('\n', '') for s in slist]

if __name__ == "__main__":
    arguments = docopt(__doc__, version='0.1a')
    print arguments

    if (arguments['--local']) == False:
        serverlist = getservers(arguments['--serverfile'])
        tldomain(serverlist, arguments['<domain>'], arguments['<username>'], arguments['<password>'])

    else:
        tllocal(arguments['<domain>'])

Answer 1

将参数作为列表中的单独元素传递，并将字符串格式应用于最后一个元素，而不是subprocess.check_output()调用的输出：

print subprocess.check_output(
    ["tasklist", "/V", "/S", serverl, "/U", domain + "\\" + username,
     "/P", password, "/FI", "USERNAME eq %s\\*" % domain])

请注意，我还从最后一个参数中删除了'引用，将其保留到subprocess模块。

这也假设domain始终是非空字符串;如果情况并非如此，请使用：

if domain:
    domain += '\\'

print subprocess.check_output(
    ["tasklist", "/V", "/S", serverl, "/U", domain + username,
     "/P", password, "/FI", "USERNAME eq %s*" % domain])

e.g。实际指定\时仅使用domain反斜杠。

使用Python子进程和字符串格式

1 个答案: