robots-auto-action/tools/ip_util.py

"""
IP工具类
"""
import socket
import statistics

import requests
from flask import json


def get_ip_info(ip):
    response = requests.get('http://whois.pconline.com.cn/ipJson.jsp?ip=%s&json=true' % (ip))

    if response.ok:
        print(response.text)
        data = json.loads(response.text)
        pro = data['pro']
        city = data['city']
        addr = data['addr']

        print('%s, %s, %s', pro, city, addr)


def get_city(ip):
    response = requests.get('http://whois.pconline.com.cn/ipJson.jsp?ip=%s&json=true' % (ip))

    if response.ok:
        data = json.loads(response.text)
        city = data['city']

        return city
    else:

        return None


"""
获取本地IP
"""


def get_local_ip():
    ip = ''
    try:
        s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        s.connect(('8.8.8.8', 80))
        ip = s.getsockname()[0]
    finally:
        s.close()
    return ip


# 获取主机ip
def get_current_ip():
    with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s:
        s.connect(("8.8.8.8", 80))
        ip = s.getsockname()[0]
    return ip