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
|
import json
from urllib.parse import quote
import logging
import requests
from project.basecommunicator import BaseCommunicator
class ShoutboxCommunicator(BaseCommunicator):
"""Class for communication with the shoutbox API"""
url = "http://localhost:8000"
interval = 30
message_limit_seconds = 50
largest_id = 0
token = ""
@staticmethod
def get_url():
id_hex = "{0:0{1}x}".format(ShoutboxCommunicator.largest_id, 24)
return "{}/api/last?&telegram=false&id={}".format(ShoutboxCommunicator.url, ShoutboxCommunicator.message_limit_seconds, id_hex)
@staticmethod
def post_url():
return ShoutboxCommunicator.url + "/api/post"
@staticmethod
def fetch():
"""Get and return a list of new messages from the API"""
try:
r = requests.get(ShoutboxCommunicator.get_url())
#logging.info("Response from API OK.")
except:
logging.warning("Failed to get response from API! ({})".format(ShoutboxCommunicator.get_url()))
return
try:
logging.info("url: {}".format(ShoutboxCommunicator.get_url()))
content = json.loads(r.text)
logging.info(content)
except:
logging.warning("Could not parse JSON from request!")
return
if len(content) > 0:
logging.info("Messages from shoutbox:")
for msg in content:
logging.info("{}: {}, id: {}".format(msg["user"], msg["text"], msg["id"]))
id_dec = int(msg["id"], 16)
if id_dec > largest_id:
largest_id = id_dec
return content
@staticmethod
def send(data):
"""Send a message to the API"""
fields = ("user", "text", "ip", "timestamp", "api_token")
post_params = "?"
logging.info("Sending message from shoutbox...")
for field in fields:
if field != fields[-1]:
post_params = "{}{}={}".format(post_params, field, quote(str(data[field]).encode("utf-8")))
post_params += "&"
else:
post_params = "{}{}={}".format(post_params, field, ShoutboxCommunicator.token)
url = ShoutboxCommunicator.post_url() + post_params
try:
requests.post(url, "")
user = data["user"]
logging.info("Sent message from {} to shoutbox, with url:\n{}".format(user, url))
except:
logging.warning("Failed to send message to shoutbox API!")
|
