python-daemon + argparse?

python-daemon + argparse?

我制作了一个脚本,它接受一些参数 运行。最初它可能 运行 自动使用参数 'auto',但我正在尝试将其守护进程以便它将 运行 具有指定参数的 运行 脚本作为守护进程.问题是 python-daemon 和 argparse 在决定谁解析什么时似乎相处不融洽。

parser = argparse.ArgumentParser(usage='pyfilter.py <file> <options> <actions>')
parser.add_argument('file', help='blacklist file containing IPs', type=str)

subparsers = parser.add_subparsers(help='help', dest='action')

parser_update = subparsers.add_parser('update', help='update help')
parser_update.add_argument('-c', '--check', help="check IPs for abuse reports", dest="check", type=str, nargs=1)

parser_blacklist = subparsers.add_parser('blacklist', help='create iptables rules for malicious IPs specified'
                                                           'in the provided file')
parser_clear = subparsers.add_parser('clear', help='clear iptables')

parser_auto = subparsers.add_parser('auto', help='automatically run update and blacklist on a loop')
parser_auto.add_argument('-i', '--interval', help='specify the loop interval', dest='interval', type=int, nargs=1)
parser_auto.add_argument('-c', '--check', help="check IPs for abuse reports", dest="check", type=str, nargs=1)

parser.add_argument('-p', '--port', help='specify the port to block', type=int)
parser.add_argument('-v', '--verbose', help='write output to screen', nargs=1)
args = parser.parse_args()

. . .

class pyfilterDaemon():
    def __init__(self):
        self.stdin_path = '/dev/null'
        self.stdout_path = '/dev/tty'
        self.stderr_path = '/dev/tty'
        self.pidfile_path = '/tmp/pyfilter.pid'
        self.pidfile_timeout = 5

    def run(self):
        while True:
            update()
            blacklist()
            time.sleep(interval)
. . .

def main():
    #handle()
    d = pyfilterDaemon()
    daemon_runner = runner.DaemonRunner(d)
    daemon_runner.start()

这是我尝试使用的命令:

root@tfof:~# ./pyfilter.py start
ERROR: File 'start' not found  # argparse parsed 'start' accidentally

root@tfof:~# ./pyfilter.py /etc/blacklist.lst -v yes auto
usage: checkfilter.py stop|restart|start  # now the argparse arguments are satisfied, but python-daemon is looking for its argument

root@tfof:~# ./pyfilter.py /etc/blacklist.lst -v yes auto start
usage: pyfilter.py <file> <options> <actions>
pyfilter.py: error: unrecognized arguments: start  # argparse is trying to parse 'start'

是否可以将 'start' 参数传递给 python-daemon 之类的?或者,如果我可以摆脱 argparsing 就好了,但是 'file' 参数是必须的。

Argparse 默认从 sys.argv 获取参数(参见 here)。 你在这里看到的行为发生并不奇怪, 因为您只需使用默认参数调用 parse_args 函数。 您可以将要解析的任何内容传递给它,而不是 sys.argv.

请参阅 示例。

因此,使用 python-deamon 所需的任何内容,然后使用 argparse 解析剩余的参数。