python检测远程tcp端口

今天帮运维的同学写了个用来检测远程服务器tcp端口的脚本，使用的是python的socket模块来做的，具体如下：

#!/usr/bin/env python
#coding:utf-8
#filename:tcp.py

'''
author:	gavingeng
date:	2011-12-14 09:35:59 
'''
import socket
import sys

NORMAL=0
ERROR=1
TIMEOUT=5

def ping(ip,port,timeout=TIMEOUT):
	try:
		cs=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
		address=(str(ip),int(port))
		status = cs.connect_ex((address))
		cs.settimeout(timeout)
		#this status is returnback from tcpserver
		if status != NORMAL :
			print ERROR
		else:
			print NORMAL	
	except Exception ,e:
		print ERROR
		print "error:%s" %e
		return ERROR
	
	return NORMAL

if __name__=='__main__':
	if len(sys.argv) < 3 :
		print ur'请按照如下格式使用: ./tcp.py 127.0.0.1 3306 5来使用'
		sys.exit(1)
	
	ip = sys.argv[1]
	port = sys.argv[2]
	try:
		timeout = sys.argv[3]
	except IndexError ,e:
		timeout=TIMEOUT
	ping(ip,port,timeout)