witchie/toot/api.py

162 lines
3.9 KiB
Python
Raw Normal View History

2017-04-16 12:14:33 +00:00
# -*- coding: utf-8 -*-
import re
2017-04-16 12:14:33 +00:00
2017-12-29 13:26:40 +00:00
from urllib.parse import urlparse, urlencode
2017-04-16 12:14:33 +00:00
2017-12-30 13:15:51 +00:00
from toot import http, CLIENT_NAME, CLIENT_WEBSITE
from toot.exceptions import AuthenticationError
2017-04-16 12:14:33 +00:00
SCOPES = 'read write follow'
2017-04-26 09:49:21 +00:00
def _account_action(app, user, account, action):
url = '/api/v1/accounts/{}/{}'.format(account, action)
2017-04-26 09:49:21 +00:00
2017-12-30 13:15:51 +00:00
return http.post(app, user, url).json()
2017-04-26 09:49:21 +00:00
def create_app(domain):
url = 'http://{}/api/v1/apps'.format(domain)
2017-04-16 12:14:33 +00:00
data = {
2017-04-16 12:14:33 +00:00
'client_name': CLIENT_NAME,
'redirect_uris': 'urn:ietf:wg:oauth:2.0:oob',
'scopes': SCOPES,
'website': CLIENT_WEBSITE,
}
2017-04-16 12:14:33 +00:00
return http.anon_post(url, data).json()
2017-04-16 12:14:33 +00:00
def login(app, username, password):
url = app.base_url + '/oauth/token'
data = {
2017-04-16 12:14:33 +00:00
'grant_type': 'password',
'client_id': app.client_id,
'client_secret': app.client_secret,
'username': username,
'password': password,
'scope': SCOPES,
}
response = http.anon_post(url, data, allow_redirects=False)
2017-04-16 12:14:33 +00:00
# If auth fails, it redirects to the login page
if response.is_redirect:
raise AuthenticationError()
2017-04-16 12:14:33 +00:00
2017-12-30 13:15:51 +00:00
return http.process_response(response).json()
2017-04-16 12:14:33 +00:00
def get_browser_login_url(app):
"""Returns the URL for manual log in via browser"""
return "{}/oauth/authorize/?{}".format(app.base_url, urlencode({
"response_type": "code",
"redirect_uri": "urn:ietf:wg:oauth:2.0:oob",
"scope": "read write follow",
"client_id": app.client_id,
}))
def request_access_token(app, authorization_code):
url = app.base_url + '/oauth/token'
data = {
'grant_type': 'authorization_code',
'client_id': app.client_id,
'client_secret': app.client_secret,
'code': authorization_code,
'redirect_uri': 'urn:ietf:wg:oauth:2.0:oob',
}
response = http.anon_post(url, data, allow_redirects=False)
2017-12-30 13:15:51 +00:00
return http.process_response(response).json()
2017-04-16 12:14:33 +00:00
def post_status(app, user, status, visibility='public', media_ids=None):
2017-12-30 13:15:51 +00:00
return http.post(app, user, '/api/v1/statuses', {
2017-04-16 12:14:33 +00:00
'status': status,
'media_ids[]': media_ids,
'visibility': visibility,
}).json()
2017-04-16 12:14:33 +00:00
def timeline_home(app, user):
2017-12-30 13:15:51 +00:00
return http.get(app, user, '/api/v1/timelines/home').json()
def _get_next_path(headers):
links = headers.get('Link', '')
matches = re.match('<([^>]+)>; rel="next"', links)
if matches:
url = matches.group(1)
return urlparse(url).path
def timeline_generator(app, user):
next_path = '/api/v1/timelines/home'
while next_path:
2017-12-30 13:15:51 +00:00
response = http.get(app, user, next_path)
yield response.json()
next_path = _get_next_path(response.headers)
2017-04-16 12:14:33 +00:00
def upload_media(app, user, file):
2017-12-30 13:15:51 +00:00
return http.post(app, user, '/api/v1/media', files={
2017-04-16 12:14:33 +00:00
'file': file
}).json()
2017-04-16 13:07:27 +00:00
def search(app, user, query, resolve):
2017-12-30 13:15:51 +00:00
return http.get(app, user, '/api/v1/search', {
2017-04-16 13:07:27 +00:00
'q': query,
'resolve': resolve,
}).json()
2017-04-16 15:15:05 +00:00
2017-04-17 09:10:57 +00:00
def search_accounts(app, user, query):
2017-12-30 13:15:51 +00:00
return http.get(app, user, '/api/v1/accounts/search', {
2017-04-17 09:10:57 +00:00
'q': query,
}).json()
2017-04-17 09:10:57 +00:00
2017-04-16 15:15:05 +00:00
def follow(app, user, account):
2017-04-26 09:49:21 +00:00
return _account_action(app, user, account, 'follow')
2017-04-16 15:15:05 +00:00
def unfollow(app, user, account):
2017-04-26 09:49:21 +00:00
return _account_action(app, user, account, 'unfollow')
2017-04-16 15:15:05 +00:00
2017-04-26 09:49:21 +00:00
def mute(app, user, account):
return _account_action(app, user, account, 'mute')
def unmute(app, user, account):
return _account_action(app, user, account, 'unmute')
def block(app, user, account):
return _account_action(app, user, account, 'block')
def unblock(app, user, account):
return _account_action(app, user, account, 'unblock')
2017-04-16 15:52:54 +00:00
def verify_credentials(app, user):
2017-12-30 13:15:51 +00:00
return http.get(app, user, '/api/v1/accounts/verify_credentials').json()
def get_notifications(app, user):
2017-12-30 13:15:51 +00:00
return http.get(app, user, '/api/v1/notifications').json()
2017-12-29 13:26:40 +00:00
def get_instance(domain):
2017-12-29 13:26:40 +00:00
url = "http://{}/api/v1/instance".format(domain)
return http.anon_get(url).json()