mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
1.4 KiB
52 lines
1.4 KiB
# ~*~ coding: utf-8 ~*~
|
|
#
|
|
|
|
from __future__ import unicode_literals
|
|
import requests
|
|
import ipaddress
|
|
|
|
from .models import LoginLog
|
|
|
|
|
|
def validate_ip(ip):
|
|
try:
|
|
ipaddress.ip_address(ip.decode('utf-8'))
|
|
return True
|
|
except ValueError:
|
|
print('valid error')
|
|
return False
|
|
|
|
|
|
def write_login_log(username, name='', login_type='',
|
|
login_ip='', user_agent=''):
|
|
if not (login_ip and validate_ip(login_ip)):
|
|
login_ip = '0.0.0.0'
|
|
if not name:
|
|
name = username
|
|
login_city = get_ip_city(login_ip)
|
|
LoginLog.objects.create(username=username, name=name, login_type=login_type,
|
|
login_ip=login_ip, login_city=login_city, user_agent=user_agent)
|
|
|
|
|
|
def get_ip_city(ip, timeout=10):
|
|
# Taobao ip api: http://ip.taobao.com//service/getIpInfo.php?ip=8.8.8.8
|
|
# Sina ip api: http://int.dpool.sina.com.cn/iplookup/iplookup.php?ip=8.8.8.8&format=json
|
|
|
|
url = 'http://int.dpool.sina.com.cn/iplookup/iplookup.php?ip=%s&format=json' % ip
|
|
try:
|
|
r = requests.get(url, timeout=timeout)
|
|
print(r)
|
|
except requests.Timeout:
|
|
r = None
|
|
city = 'Unknown'
|
|
if r and r.status_code == 200:
|
|
try:
|
|
data = r.json()
|
|
if not isinstance(data, int) and data['ret'] == 1:
|
|
city = data['country'] + ' ' + data['city']
|
|
except ValueError:
|
|
pass
|
|
return city
|
|
|
|
|