diff --git a/toot/console.py b/toot/console.py index 51c1d5e..82124aa 100644 --- a/toot/console.py +++ b/toot/console.py @@ -15,6 +15,10 @@ from toot.output import print_out, print_err VISIBILITY_CHOICES = ['public', 'unlisted', 'private', 'direct'] +def get_default_visibility(): + return os.getenv("TOOT_VISIBILITY", "public") + + def language(value): """Validates the language parameter""" if len(value) != 2: @@ -348,7 +352,7 @@ POST_COMMANDS = [ }), (["-v", "--visibility"], { "type": visibility, - "default": os.getenv("TOOT_VISIBILITY", "public"), + "default": get_default_visibility(), "help": 'post visibility, one of: %s' % ", ".join(VISIBILITY_CHOICES), }), (["-s", "--sensitive"], { @@ -437,7 +441,7 @@ STATUS_COMMANDS = [ arguments=[status_id_arg, (["-v", "--visibility"], { "type": visibility, - "default": os.getenv("TOOT_VISIBILITY", "public"), + "default": get_default_visibility(), "help": 'boost visibility, one of: %s' % ", ".join(VISIBILITY_CHOICES), })], require_auth=True, diff --git a/toot/tui/app.py b/toot/tui/app.py index 39b8529..9f917ec 100644 --- a/toot/tui/app.py +++ b/toot/tui/app.py @@ -1,10 +1,10 @@ import logging import urwid -import os from concurrent.futures import ThreadPoolExecutor from toot import api, config, __version__ +from toot.console import get_default_visibility from .compose import StatusComposer from .constants import PALETTE @@ -495,7 +495,7 @@ class TUI(urwid.Frame): def async_toggle_reblog(self, timeline, status): def _reblog(): logger.info("Reblogging {}".format(status)) - api.reblog(self.app, self.user, status.id, visibility=os.getenv("TOOT_VISIBILITY", "public")) + api.reblog(self.app, self.user, status.id, visibility=get_default_visibility()) def _unreblog(): logger.info("Unreblogging {}".format(status)) diff --git a/toot/tui/compose.py b/toot/tui/compose.py index 9eb742e..4b31c0f 100644 --- a/toot/tui/compose.py +++ b/toot/tui/compose.py @@ -1,6 +1,7 @@ import urwid import logging -import os + +from toot.console import get_default_visibility from .constants import VISIBILITY_OPTIONS from .widgets import Button, EditBox @@ -32,7 +33,7 @@ class StatusComposer(urwid.Frame): self.cw_remove_button = Button("Remove content warning", on_press=self.remove_content_warning) - self.visibility = os.getenv("TOOT_VISIBILITY", "public") + self.visibility = get_default_visibility() self.visibility_button = Button("Visibility: {}".format(self.visibility), on_press=self.choose_visibility)