【Python】Python 2 测试网络连通性脚本

文章目录

    • 前言
    • [1. 命令行传参](#1. 命令行传参)
    • [2. 代码](#2. 代码)

前言

最近在只有python2的服务器上部署服务,不能用三方类库,这里出于好奇心学习下python。这里简单做个脚本,实现了检验网络连通性的功能(类似于curl)。

1. 命令行传参

使用命令行的目的是可以一次性检验多个url的连通性,比如上线前对接多个三方接口可以用上。

bash 复制代码
python test_socket.py -u "{'https://www.baidu.com':'120.232.145.144','ws://localhost:8080/ws':'127.0.0.1','wss://hello/ws':''}" 

参数解释

json 复制代码
{
	'https://www.baidu.com':'120.232.145.144', // 期待域名解析为120.232.145.144 否则报错
	'ws://localhost:8080/ws':'127.0.0.1',  // 期待域名解析为127.0.0.1 否则报错
	'wss://hello/ws':''  // 无期望域名解析的ip,保证连通性即可
}

2. 代码

python 复制代码
# -*- coding: utf-8 -*-

# coding:utf-8
import sys
import getopt
import socket
import urlparse
import re


def get_hostname(input_url):
    if len(input_url) == 0:
        return 'url为空'
    hostname = urlparse.urlparse(input_url).hostname
    if hostname is None:
        return '域名解析为空'
    try:
        return socket.gethostbyname(hostname)
    except socket.error:
        return '域名解析异常'


def get_port(input_url):
    if len(input_url) == 0:
        return 'url为空'
    # url 上读端口号
    input_port = urlparse.urlparse(input_url).port
    if input_port:
        return input_port
    # url 上读不到端口号用协议类型区分
    scheme = urlparse.urlparse(input_url).scheme
    if scheme == 'https' or scheme == 'wss':
        return 443
    if scheme == 'http' or 'ws':
        return 80


def check_connect(input_ip, input_port):
    # 设置超时时间为3秒钟
    socket.setdefaulttimeout(3)
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    try:
        s.connect((input_ip, input_port))
        return '可联通'
    except socket.timeout:
        return '无法联通, 超时'
    except Exception as e:
        return '无法联通, 异常:' + str(e)
    finally:
        s.close()


# 从命令行中读取参数
opts, argv = getopt.getopt(sys.argv[1:], 'u:')
ops = dict(opts)
url_ip_dict = eval(ops.get('-u'))

# 循环参数内容进行连通性检测
print "-"*150
printTemplate = "%-60s %-20s %-20s %-50s"
print printTemplate % ('url', 'ip', 'port', 'result')
print "-"*150
for input_url, input_ip in url_ip_dict.iteritems():
    # 域名解析
    actual_ip = get_hostname(input_url)
    # 未能解析出ip为异常
    if not re.match(r'\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}', input_ip):
        print printTemplate % (input_url, '\\', '\\', actual_ip)
        continue

    # 解析出ip, 且入参存在ip 则校验是否是解析出来的ip
    if input_ip:
        if actual_ip != input_ip:
            print printTemplate % (input_url, input_ip, '\\', '未能联通, 实际解析的ip为' + actual_ip)
            continue

    port = get_port(input_url)
    # 连通性测试
    result = check_connect(actual_ip, port)
    print printTemplate % (input_url, actual_ip, port, result)
相关推荐
用户27784491049933 小时前
借助DeepSeek智能生成测试用例:从提示词到Excel表格的全流程实践
人工智能·python
JavaEdge在掘金6 小时前
ssl.SSLCertVerificationError报错解决方案
python
我不会编程5556 小时前
Python Cookbook-5.1 对字典排序
开发语言·数据结构·python
老歌老听老掉牙6 小时前
平面旋转与交线投影夹角计算
python·线性代数·平面·sympy
满怀10157 小时前
Python入门(7):模块
python
无名之逆7 小时前
Rust 开发提效神器:lombok-macros 宏库
服务器·开发语言·前端·数据库·后端·python·rust
大丈夫立于天地间7 小时前
ISIS协议中的数据库同步
运维·网络·信息与通信
你觉得2057 小时前
哈尔滨工业大学DeepSeek公开课:探索大模型原理、技术与应用从GPT到DeepSeek|附视频与讲义下载方法
大数据·人工智能·python·gpt·学习·机器学习·aigc
Dream Algorithm7 小时前
路由器的 WAN(广域网)口 和 LAN(局域网)口
网络·智能路由器
啊喜拔牙7 小时前
1. hadoop 集群的常用命令
java·大数据·开发语言·python·scala