Mitter/mitterlib/ui/helpers/console_utils.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/>.

import textwrap
import locale
import getpass

from mitterlib.network.networkbase import NetworkData

from timesince import timesince


def encode_print(text):
    """Try to print the text; if we get any UnicodeEncodeError, we print it
    without encoding."""
    try:
        print text
    except UnicodeEncodeError:
        encoding = locale.getdefaultlocale()[1]
        if not encoding:
            encoding = 'ascii'
        print text.encode(encoding, 'replace')
        return


def print_messages(data, connection, show_numbers=False, indent=0):
    """Print the list of messages."""
    count = 0
    # the wrapping thing
    indent_text = ' ' * (indent * 3)
    wrapper = textwrap.TextWrapper()
    wrapper.initial_indent = indent_text
    wrapper.subsequent_indent = indent_text

    if isinstance(data, NetworkData):
        # If it's a single message, just print it
        _display_message(wrapper, data, connection, show_numbers, count,
                indent_text)
        return
    elif isinstance(data, str):
        # If it just text, just print it.
        _display_text(wrapper, data)
        return

    # Twitter sends us the data from the newest to latest, which is not
    # good for displaying on a console. So we reverse the list.
    # TODO: Check if this is true for ALL networks. Otherwise, we'll have
    #       problems.
    data.reverse()

    print
    for message in data:
        count += 1

        if isinstance(message, str):
            _display_text(wrapper, message)
        else:
            _display_message(wrapper, message, connection,
                    show_numbers, count, indent_text)
    return


def _display_message(wrapper, message, connection, show_numbers,
        count, indent_text):
    """Print a single message (NetworkData)."""
    display = []
    if show_numbers:
        display.append('%d.' % (count))

    if message.username != message.name:
        display.append('%s (%s):' % (message.username, message.name))
    else:
        display.append('%s:' % (message.username))
    display.append(message.message)

    msg = ' '.join(display)
    for line in wrapper.wrap(msg):
        encode_print(line)

    footer = '(%s ago, in %s [%s])' % (timesince(message.message_time),
                connection.name(message.network), message.network)
    for line in wrapper.wrap(footer):
        encode_print(line)

    print
    print


def _display_text(wrapper, text):
    """Print a formated text."""
    text = wrapper.wrap(text)
    for line in text:
        encode_print(line)
    return


def authorization(options, config):
    for network in options:
        namespace = network['name']

        network_title = '%s credentials:' % (namespace)
        print network_title
        print '-' * len(network_title)

        for option in network['options']:
            name = option['name']
            try:
                old_value = config[namespace][name]
            except KeyError:
                # not setup yet (so it's not in the config -- it would
                # probably never happen, since we are adding all options
                # in the config right from start, but better safe than sorry)
                old_value = None

            if old_value:
                old_value_prompt = ' (%s)' % (old_value)
            else:
                old_value_prompt = ''

            prompt = '%s%s: ' % (option['prompt'], old_value_prompt)

            if option['type'] == 'str':
                value = raw_input(prompt)
            elif option['type'] == 'passwd':
                value = getpass.getpass(prompt)

            if value:
                config[namespace][name] = value
        print