85 lines
3.0 KiB
Python
85 lines
3.0 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# https://github.com/Lynnesbian/fediverse-bot-template
|
||
|
|
# Copyright (C) 2019 Lynne (@lynnesbian@fedi.lynnesbian.space)
|
||
|
|
|
||
|
|
# This program is free software: you can redistribute it and/or modify
|
||
|
|
# it under the terms of the GNU Affero General Public License as published
|
||
|
|
# by the Free Software Foundation, either version 3 of the License, or
|
||
|
|
# (at your option) any later version.
|
||
|
|
|
||
|
|
# This program is distributed in the hope that it will be useful,
|
||
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
|
# GNU Affero General Public License for more details.
|
||
|
|
|
||
|
|
# import Mastodon.py and other modules
|
||
|
|
from mastodon import Mastodon
|
||
|
|
import os, json, re, shutil, sys
|
||
|
|
|
||
|
|
# specify defaults
|
||
|
|
|
||
|
|
cfg = {
|
||
|
|
"site": "https://botsin.space",
|
||
|
|
}
|
||
|
|
|
||
|
|
# scopes define what your bot has access to.
|
||
|
|
# a list of scopes is available here: https://docs.joinmastodon.org/api/permissions/
|
||
|
|
# to see what scopes you need for a particular action, check the relevant section in the Mastodon REST API docs.
|
||
|
|
# for example, if you want to see what you need to manage statuses, see https://docs.joinmastodon.org/api/rest/statuses/
|
||
|
|
# the permissions below allow you to read posts, see who's following you (and who you're following), post statuses, and view notifications (needed for replying to people).
|
||
|
|
scopes = ["read:statuses", "read:accounts", "write:statuses", "read:notifications"]
|
||
|
|
|
||
|
|
# try to read from the existing config.json. if it's not there, use the defaults to create a new one.
|
||
|
|
try:
|
||
|
|
cfg.update(json.load(open('config.json', 'r')))
|
||
|
|
except:
|
||
|
|
json.dump(cfg, open("config.json", "w+"))
|
||
|
|
|
||
|
|
# load meta information, which defines the bot name, source code URL, etc.
|
||
|
|
# note that it is mandatory to provide a source code link, as stated by the AGPLv3 license.
|
||
|
|
try:
|
||
|
|
meta = json.load(open("meta.json", 'r'))
|
||
|
|
except OSError as e:
|
||
|
|
print("Failed to load meta.json: {}".format(e.strerror))
|
||
|
|
sys.exit(1)
|
||
|
|
except:
|
||
|
|
raise
|
||
|
|
|
||
|
|
if "client" not in cfg:
|
||
|
|
print("No application info -- registering {} with {}".format(meta['bot_name'], cfg['site']))
|
||
|
|
client_id, client_secret = Mastodon.create_app(meta['bot_name'],
|
||
|
|
api_base_url=cfg['site'],
|
||
|
|
scopes=scopes,
|
||
|
|
website=meta['source_url'])
|
||
|
|
|
||
|
|
# save application information
|
||
|
|
cfg['client'] = {
|
||
|
|
"id": client_id,
|
||
|
|
"secret": client_secret
|
||
|
|
}
|
||
|
|
|
||
|
|
if "secret" not in cfg:
|
||
|
|
print("No user credentials -- logging in to {}".format(cfg['site']))
|
||
|
|
client = Mastodon(client_id = cfg['client']['id'],
|
||
|
|
client_secret = cfg['client']['secret'],
|
||
|
|
api_base_url=cfg['site'])
|
||
|
|
|
||
|
|
print("Open this URL and authenticate to give {} access to your bot's account: {}".format(meta['bot_name'], client.auth_request_url(scopes=scopes)))
|
||
|
|
# log in and save the provided information
|
||
|
|
cfg['secret'] = client.log_in(code=input("Secret: "), scopes=scopes)
|
||
|
|
|
||
|
|
# save the current configuration
|
||
|
|
|
||
|
|
json.dump(cfg, open("config.json", "w+"))
|
||
|
|
|
||
|
|
# test login information
|
||
|
|
client = Mastodon(
|
||
|
|
client_id=cfg['client']['id'],
|
||
|
|
client_secret = cfg['client']['secret'],
|
||
|
|
access_token=cfg['secret'],
|
||
|
|
api_base_url=cfg['site'])
|
||
|
|
|
||
|
|
me = client.account_verify_credentials()
|
||
|
|
|
||
|
|
print("Done!")
|