Mitter/mitterlib/ui/ui_tty.py

#!/usr/bin/python
# -*- coding: utf-8 -*-

# Mitter, a Maemo client for Twitter.
# Copyright (C) 2007, 2008  Julio Biason, Deepak Sarda
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

NAMESPACE = 'tty'

import mitterlib.ui.console_utils as console_utils
import logging
from mitterlib.network.networkbase import NetworkError
from mitterlib.network import NetworksNoNetworkSetupError, NetworksError

_log = logging.getLogger('ui.tty')


def options(options):
    """Add command line options for this interface."""
    options.add_group(NAMESPACE, 'TTY interface')   # This surely needs a
                                                    # better description
    options.add_option('--messages',
            group=NAMESPACE,
            option='messages',
            help='Display the latest messages.',
            default=False,
            action='store_true',
            is_config_option=False,
            conflict_group='interface')
    options.add_option('--update',
            group=NAMESPACE,
            option='update',
            default=None,
            help='Update your status',
            metavar='STATUS',
            type='str',
            is_config_option=False,
            conflict_group='interface')
    options.add_option('--replies',
            group=NAMESPACE,
            option='replies',
            help='Get a list of replies instead of the friends timeline',
            default=False,
            action='store_true',
            is_config_option=False,
            conflict_group='interface')
    return


class Interface(object):
    """The console/tty interface for Mitter."""

    # -----------------------------------------------------------------------
    # Private functions
    # -----------------------------------------------------------------------

    def _config(self):
        """Set up the networks."""
        options = self._connection.settings()
        console_utils.authorization(options, self._options)
        return

    def _messages(self):
        """Show the latest messages."""
        try:
            data = self._connection.messages()
            console_utils.print_messages(data, self._connection)
        except NetworksNoNetworkSetupError:
            # call the config
            self._config()
        except NetworkError:
            print 'Network failure. Try again in a few minutes.'
        return

    def _replies(self):
        """Show the replies to you."""
        try:
            data = self._connection.replies()
            console_utils.print_messages(data, self._connection)
        except NetworksNoNetworkSetupError:
            # call the config
            self._config()
        except NetworkError:
            print 'Network failure. Try again in a few minutes.'
        return

    def _update(self, message):
        """Update your status."""
        try:
            self._connection.update(message)
        except (NetworksError, NetworkError):
            print 'Error updating your status. Try again in a few minutes.'

        return


    # -----------------------------------------------------------------------
    # Methods required by the main Mitter code
    # -----------------------------------------------------------------------

    def __init__(self, connection, options):
        """Class initialization."""
        self._connection = connection
        self._options = options

    def __call__(self):
        """The callable function, used by mitter to start the interface."""
        status_message = self._options[NAMESPACE]['update']
        if status_message:
            self._update(status_message)
            return

        if self._options[NAMESPACE]['replies']:
            self._replies()
            return

        self._messages()
        return