XF/neptune
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Implement editing aliases

Browse Source
This commit is contained in:
Mark Veidemanis
2022-08-01 19:06:56 +01:00
parent 6b0851ab6f
commit 3486f30f37
6 changed files with 81 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
core/lib/threshold.py
View File

@@ -51,13 +51,14 @@ def sort_data(data):
data[item] = OrderedDict({k: v for k, v in sorted_item})
def threshold_request(url, data, method="POST"):
def threshold_request(url, data, method="POST", esc=False):
headers = {
"ApiKey": settings.THRESHOLD_API_KEY,
"Token": settings.THRESHOLD_API_TOKEN,
}
for key in data:
data[key] = escape(data[key])
if esc:
for key in data:
data[key] = escape(data[key])
if method == "POST":
method = requests.post
@@ -96,7 +97,7 @@ def threshold_request(url, data, method="POST"):
def get_chans(net, query):
url = "chans"
payload = {"net": net, "query": query}
channels = threshold_request(url, payload)
channels = threshold_request(url, payload, esc=True)
if not channels:
return []
return channels["chans"]
@@ -105,7 +106,7 @@ def get_chans(net, query):
def get_users(net, query):
url = "users"
payload = {"net": net, "query": query}
users = threshold_request(url, payload)
users = threshold_request(url, payload, esc=True)
if not users:
return []
return users["users"]
@@ -114,7 +115,7 @@ def get_users(net, query):
def annotate_online(net, query):
url = "online"
payload = {"net": net, "query": query}
online_info = threshold_request(url, payload)
online_info = threshold_request(url, payload, esc=True)
if not online_info:
return {}
return online_info
@@ -123,7 +124,7 @@ def annotate_online(net, query):
def annotate_num_users(net, query):
url = "num_users"
payload = {"net": net, "query": query}
user_num_map = threshold_request(url, payload)
user_num_map = threshold_request(url, payload, esc=True)
if not user_num_map:
return {}
return user_num_map
@@ -132,7 +133,7 @@ def annotate_num_users(net, query):
def annotate_num_chans(net, query):
url = "num_chans"
payload = {"net": net, "query": query}
chan_num_map = threshold_request(url, payload)
chan_num_map = threshold_request(url, payload, esc=True)
if not chan_num_map:
return {}
return chan_num_map