/
util.py
108 lines (92 loc) · 4.02 KB
/
util.py
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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
# coding=utf-8
import hmac
import re
def number_format(num, places=0):
"""Format a number with grouped thousands and given decimal places"""
places = max(0, places)
tmp = "%.*f" % (places, num)
point = tmp.find(".")
integer = (point == -1) and tmp or tmp[:point]
decimal = (point != -1) and tmp[point:] or ""
count = 0
formatted = []
for i in range(len(integer), 0, -1):
count += 1
formatted.append(integer[i - 1])
if count % 3 == 0 and i - 1:
formatted.append(",")
integer = "".join(formatted[::-1])
return integer + decimal
def get_ip_hmacs(ip_addr, key):
# result looks like 192.168.1
google_style_ip_split = ip_addr.split(".")[0:3]
# we're not handling ipv6
google_style_ip_raw = ".".join(google_style_ip_split)
google_style_ip = hmac.new(key, google_style_ip_raw.encode("utf8"), 'MD5').hexdigest()
ip = hmac.new(key, ip_addr.encode("utf8"), 'MD5').hexdigest()
return ip, google_style_ip
def detect_browser_and_platform(user_agent):
browser = "other"
platform = "desktop"
if user_agent:
if re.search("firefox", user_agent, re.IGNORECASE):
browser = "firefox"
if re.search("android", user_agent, re.IGNORECASE):
platform = "android"
elif re.search("msie", user_agent, re.IGNORECASE) or \
re.search("trident", user_agent, re.IGNORECASE):
browser = "ie"
elif re.search("opr\/", user_agent, re.IGNORECASE):
browser = "opera"
if re.search("mobile", user_agent, re.IGNORECASE):
platform = "android"
elif re.search("chrome", user_agent, re.IGNORECASE):
browser = "chrome"
if re.search("android", user_agent, re.IGNORECASE):
platform = "android"
elif re.search("fxios", user_agent, re.IGNORECASE):
platform = "ios"
browser = "firefox"
elif re.search("crios", user_agent, re.IGNORECASE):
platform = "ios"
browser = "chrome"
elif re.search("safari", user_agent, re.IGNORECASE):
browser = "safari"
if re.search("mobile", user_agent, re.IGNORECASE):
platform = "ios"
return {'browser': browser, 'platform': platform}
# This function is mirrored by prepare_install_button in
# templates/results.html. if you change it here, make sure it's changed
# there as well.
def get_tool_recommendation(detection):
platform = detection['platform']
browser = detection['browser']
tool_url = tool_name = None
if platform == "desktop":
if browser == "firefox":
tool_url = "https://www.eff.org/files/privacy-badger-latest.xpi"
tool_name = "Privacy Badger"
elif browser == "chrome":
tool_url = "https://chrome.google.com/webstore/detail/pkehgijcmpdhfbdbbnkijodmdjhbjlgp"
tool_name = "Privacy Badger"
elif browser == "opera":
tool_url = "https://addons.opera.com/en/extensions/details/ublock/?display=en"
tool_name = u"µBlock"
elif browser == "ie":
tool_url = "http://windows.microsoft.com/en-us/internet-explorer/use-tracking-protection#ie=ie-11"
tool_name = "adding tracker protection lists EasyList and EasyPrivacy"
elif platform == "ios":
if browser == "safari":
tool_url = "https://itunes.apple.com/us/app/disconnect-best-blocking-award/id935480186?mt=8"
tool_name = "Disconnect"
else:
tool_url = "https://itunes.apple.com/us/app/disconnect-privacy-pro-entire/id1057771839?mt=8"
tool_name = "Disconnect Privacy Pro"
elif platform == "android":
if browser == "firefox":
tool_url = "https://addons.mozilla.org/en-US/android/addon/ublock-origin/"
tool_name = "uBlock Origin"
else:
tool_url = "https://disconnect.me/mobile/disconnect-mobile/sideload"
tool_name = "Disconnect"
return {'url': tool_url, 'name': tool_name}