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

Question

我想在运行测试之前检查api和app是否正在运行.我知道我可以使用CLI获取开放端口列表

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

但我想写一个python脚本来做到这一点.关于我应该使用什么库或者我该怎么做的任何想法？

Answer 1

我在这里找到了一个使用套接字的端口检查器,它可以工作

#!/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)