import json import logging import requests from project.basecommunicator import BaseCommunicator class ShoutboxCommunicator(BaseCommunicator): """Class for communication with the shoutbox API""" url = "http://localhost:8000" interval = 10 token = "" @staticmethod def get_url(): return "{}/api/last?seconds={}".format(ShoutboxCommunicator.url, ShoutboxCommunicator.interval) @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: content = json.loads(r.text) 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("{}: {}".format(msg["user"], msg["text"])) return content @staticmethod def send(data): """Send a message to the API""" post_params = "?user={}&text={}&id={}&ip={}×tamp={}&api_token={}".format( data["user"], data["text"], data["id"], data["ip"], data["timestamp"], ShoutboxCommunicator.token ) try: requests.post(ShoutboxCommunicator.post_url() + post_params, "") user = data["user"] logging.info("Sent message from {} to shoutbox.".format(user)) except: logging.warning("Failed to send message to shoutbox API!")