1 # getipinfo.py
2 def getipinfo(ip):
3 import requests
4 url=
'http://ip.taobao.com/service/getIpInfo.php?ip='
5 r=requests.get(url+
ip)
6 r1=
r.json()
7 a=
{}
8 a[
'country']=r1[
'data'][
'country']
9 a[
'city']=r1[
'data'][
'city']
10 a[
'region']=r1[
'data'][
'region']
11 return a
1 import getipinfo
2 import unittest
3 import time
4 class test_getipinfo(unittest.TestCase):
5 def test_01(self):
6 t=getipinfo.getipinfo(
'113.13.186.134')
7 self.assertEqual(t[
'country'],
'中国')
8 self.assertEqual(t[
'region'],
'广西')
9 self.assertEqual(t[
'city'],
'柳州')
10 time.sleep(5
)
11 def test_02(self):
12 t = getipinfo.getipinfo(
'39.108.97.124')
13 self.assertEqual(t[
'country'],
'中国')
14 self.assertEqual(t[
'region'],
'广东')
15 self.assertEqual(t[
'city'],
'深圳')
16 time.sleep(5
)
17 def test_03(self):
18 t = getipinfo.getipinfo(
'39.108.97.xxx')
19 self.assertEqual(t[
'country'],
'')
20 self.assertEqual(t[
'region'],
'')
21 self.assertEqual(t[
'city'],
'')
22 if __name__==
'__main__':
23 unittest.main()
转载于:https://www.cnblogs.com/jiawenliao/p/9016219.html
