102 lines
2.9 KiB
Python
102 lines
2.9 KiB
Python
import socket
|
|
import random
|
|
from threading import Thread
|
|
from datetime import datetime
|
|
from colorama import Fore, init, Back
|
|
#var
|
|
username="no"
|
|
class var:
|
|
join_msg=" joined"
|
|
debug_usr_name="Client_Test"
|
|
# null is disabled
|
|
# enabled is enabled
|
|
debug_enabled="null"
|
|
seperator="<SEP>"
|
|
serverid="Server"
|
|
leave="left"
|
|
init()
|
|
|
|
|
|
colors = [Fore.BLUE, Fore.CYAN, Fore.GREEN, Fore.LIGHTBLACK_EX,
|
|
Fore.LIGHTBLUE_EX, Fore.LIGHTCYAN_EX, Fore.LIGHTGREEN_EX,
|
|
Fore.LIGHTMAGENTA_EX, Fore.LIGHTRED_EX, Fore.LIGHTWHITE_EX,
|
|
Fore.LIGHTYELLOW_EX, Fore.MAGENTA, Fore.RED, Fore.WHITE, Fore.YELLOW
|
|
]
|
|
|
|
|
|
client_color = random.choice(colors)
|
|
|
|
host_ip=input("Enter host IP: ")
|
|
host_port=input("Enter host PORT: ")
|
|
serverHost = host_ip
|
|
serverPort = int(host_port)
|
|
separator_token = var.seperator # we will use this to separate the client name & message
|
|
|
|
# initialize TCP socket
|
|
s = socket.socket()
|
|
print(f"[*] Connecting to {serverHost}:{serverPort}...")
|
|
# connect to the server
|
|
s.connect((serverHost, serverPort))
|
|
print("[+] Connected.")
|
|
|
|
if var.debug_enabled != "enabled":
|
|
username = input("Enter name: ")
|
|
elif var.debug_enabled == "enabled":
|
|
username=var.debug_usr_name
|
|
|
|
print("[*] Logged in as", username)
|
|
print("Colors: Blue, Cyan, Green, Lime, Magenta, Red, Yellow, White, Anything else = Random")
|
|
colorInputFirst = input("Enter color: ")
|
|
color_input = colorInputFirst.lower()
|
|
if (color_input == "blue"):
|
|
client_color = Fore.BLUE
|
|
elif (color_input == "cyan"):
|
|
client_color = Fore.CYAN
|
|
elif (color_input == "green"):
|
|
client_color = Fore.GREEN
|
|
elif (color_input == "lime"):
|
|
client_color = Fore.LIGHTGREEN_EX
|
|
elif (color_input == "magenta"):
|
|
client_color = Fore.MAGENTA
|
|
elif (color_input == "red"):
|
|
client_color = Fore.RED
|
|
elif (color_input == "yellow"):
|
|
client_color = Fore.YELLOW
|
|
elif (color_input == "white"):
|
|
client_color = Fore.WHITE
|
|
elif (color_input != "white"):
|
|
client_color = random.choice(colors)
|
|
usr_join_msg = f"{username}{var.join_msg}"
|
|
date_now = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
|
|
usr_join_msg = f"{client_color}[{date_now}] {var.serverid}{separator_token}{usr_join_msg}\n"
|
|
s.send(usr_join_msg.encode())
|
|
|
|
|
|
def listen_for_messages():
|
|
while True:
|
|
message = s.recv(1024).decode()
|
|
print("\n" + message)
|
|
|
|
# make a thread that listens for messages to this client & print them
|
|
t = Thread(target=listen_for_messages)
|
|
# make the thread daemon so it ends whenever the main thread ends
|
|
t.daemon = True
|
|
# start the thread
|
|
t.start()
|
|
|
|
while True:
|
|
# input message we want to send to the server
|
|
to_send = input()
|
|
# a way to exit the program
|
|
if to_send.lower() == 'q':
|
|
break
|
|
# add the datetime, name & the color of the sender
|
|
date_now = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
|
|
to_send = f"{client_color}[{date_now}] {username}{separator_token}{to_send}"
|
|
# finally, send the message
|
|
s.send(to_send.encode())
|
|
|
|
# close the socket
|
|
s.close()
|
|
|