1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
import requests
import datetime
from ua_parser import user_agent_parser
from database import SessionLocal
import config
from models import Link, Log
"""
Create a new log whenever a link is visited
"""
def ip_to_location(ip):
if not config.IP_TO_LOCATION:
return "-, -", "-"
url = f"https://api.ip2location.io/?key={config.API_KEY}&ip={ip}"
response = requests.get(url)
data = response.json()
if response.status_code != 200:
config.LOG.error(
"Error with IP2Location API. Perhaps the API is down."
)
return "-, -", "-"
if "error" in data:
config.LOG.error(
"Error with IP2Location API. Likely wrong API key or insufficient"
" funds."
)
return "-, -", "-"
location = ""
# Sometimes a certain name may not be present, so always check
if "city_name" in data:
location += data["city_name"]
if "region_name" in data:
location += f', {data["region_name"]}'
if "country_name" in data:
location += f', {data["country_name"]}'
isp = data["as"]
return location, isp
def log(link, ip, user_agent):
db = SessionLocal()
# Get the redirect link and owner of the link
redirect_link, owner = (
db.query(Link.redirect_link, Link.owner)
.filter(Link.link == link)
.first()
)
# Get the location and ISP of the user
location, isp = ip_to_location(ip)
ua_string = user_agent_parser.Parse(user_agent)
browser = ua_string["user_agent"]["family"]
os = f'{ua_string["os"]["family"]} {ua_string["os"]["major"]}'
# Create the log and commit it to the database
new_log = Log(
owner=owner,
link=link,
timestamp=datetime.datetime.utcnow(),
ip=ip,
location=location,
browser=browser,
os=os,
user_agent=user_agent,
isp=isp,
)
db.add(new_log)
db.commit()
db.close()
# Return the redirect link in order to properly redirect the user
return redirect_link
|
