Custom prefix with formatting

[rainbowstream.git] / rainbowstream / util.py

import json

from twitter.util import printNicely
from .colors import *
from .config import *


def detail_twitter_error(twitterException):
    """
    Display Twitter Errors nicely
    """
    try:
        # twitterException.response_data can be byte string on Python 3
        # or nornal dict on Python 2
        loadedJson = json.loads(twitterException.response_data.decode('utf8'))
        for m in loadedJson.get('errors', dict()):
            info = "Error " + str(m.get('code')) + ": " + m.get('message')
            printNicely(yellow(info))
    except:
        info = "Error: " + twitterException.response_data.decode('utf8')
        printNicely(yellow(info))


def format_prefix(listname = '', keyword = ''):
    """
    Format the custom prefix
    """
    formattedPrefix = c['PREFIX']
    username = '@' + c['original_name']
    place = ''
    if keyword != '':
        place = '/public'
        keyword = '#' + keyword

    if listname != '':
        username, place = listname.split('/')
        place = "/" + place

    formattedPrefix = formattedPrefix.replace("#username", username)
    formattedPrefix = formattedPrefix.replace("#place", place)
    formattedPrefix = formattedPrefix.replace("#keyword", keyword)

    return formattedPrefix