pytools/fancy_message.py at master · dlinten/pytools · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
#!/usr/bin/env python3

########################################################################
#                                                                      #
# Copyright 2016 (c) Araluc Ltd.  All rights reserved.                 #
# For copying and use see:                                             #
#  https://github.com/dlinten/pytools/blob/master/LICENSE              #
#                                                                      #
# Company     : Araluc Ltd.                                            #
# Web site    : araluc.com                                             #
# Email       : info@araluc.com                                        #
# Author      : David Linten                                           #
# Date        : 2016-11-09 12:22:43.653640                             #
#                                                                      #
# Name        : fancy_message                                          #
# Description : Prints our coloured fancy messages to the console      #
#                                                                      #
########################################################################


import os


class Colour:
    HEADER = '\033[95m'
    OKBLUE = '\033[94m'
    OKGREEN = '\033[92m'
    WARNING = '\033[93m'
    FAIL = '\033[91m'
    ENDC = '\033[0m'
    BOLD = '\033[1m'
    UNDERLINE = '\033[4m'


def check_path_arg(console_arg, question: str, default_value: str, is_path=False):
    if console_arg is None or console_arg == "":
        user_input = input("{cstart}{question}{cend} {cbold}[{default_value}]{cstart}?{cend} ".format(
            cstart=Colour.OKGREEN,
            cend=Colour.ENDC,
            cbold=Colour.BOLD,
            question=question,
            default_value=default_value))
        console_arg = default_value if user_input == "" else user_input
    if is_path:
        console_arg = os.path.abspath(os.path.expanduser(console_arg))
        if not os.path.exists(console_arg) and not os.path.isfile(console_arg):
            Message.error("Invalid argument, path or file does not exist!")
            console_arg = check_path_arg("", question, default_value, is_path)
    return console_arg


class Message:
    @staticmethod
    def header(message: str, ending='\n'):
        print('{}{}{}'.format(Colour.HEADER, message, Colour.ENDC), end=ending)

    @staticmethod
    def info(message: str, ending='\n'):
        print('{}{}{}'.format(Colour.OKGREEN, message, Colour.ENDC), end=ending)

    @staticmethod
    def warn(message: str, ending='\n'):
        print('{}{}{}'.format(Colour.WARNING, message, Colour.ENDC), end=ending)

    @staticmethod
    def error(message: str, ending='\n'):
        print('{}{}{}'.format(Colour.FAIL, message, Colour.ENDC), end=ending)

    @staticmethod
    def output(message: str, ending='\n'):
        print(message, end=ending)