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

276 lines
7.4 KiB
Python
Raw Normal View History

from __future__ import print_function
2017-04-12 16:42:04 +02:00
import os
import sys
2017-04-13 13:52:28 +02:00
import logging
2017-04-12 16:42:04 +02:00
2017-04-13 13:14:01 +02:00
from bs4 import BeautifulSoup
2017-04-12 16:42:04 +02:00
from builtins import input
2017-04-13 13:14:01 +02:00
from datetime import datetime
from future.moves.itertools import zip_longest
2017-04-12 16:42:04 +02:00
from getpass import getpass
2017-04-13 13:14:01 +02:00
from itertools import chain
from optparse import OptionParser
2017-04-13 13:14:01 +02:00
from textwrap import TextWrapper
2017-04-12 16:42:04 +02:00
from .config import save_user, load_user, load_app, save_app, CONFIG_APP_FILE, CONFIG_USER_FILE
from . import create_app, login, post_status, timeline_home, upload_media, DEFAULT_INSTANCE
2017-04-12 16:42:04 +02:00
2017-04-13 13:52:28 +02:00
class ConsoleError(Exception):
pass
def red(text):
return "\033[31m{}\033[0m".format(text)
2017-04-12 16:42:04 +02:00
def green(text):
return "\033[32m{}\033[0m".format(text)
2017-04-12 16:42:04 +02:00
def yellow(text):
return "\033[33m{}\033[0m".format(text)
def print_error(text):
print(red(text), file=sys.stderr)
2017-04-12 16:42:04 +02:00
def create_app_interactive():
2017-04-13 13:52:28 +02:00
instance = input("Choose an instance [%s]: " % green(DEFAULT_INSTANCE))
2017-04-12 16:42:04 +02:00
if not instance:
instance = DEFAULT_INSTANCE
base_url = 'https://{}'.format(instance)
2017-04-13 13:52:28 +02:00
print("Registering application with %s" % green(base_url))
try:
app = create_app(base_url)
except:
raise ConsoleError("Failed authenticating application. Did you enter a valid instance?")
2017-04-12 16:42:04 +02:00
save_app(app)
2017-04-13 13:52:28 +02:00
print("Application tokens saved to: {}".format(green(CONFIG_APP_FILE)))
2017-04-12 16:42:04 +02:00
2017-04-12 17:12:47 +02:00
return app
2017-04-12 16:42:04 +02:00
def login_interactive(app):
print("\nLog in to " + green(app.base_url))
email = input('Email: ')
password = getpass('Password: ')
print("Authenticating...")
2017-04-13 13:52:28 +02:00
try:
user = login(app, email, password)
except:
raise ConsoleError("Login failed")
2017-04-12 16:42:04 +02:00
save_user(user)
print("User token saved to " + green(CONFIG_USER_FILE))
return user
def print_usage():
print("toot - interact with Mastodon from the command line")
print("")
print("Usage:")
2017-04-13 13:52:28 +02:00
print(" toot login - log into a Mastodon instance (saves access tokens to `~/.config/toot/`)")
print(" toot logout - log out (delete saved access tokens)")
print(" toot auth - shows currently logged in user and instance")
print(" toot post <msg> - toot a new post to your timeline")
print(" toot timeline - shows your public timeline")
2017-04-12 16:42:04 +02:00
print("")
2017-04-15 12:12:33 +02:00
print("To get help for each command run:")
print(" toot <command> --help")
print("")
2017-04-12 16:42:04 +02:00
print("https://github.com/ihabunek/toot")
2017-04-13 13:14:01 +02:00
def print_timeline(item):
def wrap_text(text, width):
wrapper = TextWrapper(width=width, break_long_words=False, break_on_hyphens=False)
return chain(*[wrapper.wrap(l) for l in text.split("\n")])
def timeline_rows(item):
name = item['name']
time = item['time'].strftime('%Y-%m-%d %H:%M%Z')
left_column = [name, time]
if 'reblogged' in item:
left_column.append(item['reblogged'])
text = item['text']
right_column = wrap_text(text, 80)
return zip_longest(left_column, right_column, fillvalue="")
for left, right in timeline_rows(item):
print("{:30}{}".format(left, right))
def parse_timeline(item):
content = item['reblog']['content'] if item['reblog'] else item['content']
reblogged = item['reblog']['account']['username'] if item['reblog'] else ""
name = item['account']['display_name'] + " @" + item['account']['username']
soup = BeautifulSoup(content, "html.parser")
text = soup.get_text().replace('&apos;', "'")
time = datetime.strptime(item['created_at'], "%Y-%m-%dT%H:%M:%S.%fZ")
return {
"name": name,
"text": text,
"time": time,
"reblogged": reblogged,
}
def cmd_timeline(app, user):
items = timeline_home(app, user)
parsed_items = [parse_timeline(t) for t in items]
print("" * 31 + "" + "" * 88)
for item in parsed_items:
print_timeline(item)
print("" * 31 + "" + "" * 88)
2017-04-12 16:42:04 +02:00
def cmd_post_status(app, user):
parser = OptionParser(usage="toot post [options] TEXT")
parser.add_option("-m", "--media", dest="media", type="string",
help="path to the media file to attach")
parser.add_option("-v", "--visibility", dest="visibility", type="string", default="public",
help='post visibility, either "public" (default), "direct", "private", or "unlisted"')
(options, args) = parser.parse_args()
if len(args) < 2:
parser.print_help()
raise ConsoleError("No text given")
if options.visibility not in ['public', 'unlisted', 'private', 'direct']:
raise ConsoleError("Invalid visibility value given: '{}'".format(options.visibility))
if options.media:
media = do_upload(app, user, options.media)
media_ids = [media['id']]
else:
media_ids = None
2017-04-12 16:42:04 +02:00
response = post_status(
app, user, args[1], media_ids=media_ids, visibility=options.visibility)
2017-04-12 16:42:04 +02:00
2017-04-12 17:12:47 +02:00
print("Toot posted: " + green(response.get('url')))
2017-04-12 16:42:04 +02:00
def cmd_auth(app, user):
2017-04-15 12:12:33 +02:00
parser = OptionParser(usage='%prog auth')
parser.parse_args()
2017-04-12 16:42:04 +02:00
if app and user:
2017-04-13 13:14:01 +02:00
print("You are logged in to " + green(app.base_url))
2017-04-12 16:42:04 +02:00
print("Username: " + green(user.username))
2017-04-13 13:14:01 +02:00
print("App data: " + green(CONFIG_APP_FILE))
print("User data: " + green(CONFIG_USER_FILE))
2017-04-12 16:42:04 +02:00
else:
print("You are not logged in")
2017-04-15 12:12:33 +02:00
def cmd_login():
parser = OptionParser(usage='%prog login')
parser.parse_args()
app = create_app_interactive()
user = login_interactive(app)
return app, user
2017-04-13 13:52:28 +02:00
def cmd_logout(app, user):
2017-04-15 12:12:33 +02:00
parser = OptionParser(usage='%prog logout')
parser.parse_args()
2017-04-13 13:52:28 +02:00
os.unlink(CONFIG_APP_FILE)
os.unlink(CONFIG_USER_FILE)
print("You are now logged out")
def cmd_upload(app, user):
2017-04-15 12:12:33 +02:00
parser = OptionParser(usage='%prog upload <path_to_media>')
parser.parse_args()
if len(sys.argv) < 3:
print_error("No status text given")
return
2017-04-15 12:12:33 +02:00
response = do_upload(sys.argv[2])
print("\nSuccessfully uploaded media ID {}, type '{}'".format(
yellow(response['id']), yellow(response['type'])))
print("Original URL: " + green(response['url']))
print("Preview URL: " + green(response['preview_url']))
print("Text URL: " + green(response['text_url']))
def do_upload(app, user, path):
if not os.path.exists(path):
raise ConsoleError("File does not exist: " + path)
with open(path, 'rb') as f:
print("Uploading media: {}".format(green(f.name)))
return upload_media(app, user, f)
2017-04-13 13:52:28 +02:00
def run_command(command):
app = load_app()
user = load_user()
# Commands which can run when not logged in
if command == 'login':
2017-04-15 12:12:33 +02:00
return cmd_login()
2017-04-13 13:52:28 +02:00
if command == 'auth':
return cmd_auth(app, user)
# Commands which require user to be logged in
if not app or not user:
print(red("You are not logged in."))
print(red("Please run `toot login` first."))
return
if command == 'logout':
return cmd_logout(app, user)
2017-04-12 16:42:04 +02:00
2017-04-13 13:52:28 +02:00
if command == 'post':
return cmd_post_status(app, user)
if command == 'timeline':
return cmd_timeline(app, user)
if command == 'upload':
return cmd_upload(app, user)
2017-04-13 13:52:28 +02:00
print(red("Unknown command '{}'\n".format(command)))
print_usage()
def main():
2017-04-12 16:42:04 +02:00
if os.getenv('TOOT_DEBUG'):
logging.basicConfig(level=logging.DEBUG)
2017-04-13 13:52:28 +02:00
command = sys.argv[1] if len(sys.argv) > 1 else None
2017-04-12 16:42:04 +02:00
2017-04-13 13:52:28 +02:00
if not command:
return print_usage()
try:
run_command(command)
except ConsoleError as e:
print_error(str(e))