Toot-Mastodon-CLI-TUI-clien.../toot/commands.py

296 lines
8.9 KiB
Python
Raw Normal View History

2017-04-19 14:47:30 +02:00
# -*- coding: utf-8 -*-
2017-12-30 13:32:52 +01:00
from toot import api, config
from toot.auth import login_interactive, login_browser_interactive, create_app_interactive
from toot.exceptions import ConsoleError, NotFoundError
from toot.output import print_out, print_instance, print_account, print_search_results, print_timeline
from toot.utils import assert_domain_exists, multiline_input, EOF_KEY
2017-04-19 14:47:30 +02:00
def timeline(app, user, args):
2018-06-12 10:40:36 +02:00
# Make sure tag, list and public are not used simultaneously
if len([arg for arg in [args.tag, args.list, args.public] if arg]) > 1:
raise ConsoleError("Only one of --public, --tag, or --list can be used at one time.")
2018-06-12 10:40:36 +02:00
if args.local and not (args.public or args.tag):
raise ConsoleError("The --local option is only valid alongside --public or --tag.")
if args.public:
gen = api.public_timeline_generator(app.instance, local=args.local, limit=args.count)
2018-06-12 10:40:36 +02:00
elif args.tag:
gen = api.tag_timeline_generator(app, user, args.tag, local=args.local, limit=args.count)
elif args.list:
gen = api.timeline_list_genertor(app, user, args.list)
else:
gen = api.home_timeline_generator(app, user, limit=args.count)
2018-06-12 10:40:36 +02:00
while(True):
items = next(gen)
if args.reverse:
items = reversed(items)
print_timeline(items)
if args.once:
break
char = input("\nContinue? [Y/n] ")
if char.lower() == "n":
break
2017-04-19 14:47:30 +02:00
def thread(app, user, args):
toot = api.single_status(app, user, args.status_id)
context = api.context(app, user, args.status_id)
thread = []
for item in context['ancestors']:
thread.append(item)
thread.append(toot)
for item in context['descendants']:
thread.append(item)
print_timeline(thread)
2017-04-19 14:47:30 +02:00
def curses(app, user, args):
2018-01-06 11:56:52 +01:00
from toot.ui.app import TimelineApp
2018-01-06 11:25:05 +01:00
# Make sure tag, list and public are not used simultaneously
if len([arg for arg in [args.tag, args.public] if arg]) > 1:
raise ConsoleError("Only one of --public or --tag can be used at one time.")
if args.local and not (args.public or args.tag):
raise ConsoleError("The --local option is only valid alongside --public or --tag.")
2018-01-06 11:25:05 +01:00
if not args.public and (not app or not user):
raise ConsoleError("You must be logged in to view the home timeline.")
if args.public:
instance = args.instance or app.instance
generator = api.public_timeline_generator(instance, local=args.local)
elif args.tag:
generator = api.tag_timeline_generator(app, user, args.tag, local=args.local)
2018-01-06 11:25:05 +01:00
else:
generator = api.home_timeline_generator(app, user)
TimelineApp(generator).run()
2017-04-19 14:47:30 +02:00
def post(app, user, args):
if args.media:
media = _do_upload(app, user, args.media)
media_ids = [media['id']]
else:
media = None
2017-04-19 14:47:30 +02:00
media_ids = None
if media and not args.text:
args.text = media['text_url']
if not args.text:
print_out("Write or paste your toot. Press <yellow>{}</yellow> to post it.".format(EOF_KEY))
args.text = multiline_input()
if not args.text:
raise ConsoleError("You must specify either text or media to post.")
response = api.post_status(
app, user, args.text,
visibility=args.visibility,
media_ids=media_ids,
sensitive=args.sensitive,
spoiler_text=args.spoiler_text,
2018-06-13 12:43:31 +02:00
in_reply_to_id=args.reply_to,
)
2017-04-19 14:47:30 +02:00
print_out("Toot posted: <green>{}</green>".format(response.get('url')))
2017-04-19 14:47:30 +02:00
2018-06-14 10:40:16 +02:00
def delete(app, user, args):
api.delete_status(app, user, args.status_id)
print_out("<green>✓ Status deleted</green>")
def favourite(app, user, args):
api.favourite(app, user, args.status_id)
print_out("<green>✓ Status favourited</green>")
def unfavourite(app, user, args):
api.unfavourite(app, user, args.status_id)
print_out("<green>✓ Status unfavourited</green>")
def reblog(app, user, args):
api.reblog(app, user, args.status_id)
print_out("<green>✓ Status reblogged</green>")
def unreblog(app, user, args):
api.unreblog(app, user, args.status_id)
print_out("<green>✓ Status unreblogged</green>")
def pin(app, user, args):
api.pin(app, user, args.status_id)
print_out("<green>✓ Status pinned</green>")
def unpin(app, user, args):
api.unpin(app, user, args.status_id)
print_out("<green>✓ Status unpinned</green>")
2017-04-19 14:47:30 +02:00
def auth(app, user, args):
config_data = config.load_config()
if not config_data["users"]:
print_out("You are not logged in to any accounts")
return
active_user = config_data["active_user"]
print_out("Authenticated accounts:")
for uid, u in config_data["users"].items():
active_label = "ACTIVE" if active_user == uid else ""
print_out("* <green>{}</green> <yellow>{}</yellow>".format(uid, active_label))
path = config.get_config_file_path()
print_out("\nAuth tokens are stored in: <blue>{}</blue>".format(path))
2017-04-19 14:47:30 +02:00
2018-06-15 09:39:28 +02:00
def login_cli(app, user, args):
2018-12-25 02:20:30 +01:00
app = create_app_interactive(instance=args.instance, scheme=args.scheme)
login_interactive(app, args.email)
2017-04-19 14:47:30 +02:00
print_out()
print_out("<green>✓ Successfully logged in.</green>")
2017-04-19 14:47:30 +02:00
2018-06-15 09:39:28 +02:00
def login(app, user, args):
2018-12-25 02:20:30 +01:00
app = create_app_interactive(instance=args.instance, scheme=args.scheme)
login_browser_interactive(app)
print_out()
print_out("<green>✓ Successfully logged in.</green>")
2017-04-19 14:47:30 +02:00
def logout(app, user, args):
user = config.load_user(args.account, throw=True)
config.delete_user(user)
print_out("<green>✓ User {} logged out</green>".format(config.user_id(user)))
2017-04-19 14:47:30 +02:00
def activate(app, user, args):
user = config.load_user(args.account, throw=True)
config.activate_user(user)
print_out("<green>✓ User {} active</green>".format(config.user_id(user)))
2017-04-19 14:47:30 +02:00
def upload(app, user, args):
response = _do_upload(app, user, args.file)
2017-12-30 13:14:37 +01:00
msg = "Successfully uploaded media ID <yellow>{}</yellow>, type '<yellow>{}</yellow>'"
print_out()
2017-12-30 13:14:37 +01:00
print_out(msg.format(response['id'], response['type']))
print_out("Original URL: <green>{}</green>".format(response['url']))
print_out("Preview URL: <green>{}</green>".format(response['preview_url']))
print_out("Text URL: <green>{}</green>".format(response['text_url']))
2017-04-19 14:47:30 +02:00
def search(app, user, args):
response = api.search(app, user, args.query, args.resolve)
2017-12-29 14:42:51 +01:00
print_search_results(response)
2017-04-19 14:47:30 +02:00
def _do_upload(app, user, file):
print_out("Uploading media: <green>{}</green>".format(file.name))
2017-04-19 14:47:30 +02:00
return api.upload_media(app, user, file)
def _find_account(app, user, account_name):
"""For a given account name, returns the Account object.
2017-04-19 14:47:30 +02:00
Raises an exception if not found.
"""
if not account_name:
raise ConsoleError("Empty account name given")
accounts = api.search_accounts(app, user, account_name)
if account_name[0] == "@":
account_name = account_name[1:]
for account in accounts:
if account['acct'] == account_name:
2017-04-19 14:47:30 +02:00
return account
2017-04-19 15:29:40 +02:00
raise ConsoleError("Account not found")
2017-04-19 14:47:30 +02:00
def follow(app, user, args):
account = _find_account(app, user, args.account)
api.follow(app, user, account['id'])
print_out("<green>✓ You are now following {}</green>".format(args.account))
2017-04-19 14:47:30 +02:00
def unfollow(app, user, args):
account = _find_account(app, user, args.account)
2017-04-26 11:49:21 +02:00
api.unfollow(app, user, account['id'])
print_out("<green>✓ You are no longer following {}</green>".format(args.account))
2017-04-19 14:47:30 +02:00
2017-04-26 11:49:21 +02:00
def mute(app, user, args):
account = _find_account(app, user, args.account)
api.mute(app, user, account['id'])
print_out("<green>✓ You have muted {}</green>".format(args.account))
2017-04-19 14:47:30 +02:00
2017-04-26 11:49:21 +02:00
def unmute(app, user, args):
account = _find_account(app, user, args.account)
api.unmute(app, user, account['id'])
print_out("<green>✓ {} is no longer muted</green>".format(args.account))
2017-04-26 11:49:21 +02:00
def block(app, user, args):
account = _find_account(app, user, args.account)
api.block(app, user, account['id'])
print_out("<green>✓ You are now blocking {}</green>".format(args.account))
2017-04-26 11:49:21 +02:00
def unblock(app, user, args):
account = _find_account(app, user, args.account)
api.unblock(app, user, account['id'])
print_out("<green>✓ {} is no longer blocked</green>".format(args.account))
2017-04-19 14:47:30 +02:00
def whoami(app, user, args):
2017-04-19 15:29:40 +02:00
account = api.verify_credentials(app, user)
2017-12-29 14:42:51 +01:00
print_account(account)
2017-04-19 14:47:30 +02:00
2017-04-19 15:29:40 +02:00
def whois(app, user, args):
account = _find_account(app, user, args.account)
2017-12-29 14:42:51 +01:00
print_account(account)
2017-12-29 14:26:40 +01:00
def instance(app, user, args):
name = args.instance or (app and app.instance)
if not name:
raise ConsoleError("Please specify instance name.")
assert_domain_exists(name)
try:
instance = api.get_instance(name, args.scheme)
print_instance(instance)
except NotFoundError:
raise ConsoleError(
"Instance not found at {}.\n"
"The given domain probably does not host a Mastodon instance.".format(name)
)