alone-chat/alone-chat.py at master · VonKavalier/alone-chat · 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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
#!/usr/bin/env python3
# -*- coding: utf8 -*-
"""Alone-chat.py: Express yourself without hurting anyone"""

import os
from datetime import datetime

__author__ = "Tom Celestin"
__copyright__ = "Copyright 2018, Planet Earth"

def print_welcome():
    """Display welcome message."""
    os.system('clear')
    welcome = """
=================================================================================
    Welcome to the

   mm   \"\"#                                  mmm  #               m
   ##     #     mmm   m mm    mmm          m\"   \" # mm    mmm   mm#mm
  #  #    #    #\" \"#  #\"  #  #\"  #         #      #\"  #  \"   #    #
  #mm#    #    #   #  #   #  #\"\"\"\"   \"\"\"   #      #   #  m\"\"\"#    #
 #    #   \"mm  \"#m#\"  #   #  \"#mm\"          \"mmm\" #   #  \"mm\"#    \"mm

    You can express yourself without hurting anyone
=================================================================================
    """
    print(welcome)


def ask_username():
    """Ask user's username."""
    username = input("Please enter your username: ")
    return username


def get_time(format):
    """Get current time formatted."""
    i = datetime.now()
    date_string = i.strftime(format)
    return date_string


def message_as(username):
    "Format prompt according to username."
    formatting = "[\033[94m" + username + "\033[0m] > "
    if username == "system":
        formatting = "[\033[93msystem\033[0m] > "
    return get_time('%H:%M:%S') + " " + formatting


def system_message(username, joins):
    """Displays join and quit messages."""
    message = ""
    status = " \033[91mhas quit\033[0m"
    if joins:
        message += "\n"
        status =  " \033[92mhas joined\033[0m"
    message += message_as("system") + username + status
    print(message)


def save_logs(logs):
    """Save logs array in a file."""
    date_string = get_time('%y%m%d%H%M%S')
    filename = date_string + "_alone-chat_logs.txt"
    file = open(filename,"w")
    for message in logs:
        file.write(message+"\n")
    file.close()


def read_command(command):
    """Return value depending on the command."""
    if command.startswith("/"):
        if command == "/help" or command == "/?":
            print(message_as("system") + "/hide   : Toggle hiding or showing following messages")
            print(message_as("system") + "/export : Export messages from current conversation and quit")
            print(message_as("system") + "/quit   : Quit Alone-Chat")
            print(message_as("system") + "/help   : Display this help")
            return 0
        elif command == "/export":
            return 1
        elif command == "/hide":
            return 2
        else:
            return 3
    else:
        return 0


def interface(username):
    """Display prompt and manage imputs."""
    system_message(username, True)
    formatted_username = message_as(username)
    hide = False
    check_command = 0
    logs = []
    while check_command not in [1,3]:
        message = input(formatted_username)
        logs.append(message)
        check_command = read_command(message)
        if check_command == 2:
            hide = not hide
        if hide:
            CURSOR_UP_ONE = '\x1b[1A'
            ERASE_LINE = '\x1b[2K'
            print(CURSOR_UP_ONE + ERASE_LINE)
    if read_command(message) == 1:
        save_logs(logs)

    system_message(username, False)


def main():
    """The beginning of everything."""
    print_welcome()
    username = ask_username()
    interface(username)
    exit(1)

if __name__ == "__main__":
    main()