Docstring cleanup

This commit is contained in:
Lorenz Diener 2016-11-24 00:36:00 +01:00
pare e89c7bda86
commit 98e1bf7f40

Veure arxiu

@ -5,16 +5,17 @@ import os
import os.path import os.path
class Mastodon: class Mastodon:
""" Super basic but thorough and easy to use mastodon.social """
api wrapper in python. Super basic but thorough and easy to use mastodon.social
api wrapper in python.
If anything is unclear, check the official API docs at If anything is unclear, check the official API docs at
https://github.com/Gargron/mastodon/wiki/API https://github.com/Gargron/mastodon/wiki/API
Presently, only username-password login is supported, somebody please Presently, only username-password login is supported, somebody please
patch in Real Proper OAuth if desired. patch in Real Proper OAuth if desired.
KNOWN BUGS: Media api does not work, reason unclear. KNOWN BUGS: Media api does not work, reason unclear.
""" """
__DEFAULT_BASE_URL = 'https://mastodon.social' __DEFAULT_BASE_URL = 'https://mastodon.social'
@ -23,13 +24,14 @@ class Mastodon:
### ###
@staticmethod @staticmethod
def create_app(client_name, scopes = ['read', 'write', 'follow'], redirect_uris = None, to_file = None, api_base_url = __DEFAULT_BASE_URL): def create_app(client_name, scopes = ['read', 'write', 'follow'], redirect_uris = None, to_file = None, api_base_url = __DEFAULT_BASE_URL):
"""Creates a new app with given client_name and scopes (read, write, follow) """
Creates a new app with given client_name and scopes (read, write, follow)
Specify redirect_uris if you want users to be redirected to a certain page after authenticating. Specify redirect_uris if you want users to be redirected to a certain page after authenticating.
Specify to_file to persist your apps info to a file so you can use them in the constructor. Specify to_file to persist your apps info to a file so you can use them in the constructor.
Specify api_base_url if you want to register an app on an instance different from the flagship one. Specify api_base_url if you want to register an app on an instance different from the flagship one.
returns client_id and client_secret Returns client_id and client_secret.
""" """
request_data = { request_data = {
'client_name': client_name, 'client_name': client_name,
@ -54,13 +56,14 @@ class Mastodon:
# Authentication, including constructor # Authentication, including constructor
### ###
def __init__(self, client_id, client_secret = None, access_token = None, api_base_url = __DEFAULT_BASE_URL): def __init__(self, client_id, client_secret = None, access_token = None, api_base_url = __DEFAULT_BASE_URL):
"""Create a new API wrapper instance based on the given client_secret and client_id. If you """
give a client_id and it is not a file, you must also give a secret. Creates a new API wrapper instance based on the given client_secret and client_id. If you
give a client_id and it is not a file, you must also give a secret.
You can also directly specify an access_token, directly or as a file. You can also directly specify an access_token, directly or as a file.
Specify api_base_url if you wish to talk to an instance other than the flagship one. Specify api_base_url if you wish to talk to an instance other than the flagship one.
If a file is given as client_id, read client ID and secret from that file If a file is given as client_id, read client ID and secret from that file
""" """
self.api_base_url = api_base_url self.api_base_url = api_base_url
self.client_id = client_id self.client_id = client_id
@ -80,10 +83,11 @@ class Mastodon:
self.access_token = token_file.readline().rstrip() self.access_token = token_file.readline().rstrip()
def log_in(self, username, password, scopes = ['read', 'write', 'follow'], to_file = None): def log_in(self, username, password, scopes = ['read', 'write', 'follow'], to_file = None):
"""Logs in and sets access_token to what was returned. """
Can persist access token to file. Logs in and sets access_token to what was returned.
Can persist access token to file.
Returns the access_token, as well. Returns the access_token, as well.
""" """
params = self.__generate_params(locals()) params = self.__generate_params(locals())
params['client_id'] = self.client_id params['client_id'] = self.client_id
@ -104,8 +108,10 @@ class Mastodon:
# Reading data: Timelines # Reading data: Timelines
## ##
def timeline(self, timeline = 'home', max_id = None, since_id = None, limit = None): def timeline(self, timeline = 'home', max_id = None, since_id = None, limit = None):
"""Returns statuses, most recent ones first. Timeline can be home, mentions, public """
or tag/:hashtag""" Returns statuses, most recent ones first. Timeline can be home, mentions, public
or tag/:hashtag
"""
params = self.__generate_params(locals(), ['timeline']) params = self.__generate_params(locals(), ['timeline'])
return self.__api_request('GET', '/api/v1/timelines/' + timeline, params) return self.__api_request('GET', '/api/v1/timelines/' + timeline, params)
@ -113,58 +119,82 @@ class Mastodon:
# Reading data: Statuses # Reading data: Statuses
### ###
def status(self, id): def status(self, id):
"""Returns status.""" """
Returns a status.
"""
return self.__api_request('GET', '/api/v1/statuses/' + str(id)) return self.__api_request('GET', '/api/v1/statuses/' + str(id))
def status_context(self, id): def status_context(self, id):
"""Returns ancestors and descendants of the status.""" """
Returns ancestors and descendants of the status.
"""
return self.__api_request('GET', '/api/v1/statuses/' + str(id) + '/context') return self.__api_request('GET', '/api/v1/statuses/' + str(id) + '/context')
def status_reblogged_by(self, id): def status_reblogged_by(self, id):
"""Returns a list of users that have reblogged a status.""" """
Returns a list of users that have reblogged a status.
"""
return self.__api_request('GET', '/api/v1/statuses/' + str(id) + '/reblogged_by') return self.__api_request('GET', '/api/v1/statuses/' + str(id) + '/reblogged_by')
def status_favourited_by(self, id): def status_favourited_by(self, id):
"""Returns a list of users that have favourited a status.""" """
Returns a list of users that have favourited a status.
"""
return self.__api_request('GET', '/api/v1/statuses/' + str(id) + '/favourited_by') return self.__api_request('GET', '/api/v1/statuses/' + str(id) + '/favourited_by')
### ###
# Reading data: Accounts # Reading data: Accounts
### ###
def account(self, id): def account(self, id):
"""Returns account.""" """
Returns account.
"""
return self.__api_request('GET', '/api/v1/accounts/' + str(id)) return self.__api_request('GET', '/api/v1/accounts/' + str(id))
def account_verify_credentials(self): def account_verify_credentials(self):
"""Returns authenticated user's account.""" """
Returns authenticated user's account.
"""
return self.__api_request('GET', '/api/v1/accounts/verify_credentials') return self.__api_request('GET', '/api/v1/accounts/verify_credentials')
def account_statuses(self, id, max_id = None, since_id = None, limit = None): def account_statuses(self, id, max_id = None, since_id = None, limit = None):
"""Returns statuses by user. Same options as timeline are permitted.""" """
Returns statuses by user. Same options as timeline are permitted.
"""
params = self.__generate_params(locals(), ['id']) params = self.__generate_params(locals(), ['id'])
return self.__api_request('GET', '/api/v1/accounts/' + str(id) + '/statuses') return self.__api_request('GET', '/api/v1/accounts/' + str(id) + '/statuses')
def account_following(self, id): def account_following(self, id):
"""Returns users the given user is following.""" """
Returns users the given user is following.
"""
return self.__api_request('GET', '/api/v1/accounts/' + str(id) + '/following') return self.__api_request('GET', '/api/v1/accounts/' + str(id) + '/following')
def account_followers(self, id): def account_followers(self, id):
"""Returns users the given user is followed by.""" """
Returns users the given user is followed by.
"""
return self.__api_request('GET', '/api/v1/accounts/' + str(id) + '/followers') return self.__api_request('GET', '/api/v1/accounts/' + str(id) + '/followers')
def account_relationships(self, id): def account_relationships(self, id):
"""Returns relationships (following, followed_by, blocking) of the logged in user to """
a given account. id can be a list.""" Returns relationships (following, followed_by, blocking) of the logged in user to
a given account. id can be a list.
"""
params = self.__generate_params(locals()) params = self.__generate_params(locals())
return self.__api_request('GET', '/api/v1/accounts/relationships', params) return self.__api_request('GET', '/api/v1/accounts/relationships', params)
def account_suggestions(self): def account_suggestions(self):
"""Returns accounts that the system suggests the authenticated user to follow.""" """
Returns accounts that the system suggests the authenticated user to follow.
"""
return self.__api_request('GET', '/api/v1/accounts/suggestions') return self.__api_request('GET', '/api/v1/accounts/suggestions')
def account_search(self, q, limit = None): def account_search(self, q, limit = None):
"""Returns matching accounts. Will lookup an account remotely if the search term is """
in the username@domain format and not yet in the database.""" Returns matching accounts. Will lookup an account remotely if the search term is
in the username@domain format and not yet in the database.
"""
params = self.__generate_params(locals()) params = self.__generate_params(locals())
return self.__api_request('GET', '/api/v1/accounts/search', params) return self.__api_request('GET', '/api/v1/accounts/search', params)
@ -172,19 +202,25 @@ class Mastodon:
# Writing data: Statuses # Writing data: Statuses
### ###
def status_post(self, status, in_reply_to_id = None, media_ids = None): def status_post(self, status, in_reply_to_id = None, media_ids = None):
"""Posts a status. Can optionally be in reply to another status and contain """
up to four pieces of media (Uploaded via media_post()). Posts a status. Can optionally be in reply to another status and contain
up to four pieces of media (Uploaded via media_post()).
Returns the new status.""" Returns the new status.
"""
params = self.__generate_params(locals()) params = self.__generate_params(locals())
return self.__api_request('POST', '/api/v1/statuses', params) return self.__api_request('POST', '/api/v1/statuses', params)
def toot(self, status): def toot(self, status):
"""Synonym for status_post that only takes the status text as input.""" """
Synonym for status_post that only takes the status text as input.
"""
return self.status_post(status) return self.status_post(status)
def status_delete(self, id): def status_delete(self, id):
"""Deletes a status""" """
Deletes a status
"""
return self.__api_request('DELETE', '/api/v1/statuses/' + str(id)) return self.__api_request('DELETE', '/api/v1/statuses/' + str(id))
def status_reblog(self, id): def status_reblog(self, id):
@ -194,58 +230,73 @@ class Mastodon:
return self.__api_request('POST', '/api/v1/statuses/' + str(id) + "/reblog") return self.__api_request('POST', '/api/v1/statuses/' + str(id) + "/reblog")
def status_unreblog(self, id): def status_unreblog(self, id):
"""Un-reblogs a status. """
Un-reblogs a status.
Returns the status that used to be reblogged.""" Returns the status that used to be reblogged.
"""
return self.__api_request('POST', '/api/v1/statuses/' + str(id) + "/unreblog") return self.__api_request('POST', '/api/v1/statuses/' + str(id) + "/unreblog")
def status_favourite(self, id): def status_favourite(self, id):
"""Favourites a status. """
Favourites a status.
Returns the favourited status.""" Returns the favourited status.
"""
return self.__api_request('POST', '/api/v1/statuses/' + str(id) + "/favourite") return self.__api_request('POST', '/api/v1/statuses/' + str(id) + "/favourite")
def status_unfavourite(self, id): def status_unfavourite(self, id):
"""Favourites a status. """Favourites a status.
Returns the un-favourited status.""" Returns the un-favourited status.
"""
return self.__api_request('POST', '/api/v1/statuses/' + str(id) + "/unfavourite") return self.__api_request('POST', '/api/v1/statuses/' + str(id) + "/unfavourite")
### ###
# Writing data: Statuses # Writing data: Statuses
### ###
def account_follow(self, id): def account_follow(self, id):
"""Follows a user. """
Follows a user.
Returns the updated relationship to the user.""" Returns the updated relationship to the user.
"""
return self.__api_request('POST', '/api/v1/accounts/' + str(id) + "/follow") return self.__api_request('POST', '/api/v1/accounts/' + str(id) + "/follow")
def account_unfollow(self, id): def account_unfollow(self, id):
"""Unfollows a user. """
Unfollows a user.
Returns the updated relationship to the user.""" Returns the updated relationship to the user.
"""
return self.__api_request('POST', '/api/v1/accounts/' + str(id) + "/unfollow") return self.__api_request('POST', '/api/v1/accounts/' + str(id) + "/unfollow")
def account_block(self, id): def account_block(self, id):
"""Blocks a user. """
Blocks a user.
Returns the updated relationship to the user.""" Returns the updated relationship to the user.
"""
return self.__api_request('POST', '/api/v1/accounts/' + str(id) + "/block") return self.__api_request('POST', '/api/v1/accounts/' + str(id) + "/block")
def account_unblock(self, id): def account_unblock(self, id):
"""Unblocks a user. """
Unblocks a user.
Returns the updated relationship to the user.""" Returns the updated relationship to the user.
"""
return self.__api_request('POST', '/api/v1/accounts/' + str(id) + "/unblock") return self.__api_request('POST', '/api/v1/accounts/' + str(id) + "/unblock")
### ###
# Writing data: Media # Writing data: Media
### ###
def media_post(self, media_file): def media_post(self, media_file):
"""Posts an image. media_file can either be image data or """
a file name. Posts an image. media_file can either be image data or
a file name.
Returns the ID of the media that can then be used in status_post().""" Returns the ID of the media that can then be used in status_post().
"""
if os.path.isfile(media_file): if os.path.isfile(media_file):
media_file = open(media_file, 'rb') media_file = open(media_file, 'rb')
@ -255,7 +306,9 @@ class Mastodon:
# Internal helpers, dragons probably # Internal helpers, dragons probably
### ###
def __api_request(self, method, endpoint, params = {}, files = {}): def __api_request(self, method, endpoint, params = {}, files = {}):
""" Internal API request helper.""" """
Internal API request helper.
"""
response = None response = None
headers = None headers = None
@ -280,7 +333,9 @@ class Mastodon:
return response.json() return response.json()
def __generate_params(self, params, exclude = []): def __generate_params(self, params, exclude = []):
"""Internal named-parameters-to-dict helper""" """
Internal named-parameters-to-dict helper.
"""
params = dict(params) params = dict(params)
del params['self'] del params['self']