source: pycodebattler/trunk/pycodebattler/util.py @ 71

# -*- coding: utf-8 -*-
#
#  Copyright (C) 2010 by Atzm WATANABE <atzm@atzm.org>
#
#  This program is free software; you can redistribute it and/or modify it
#  under the terms of the GNU General Public License (version 2) as
#  published by the Free Software Foundation.  It 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.
#
# $Id$
#

import os


_COLOR_ENABLE = True


def set_color_enable(flag):
    global _COLOR_ENABLE
    _COLOR_ENABLE = flag


def _color(str_, color):
    global _COLOR_ENABLE
    list_ = [str(str_)]
    if _COLOR_ENABLE:
        list_.insert(0, color)
        list_.append('\033[0m')
    return ''.join(list_)


def red(str_):
    return _color(str_, '\033[1;31m')


def green(str_):
    return _color(str_, '\033[1;32m')


def yellow(str_):
    return _color(str_, '\033[1;33m')


def cyan(str_):
    return _color(str_, '\033[1;36m')


def isupper_one(str_):
    return len([c for c in str_ if c.isupper()]) > 0


def path2name(path):
    name = os.path.splitext(os.path.basename(path))[0]
    if name.find('_') < 0 and isupper_one(name):
        return name
    return ''.join([p.capitalize() for p in name.split('_')])