python - 如何使用 Python 脚本检查特定端口是否正在监听？

Question

我想在对它们运行测试之前检查 api 和应用程序是否正在运行。我知道我可以使用

在 CLI 中获取开放端口列表

sudo lsof -iTCP -sTCP:LISTEN -n -P

但是我想写一个 python 脚本来这样做。关于我应该使用什么库或我应该如何做有什么想法吗？

Answer 1

我找到了一个使用套接字的端口检查器 here并且有效。

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import socket
import re
import sys
def check_server(address, port):
    # Create a TCP socket
    s = socket.socket()
    print "Attempting to connect to %s on port %s" % (address, port)
    try:
        s.connect((address, port))
        print "Connected to %s on port %s" % (address, port)
        return True
    except socket.error, e:
        print "Connection to %s on port %s failed: %s" % (address, port, e)
        return False
    finally:
        s.close()

if __name__ == '__main__':
    from optparse import OptionParser
    parser = OptionParser()

    parser.add_option("-a", "--address", dest="address", default='localhost', help="ADDRESS for server", metavar="ADDRESS")

    parser.add_option("-p", "--port", dest="port", type="int", default=80, help="PORT for server", metavar="PORT")

    (options, args) = parser.parse_args()
    print 'options: %s, args: %s' % (options, args)
    check = check_server(options.address, options.port)
    print 'check_server returned %s' % check

    sys.exit(not check)