ymyzk/kawasemi

# -*- coding: utf-8 -*-
from __future__ import unicode_literals

import requests
from requests_oauthlib import OAuth1

from .base import BaseChannel
from ..exceptions import HttpError


class TwitterChannel(BaseChannel):
    url = "https://api.twitter.com/1.1/statuses/update.json"

    def __init__(self, api_key, api_secret, access_token, access_token_secret,
                 *args, **kwargs):
        """A channel for posting a tweet

        :type api_key: unicode | str
        :type api_secret: unicode | str
        :type access_token: unicode | str
        :type access_token_secret: unicode | str
        """
        self.api_key = api_key
        self.api_secret = api_secret
        self.access_token = access_token
        self.access_token_secret = access_token_secret

    def send(self, message, fail_silently=False, options=None):
        payload = {
            "status": message
        }

        self._set_payload_from_options(payload, options, "twitter", [
            "in_reply_to_status_id", "possibly_sensitive", "lat", "long",
            "place_id", "display_coordinates", "trim_user", "media_ids"])

        auth = OAuth1(self.api_key, self.api_secret, self.access_token,
                      self.access_token_secret)

        try:
            response = requests.post(self.url, auth=auth, data=payload)
            if response.status_code != requests.codes.ok:
                raise HttpError(response.status_code, response.text)
        except Exception:
            if not fail_silently:
                raise