def upload(app, user, args):
response = _do_upload(app, user, args.file)
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 auth(app, user, args):
if app and user:
print("You are logged in to {} as {}\n".format(yellow(app.instance),
yellow(user.username)))
print("User data: " + green(config.get_user_config_path()))
print("App data: " +
green(config.get_instance_config_path(app.instance)))
else:
print("You are not logged in")
def two_factor_login_interactive(app):
"""Hacky implementation of two factor authentication"""
print("Log in to " + green(app.instance))
email = input('Email: ')
password = getpass('Password: '******'/auth/sign_in'
session = requests.Session()
# Fetch sign in form
response = session.get(sign_in_url)
response.raise_for_status()
soup = BeautifulSoup(response.content, "html.parser")
form = soup.find('form')
inputs = form.find_all('input')
data = {i.attrs.get('name'): i.attrs.get('value') for i in inputs}
data['user[email]'] = email
data['user[password]'] = password
# Submit form, get 2FA entry form
response = session.post(sign_in_url, data)
response.raise_for_status()
soup = BeautifulSoup(response.content, "html.parser")
form = soup.find('form')
inputs = form.find_all('input')
data = {i.attrs.get('name'): i.attrs.get('value') for i in inputs}
data['user[otp_attempt]'] = input("2FA Token: ")
# Submit token
response = session.post(sign_in_url, data)
response.raise_for_status()
# Extract access token from response
soup = BeautifulSoup(response.content, "html.parser")
initial_state = soup.find('script', id='initial-state')
if not initial_state:
raise ConsoleError("Login failed: Invalid 2FA token?")
data = json.loads(initial_state.get_text())
access_token = data['meta']['access_token']
user = User(app.instance, email, access_token)
path = config.save_user(user)
print("Access token saved to: " + green(path))
def register_app(instance):
print("Registering application with %s" % green(instance))
try:
response = api.create_app(instance)
except:
raise ConsoleError(
"Registration failed. Did you enter a valid instance?")
base_url = 'https://' + instance
app = App(instance, base_url, response['client_id'],
response['client_secret'])
path = config.save_app(app)
print("Application tokens saved to: {}\n".format(green(path)))
return app
def _print_accounts(accounts):
if not accounts:
return
print("\nAccounts:")
for account in accounts:
acct = green("@{}".format(account['acct']))
display_name = account['display_name']
print("* {} {}".format(acct, display_name))
def _print_account(account):
print("{} {}".format(green("@" + account['acct']),
account['display_name']))
if account['note']:
print("")
note = BeautifulSoup(account['note'], "html.parser")
print("\n".join(wrap(note.get_text())))
print("")
print("ID: " + green(account['id']))
print("Since: " + green(account['created_at'][:19].replace('T', ' @ ')))
print("")
print("Followers: " + yellow(account['followers_count']))
print("Following: " + yellow(account['following_count']))
print("Statuses: " + yellow(account['statuses_count']))
print("")
print(account['url'])
def unfollow(app, user, args):
account = _find_account(app, user, args.account)
if not account:
print_error("Account not found")
return
api.unfollow(app, user, account['id'])
print(green("✓ You are no longer following %s" % args.account))
def login_interactive(app):
print("\nLog in to " + green(app.instance))
email = input('Email: ')
password = getpass('Password: '******'access_token'])
path = config.save_user(user)
print("Access token saved to: " + green(path))
return user
def login_2fa(app, user, args):
print()
print(yellow("Two factor authentication is experimental."))
print(yellow("If you have problems logging in, please open an issue:"))
print(yellow("https://github.com/ihabunek/toot/issues"))
print()
app = create_app_interactive()
two_factor_login_interactive(app)
print()
print(green("✓ Successfully logged in."))
def post(app, user, args):
if args.media:
media = _do_upload(app, user, args.media)
media_ids = [media['id']]
else:
media_ids = None
response = api.post_status(app,
user,
args.text,
media_ids=media_ids,
visibility=args.visibility)
print("Toot posted: " + green(response.get('url')))
def _print_hashtags(hashtags):
if not hashtags:
return
print("\nHashtags:")
print(", ".join([green("#" + t) for t in hashtags]))
def create_app_interactive():
instance = input("Choose an instance [%s]: " % green(DEFAULT_INSTANCE))
if not instance:
instance = DEFAULT_INSTANCE
return config.load_app(instance) or register_app(instance)
def unblock(app, user, args):
account = _find_account(app, user, args.account)
api.unblock(app, user, account['id'])
print(green("✓ %s is no longer blocked" % args.account))
def block(app, user, args):
account = _find_account(app, user, args.account)
api.block(app, user, account['id'])
print(green("✓ You are now blocking %s" % args.account))
def mute(app, user, args):
account = _find_account(app, user, args.account)
api.mute(app, user, account['id'])
print(green("✓ You have muted %s" % args.account))
def unfollow(app, user, args):
account = _find_account(app, user, args.account)
api.unfollow(app, user, account['id'])
print(green("✓ You are no longer following %s" % args.account))
def login(app, user, args):
app = create_app_interactive()
login_interactive(app)
print()
print(green("✓ Successfully logged in."))
def _do_upload(app, user, file):
print("Uploading media: {}".format(green(file.name)))
return api.upload_media(app, user, file)
def logout(app, user, args):
config.delete_user()
print(green("✓ You are now logged out"))
