forked from niannian1145/python-chat-box
上传文件至「/」
正式版1.0
This commit is contained in:
+756
@@ -0,0 +1,756 @@
|
||||
import socket
|
||||
import threading
|
||||
import time
|
||||
import os
|
||||
import sys
|
||||
import json
|
||||
|
||||
# 终端颜色定义
|
||||
class Color:
|
||||
R = '\033[0m' # 重置
|
||||
C = '\033[96m' # 青色
|
||||
Y = '\033[93m' # 黄色
|
||||
R_ = '\033[91m' # 红色
|
||||
G = '\033[92m' # 绿色
|
||||
|
||||
# 聊天工具核心类
|
||||
class ChatRoom:
|
||||
# 广播配置
|
||||
ROOM_BROADCAST_PORT = 5000
|
||||
ROOM_BROADCAST_INTERVAL = 3
|
||||
ROOM_TIMEOUT = 10
|
||||
|
||||
def __init__(self):
|
||||
# 基础信息
|
||||
self.username = ""
|
||||
self.client_socket = None
|
||||
self.is_host = False
|
||||
self.room_ip = ""
|
||||
self.room_port = 0
|
||||
self.user_color = Color.Y
|
||||
self.running = True
|
||||
self.connected = False
|
||||
|
||||
# 房间管理
|
||||
self.clients = {}
|
||||
self.discovered_rooms = {}
|
||||
self.broadcast_thread = None
|
||||
self.discovery_thread = None
|
||||
self.room_locked = False
|
||||
self.room_password = ""
|
||||
|
||||
# 权限控制
|
||||
self.muted_users = set()
|
||||
self.global_mute = False
|
||||
|
||||
# 用户注册
|
||||
def register(self):
|
||||
os.system('cls' if os.name == 'nt' else 'clear')
|
||||
while True:
|
||||
name = input(f"{Color.C}[系统提示] 请输入用户名：{Color.R}".strip())
|
||||
if name and not ' ' in name and len(name) <= 10:
|
||||
self.username = name
|
||||
break
|
||||
print(f"{Color.C}[系统提示] 用户名无效 需1-10个字符且不含空格{Color.R}")
|
||||
|
||||
# 清屏
|
||||
def clear_screen(self):
|
||||
os.system('cls' if os.name == 'nt' else 'clear')
|
||||
|
||||
# 主菜单
|
||||
def show_main_menu(self):
|
||||
while self.running and not self.connected:
|
||||
self.clear_screen()
|
||||
print(f"{Color.C}{'='*40}{Color.R}")
|
||||
print(f"{Color.C}{'[Python简易聊天工具]':^40}{Color.R}")
|
||||
print(f"{Color.C}{'='*40}{Color.R}")
|
||||
print(f"{Color.C}{'1. 创建房间':<38}{'[ 创建专属房间 ]'}{Color.R}")
|
||||
print(f"{Color.C}{'2. 加入房间':<38}{'[ 输入地址加入 ]'}{Color.R}")
|
||||
print(f"{Color.C}{'3. 查找房间':<38}{'[ 自动搜索局域网 ]'}{Color.R}")
|
||||
print(f"{Color.C}{'4. 退出脚本':<38}{'[ 关闭聊天工具 ]'}{Color.R}")
|
||||
print(f"{Color.C}{'='*40}{Color.R}")
|
||||
choice = input(f"{Color.C}[请选择功能(1-4)]：{Color.R}".strip())
|
||||
if choice == "1":
|
||||
self.create_room()
|
||||
elif choice == "2":
|
||||
self.join_room_manual()
|
||||
elif choice == "3":
|
||||
self.discover_rooms()
|
||||
elif choice == "4":
|
||||
self.exit_script()
|
||||
else:
|
||||
print(f"{Color.C}[系统提示] 无效选择 请输入1-4之间的数字{Color.R}")
|
||||
time.sleep(1)
|
||||
|
||||
# 获取本地IP
|
||||
def get_local_ip(self):
|
||||
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||||
try:
|
||||
s.connect(('8.8.8.8', 80))
|
||||
ip = s.getsockname()[0]
|
||||
except Exception:
|
||||
ip = '127.0.0.1'
|
||||
finally:
|
||||
s.close()
|
||||
return ip
|
||||
|
||||
# 房间信息广播
|
||||
def broadcast_room_info(self):
|
||||
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||||
sock.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
|
||||
while self.connected and self.is_host:
|
||||
try:
|
||||
room_info = json.dumps({
|
||||
'ip': self.room_ip,
|
||||
'port': self.room_port,
|
||||
'host': self.username,
|
||||
'online': len(self.clients) + 1,
|
||||
'locked': self.room_locked,
|
||||
'timestamp': time.time()
|
||||
})
|
||||
sock.sendto(room_info.encode('utf-8'), ('<broadcast>', self.ROOM_BROADCAST_PORT))
|
||||
time.sleep(self.ROOM_BROADCAST_INTERVAL)
|
||||
except Exception:
|
||||
break
|
||||
sock.close()
|
||||
|
||||
# 查找房间
|
||||
def discover_rooms(self):
|
||||
self.clear_screen()
|
||||
print(f"{Color.C}{'='*40}{Color.R}")
|
||||
print(f"{Color.C}{'[ 房间查找 ]':^40}{Color.R}")
|
||||
print(f"{Color.C}{'='*40}{Color.R}")
|
||||
print(f"{Color.C}[系统提示] 正在查找局域网房间 按Ctrl+C停止{Color.R}")
|
||||
print(f"{Color.C}{'-'*40}{Color.R}")
|
||||
self.discovered_rooms = {}
|
||||
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||||
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||||
sock.bind(('', self.ROOM_BROADCAST_PORT))
|
||||
sock.settimeout(1.0)
|
||||
|
||||
def discovery():
|
||||
start_time = time.time()
|
||||
while time.time() - start_time < 5 and self.running:
|
||||
try:
|
||||
data, addr = sock.recvfrom(1024)
|
||||
info = json.loads(data.decode('utf-8'))
|
||||
if time.time() - info['timestamp'] < self.ROOM_TIMEOUT:
|
||||
self.discovered_rooms[f"{info['ip']}:{info['port']}"] = info
|
||||
except socket.timeout:
|
||||
continue
|
||||
except Exception:
|
||||
break
|
||||
|
||||
self.discovery_thread = threading.Thread(target=discovery, daemon=True)
|
||||
self.discovery_thread.start()
|
||||
self.discovery_thread.join()
|
||||
sock.close()
|
||||
|
||||
self.clear_screen()
|
||||
print(f"{Color.C}{'='*40}{Color.R}")
|
||||
print(f"{Color.C}{'[ 查找结果 ]':^40}{Color.R}")
|
||||
print(f"{Color.C}{'='*40}{Color.R}")
|
||||
if not self.discovered_rooms:
|
||||
print(f"{Color.C}[系统提示] 未找到任何房间{Color.R}")
|
||||
print(f"{Color.C}{'-'*40}{Color.R}")
|
||||
input(f"{Color.C}[系统提示] 按回车键返回主菜单{Color.R}")
|
||||
return
|
||||
print(f"{Color.C}{'序号':<5}{'房间地址':<20}{'房主':<10}{'在线人数'}{'是否锁定'}{Color.R}")
|
||||
print(f"{Color.C}{'-'*40}{Color.R}")
|
||||
rooms = list(self.discovered_rooms.values())
|
||||
for i, room in enumerate(rooms, 1):
|
||||
addr = f"{room['ip']}:{room['port']}"
|
||||
locked = "是" if room['locked'] else "否"
|
||||
print(f"{Color.C}{i:<5}{addr:<20}{room['host']:<10}{room['online']:<8}{locked}{Color.R}")
|
||||
print(f"{Color.C}{'-'*40}{Color.R}")
|
||||
while True:
|
||||
choice = input(f"{Color.C}[请选择房间序号加入 0返回主菜单]：{Color.R}".strip())
|
||||
if choice == "0":
|
||||
return
|
||||
if choice.isdigit() and 1 <= int(choice) <= len(rooms):
|
||||
room = rooms[int(choice)-1]
|
||||
self.room_ip = room['ip']
|
||||
self.room_port = room['port']
|
||||
self.room_locked = room['locked']
|
||||
self.join_room()
|
||||
break
|
||||
print(f"{Color.C}[系统提示] 无效选择 请输入正确序号{Color.R}")
|
||||
|
||||
# 创建房间
|
||||
def create_room(self):
|
||||
self.clear_screen()
|
||||
print(f"{Color.C}{'='*40}{Color.R}")
|
||||
print(f"{Color.C}{'[ 创建房间 ]':^40}{Color.R}")
|
||||
print(f"{Color.C}{'='*40}{Color.R}")
|
||||
self.is_host = True
|
||||
self.room_ip = self.get_local_ip()
|
||||
|
||||
while True:
|
||||
port_input = input(f"{Color.C}[请输入端口号 1024-65535 默认8888]：{Color.R}".strip())
|
||||
self.room_port = 8888 if not port_input else int(port_input)
|
||||
if 1024 <= self.room_port <= 65535:
|
||||
break
|
||||
print(f"{Color.C}[系统提示] 端口号无效 需在1024-65535之间{Color.R}")
|
||||
|
||||
while True:
|
||||
lock_choice = input(f"{Color.C}[是否开启房间锁 1=开启 0=关闭 默认0]：{Color.R}".strip())
|
||||
lock_choice = lock_choice if lock_choice else "0"
|
||||
if lock_choice in ["0", "1"]:
|
||||
self.room_locked = (lock_choice == "1")
|
||||
break
|
||||
print(f"{Color.C}[系统提示] 无效选择 请输入0或1{Color.R}")
|
||||
|
||||
if self.room_locked:
|
||||
while True:
|
||||
pwd = input(f"{Color.C}[请设置房间密码 4-8位字符]：{Color.R}".strip())
|
||||
if 4 <= len(pwd) <= 8 and not ' ' in pwd:
|
||||
self.room_password = pwd
|
||||
break
|
||||
print(f"{Color.C}[系统提示] 密码无效 需4-8位字符且不含空格{Color.R}")
|
||||
|
||||
print(f"{Color.C}{'-'*40}{Color.R}")
|
||||
print(f"{Color.C}[系统提示] 正在创建房间{Color.R}")
|
||||
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||||
try:
|
||||
server.bind((self.room_ip, self.room_port))
|
||||
server.listen(5)
|
||||
self.connected = True
|
||||
self.broadcast_thread = threading.Thread(target=self.broadcast_room_info, daemon=True)
|
||||
self.broadcast_thread.start()
|
||||
print(f"{Color.C}[系统提示] 房间创建成功{Color.R}")
|
||||
print(f"{Color.C}[房间地址] {self.room_ip}:{self.room_port}{Color.R}")
|
||||
print(f"{Color.C}[房间状态] {'已锁定 需密码加入' if self.room_locked else '未锁定 公开加入'}{Color.R}")
|
||||
print(f"{Color.C}[操作提示] 输入/房主菜单 查看专属功能{Color.R}")
|
||||
print(f"{Color.Y}[系统提示] {self.username} 创建并加入房间{Color.R}")
|
||||
print(f"{Color.C}{'-'*40}{Color.R}")
|
||||
threading.Thread(target=self.accept_clients, args=(server,), daemon=True).start()
|
||||
self.client_chat_loop()
|
||||
except PermissionError:
|
||||
print(f"{Color.C}[系统提示] 创建房间失败 该端口已被占用 请更换端口{Color.R}")
|
||||
self.is_host = False
|
||||
time.sleep(2)
|
||||
except Exception as e:
|
||||
print(f"{Color.C}[系统提示] 创建房间失败 原因：{str(e)}{Color.R}")
|
||||
self.is_host = False
|
||||
time.sleep(2)
|
||||
|
||||
# 手动加入房间
|
||||
def join_room_manual(self):
|
||||
self.clear_screen()
|
||||
print(f"{Color.C}{'='*40}{Color.R}")
|
||||
print(f"{Color.C}{'[ 手动加入 ]':^40}{Color.R}")
|
||||
print(f"{Color.C}{'='*40}{Color.R}")
|
||||
self.room_ip = input(f"{Color.C}[请输入房间IP]：{Color.R}".strip())
|
||||
while True:
|
||||
port_input = input(f"{Color.C}[请输入房间端口 默认8888]：{Color.R}".strip())
|
||||
self.room_port = 8888 if not port_input else int(port_input)
|
||||
if 1024 <= self.room_port <= 65535:
|
||||
break
|
||||
print(f"{Color.C}[系统提示] 端口号无效 需在1024-65535之间{Color.R}")
|
||||
|
||||
self.room_locked = False
|
||||
test_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
test_socket.settimeout(2)
|
||||
try:
|
||||
test_socket.connect((self.room_ip, self.room_port))
|
||||
test_socket.send(b"CHECK_LOCK")
|
||||
lock_response = test_socket.recv(1024).decode('utf-8')
|
||||
if lock_response == "LOCKED":
|
||||
self.room_locked = True
|
||||
test_socket.close()
|
||||
except Exception:
|
||||
test_socket.close()
|
||||
|
||||
self.room_password = ""
|
||||
if self.room_locked:
|
||||
self.room_password = input(f"{Color.C}[该房间已锁定 请输入密码]：{Color.R}".strip())
|
||||
|
||||
print(f"{Color.C}{'-'*40}{Color.R}")
|
||||
self.join_room()
|
||||
|
||||
# 加入房间处理
|
||||
def join_room(self):
|
||||
self.client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
try:
|
||||
self.client_socket.connect((self.room_ip, self.room_port))
|
||||
if self.room_locked:
|
||||
self.client_socket.send(f"JOIN_WITH_PWD:{self.username}:{self.room_password}".encode('utf-8'))
|
||||
else:
|
||||
self.client_socket.send(f"JOIN:{self.username}".encode('utf-8'))
|
||||
|
||||
response = self.client_socket.recv(1024).decode('utf-8')
|
||||
if response.startswith("SUCCESS"):
|
||||
self.connected = True
|
||||
print(f"{Color.Y}[系统提示] 成功加入房间{Color.R}")
|
||||
print(f"{Color.C}{'-'*40}{Color.R}")
|
||||
threading.Thread(target=self.receive_messages, daemon=True).start()
|
||||
self.client_chat_loop()
|
||||
elif response.startswith("FAIL:用户名已存在"):
|
||||
print(f"{Color.C}[系统提示] 加入失败 该用户名已被占用 请重新注册{Color.R}")
|
||||
self.client_socket.close()
|
||||
self.register()
|
||||
time.sleep(2)
|
||||
elif response == "FAIL:密码错误":
|
||||
print(f"{Color.C}[系统提示] 加入失败 房间密码错误{Color.R}")
|
||||
self.client_socket.close()
|
||||
time.sleep(2)
|
||||
else:
|
||||
print(f"{Color.C}[系统提示] 加入失败 原因：{response.split(':', 1)[1] if ':' in response else response}{Color.R}")
|
||||
self.client_socket.close()
|
||||
time.sleep(2)
|
||||
except ConnectionRefusedError:
|
||||
print(f"{Color.C}[系统提示] 加入失败 目标房间未开启或地址错误{Color.R}")
|
||||
self.client_socket.close()
|
||||
time.sleep(2)
|
||||
except socket.timeout:
|
||||
print(f"{Color.C}[系统提示] 加入失败 连接超时{Color.R}")
|
||||
self.client_socket.close()
|
||||
time.sleep(2)
|
||||
except Exception as e:
|
||||
print(f"{Color.C}[系统提示] 加入失败 原因：{str(e)}{Color.R}")
|
||||
self.client_socket.close()
|
||||
time.sleep(2)
|
||||
|
||||
# 接收客户端连接
|
||||
def accept_clients(self, server):
|
||||
while self.connected and self.is_host:
|
||||
try:
|
||||
client_socket, addr = server.accept()
|
||||
data = client_socket.recv(1024).decode('utf-8')
|
||||
if data == "CHECK_LOCK":
|
||||
client_socket.send(b"LOCKED" if self.room_locked else b"UNLOCKED")
|
||||
client_socket.close()
|
||||
continue
|
||||
|
||||
username = ""
|
||||
password = ""
|
||||
if data.startswith("JOIN_WITH_PWD:"):
|
||||
parts = data.split(':', 2)
|
||||
if len(parts) != 3:
|
||||
client_socket.send("FAIL:无效请求".encode('utf-8'))
|
||||
client_socket.close()
|
||||
continue
|
||||
username = parts[1]
|
||||
password = parts[2]
|
||||
if password != self.room_password:
|
||||
client_socket.send("FAIL:密码错误".encode('utf-8'))
|
||||
client_socket.close()
|
||||
continue
|
||||
elif data.startswith("JOIN:"):
|
||||
username = data.split(':', 1)[1]
|
||||
else:
|
||||
client_socket.send("FAIL:无效请求".encode('utf-8'))
|
||||
client_socket.close()
|
||||
continue
|
||||
|
||||
if username in self.clients or username == self.username:
|
||||
client_socket.send("FAIL:用户名已存在".encode('utf-8'))
|
||||
client_socket.close()
|
||||
continue
|
||||
|
||||
client_socket.send("SUCCESS:加入成功".encode('utf-8'))
|
||||
self.clients[username] = client_socket
|
||||
print(f"{Color.Y}[系统提示] {username} 加入房间{Color.R}")
|
||||
self.broadcast(f"{Color.Y}[系统提示] {username} 加入房间{Color.R}", exclude=None)
|
||||
threading.Thread(target=self.handle_client, args=(client_socket, username), daemon=True).start()
|
||||
except Exception:
|
||||
break
|
||||
|
||||
# 客户端消息处理
|
||||
def handle_client(self, client_socket, username):
|
||||
while self.connected and username in self.clients:
|
||||
try:
|
||||
data = client_socket.recv(1024).decode('utf-8')
|
||||
if not data:
|
||||
self.remove_client(username)
|
||||
break
|
||||
if data == "/退出房间":
|
||||
self.remove_client(username)
|
||||
break
|
||||
if self.global_mute or username in self.muted_users:
|
||||
client_socket.send(f"{Color.C}[系统提示] 你已被禁言 无法发送消息{Color.R}".encode('utf-8'))
|
||||
continue
|
||||
if data.startswith("/"):
|
||||
self.handle_command(data, username, client_socket)
|
||||
else:
|
||||
self.process_message(username, data)
|
||||
except Exception:
|
||||
self.remove_client(username)
|
||||
break
|
||||
|
||||
# 消息处理
|
||||
def process_message(self, username, content):
|
||||
message = f"{Color.C}[{username}]{Color.R}: {self.user_color}{content}{Color.R}"
|
||||
print(message)
|
||||
|
||||
if "@全体" in content:
|
||||
at_msg = f"{Color.R_}[提醒] {Color.C}[{username}]{Color.R} @全体成员：{self.user_color}{content}{Color.R}"
|
||||
self.broadcast(at_msg, exclude=username)
|
||||
else:
|
||||
at_users = [user for user in self.clients if f"@{user}" in content]
|
||||
if self.username in content:
|
||||
at_users.append(self.username)
|
||||
self.broadcast(message, exclude=username)
|
||||
for user in at_users:
|
||||
at_msg = f"{Color.R_}[提醒] {Color.C}[{user}]{Color.R} 你被@了：{self.user_color}{content}{Color.R}"
|
||||
if user == self.username:
|
||||
print(at_msg)
|
||||
else:
|
||||
try:
|
||||
self.clients[user].send(at_msg.encode('utf-8'))
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
# 消息广播
|
||||
def broadcast(self, message, exclude):
|
||||
if self.is_host:
|
||||
for username, sock in self.clients.items():
|
||||
if username != exclude:
|
||||
try:
|
||||
sock.send(message.encode('utf-8'))
|
||||
except Exception:
|
||||
self.remove_client(username)
|
||||
else:
|
||||
try:
|
||||
self.client_socket.send(message.encode('utf-8'))
|
||||
except Exception:
|
||||
self.connected = False
|
||||
|
||||
# 移除客户端
|
||||
def remove_client(self, username):
|
||||
if username in self.clients:
|
||||
try:
|
||||
self.clients[username].send(f"{Color.C}[系统提示] 你已被移出房间{Color.R}".encode('utf-8'))
|
||||
except Exception:
|
||||
pass
|
||||
self.clients[username].close()
|
||||
del self.clients[username]
|
||||
if username in self.muted_users:
|
||||
self.muted_users.remove(username)
|
||||
print(f"{Color.Y}[系统提示] {username} 退出房间{Color.R}")
|
||||
self.broadcast(f"{Color.Y}[系统提示] {username} 退出房间{Color.R}", exclude=None)
|
||||
|
||||
# 接收消息
|
||||
def receive_messages(self):
|
||||
while self.connected:
|
||||
try:
|
||||
data = self.client_socket.recv(1024).decode('utf-8')
|
||||
if not data:
|
||||
print(f"{Color.C}[系统提示] 与房间断开连接{Color.R}")
|
||||
self.connected = False
|
||||
break
|
||||
print(data)
|
||||
except Exception:
|
||||
print(f"{Color.C}[系统提示] 接收消息失败 已断开连接{Color.R}")
|
||||
self.connected = False
|
||||
break
|
||||
|
||||
# 聊天循环
|
||||
def client_chat_loop(self):
|
||||
while self.connected:
|
||||
message = input().strip()
|
||||
if message == "/清屏":
|
||||
self.clear_screen()
|
||||
elif message == "/退出脚本":
|
||||
self.exit_script()
|
||||
elif message == "/退出房间":
|
||||
self.exit_room()
|
||||
elif message == "/更换聊天颜色":
|
||||
self.change_color()
|
||||
elif message == "/房主菜单" and self.is_host:
|
||||
self.show_host_menu()
|
||||
elif message.startswith("/"):
|
||||
if self.is_host:
|
||||
self.handle_command(message, self.username, None)
|
||||
else:
|
||||
print(f"{Color.C}[系统提示] 无效指令{Color.R}")
|
||||
else:
|
||||
if self.is_host:
|
||||
self.process_message(self.username, message)
|
||||
else:
|
||||
if message:
|
||||
self.client_socket.send(message.encode('utf-8'))
|
||||
self.process_message(self.username, message)
|
||||
|
||||
# 指令处理
|
||||
def handle_command(self, cmd, username, client_socket):
|
||||
if not self.is_host:
|
||||
if client_socket:
|
||||
client_socket.send(f"{Color.C}[系统提示] 无权限使用该指令{Color.R}".encode('utf-8'))
|
||||
else:
|
||||
print(f"{Color.C}[系统提示] 无权限使用该指令{Color.R}")
|
||||
return
|
||||
|
||||
if cmd == "/房主菜单":
|
||||
self.show_host_menu()
|
||||
elif cmd.startswith("/踢人 "):
|
||||
parts = cmd.split(' ', 1)
|
||||
if len(parts) < 2 or not parts[1].strip():
|
||||
print(f"{Color.C}[系统提示] 请输入要踢的用户名 格式：/踢人 用户名{Color.R}")
|
||||
return
|
||||
target = parts[1].strip()
|
||||
if target == self.username:
|
||||
print(f"{Color.C}[系统提示] 无法踢自己{Color.R}")
|
||||
elif target in self.clients:
|
||||
self.remove_client(target)
|
||||
print(f"{Color.C}[系统提示] 已将{target} 踢出房间{Color.R}")
|
||||
else:
|
||||
print(f"{Color.C}[系统提示] 该用户不在房间内{Color.R}")
|
||||
elif cmd.startswith("/禁言 "):
|
||||
parts = cmd.split(' ', 1)
|
||||
if len(parts) < 2 or not parts[1].strip():
|
||||
print(f"{Color.C}[系统提示] 请输入要禁言的对象 格式：/禁言 用户名/全体{Color.R}")
|
||||
return
|
||||
target = parts[1].strip()
|
||||
if target == "全体":
|
||||
self.global_mute = True
|
||||
print(f"{Color.C}[系统提示] 已开启全体禁言{Color.R}")
|
||||
self.broadcast(f"{Color.C}[系统提示] 房主已开启全体禁言{Color.R}", exclude=None)
|
||||
elif target == self.username:
|
||||
print(f"{Color.C}[系统提示] 无法禁言自己{Color.R}")
|
||||
elif target in self.clients:
|
||||
self.muted_users.add(target)
|
||||
print(f"{Color.C}[系统提示] 已禁言{target}{Color.R}")
|
||||
try:
|
||||
self.clients[target].send(f"{Color.C}[系统提示] 你已被房主禁言{Color.R}".encode('utf-8'))
|
||||
except Exception:
|
||||
pass
|
||||
else:
|
||||
print(f"{Color.C}[系统提示] 该用户不在房间内{Color.R}")
|
||||
elif cmd.startswith("/解禁 "):
|
||||
parts = cmd.split(' ', 1)
|
||||
if len(parts) < 2 or not parts[1].strip():
|
||||
print(f"{Color.C}[系统提示] 请输入要解禁的对象 格式：/解禁 用户名/全体{Color.R}")
|
||||
return
|
||||
target = parts[1].strip()
|
||||
if target == "全体":
|
||||
self.global_mute = False
|
||||
self.muted_users.clear()
|
||||
print(f"{Color.C}[系统提示] 已解除全体禁言{Color.R}")
|
||||
self.broadcast(f"{Color.C}[系统提示] 房主已解除全体禁言{Color.R}", exclude=None)
|
||||
elif target in self.muted_users:
|
||||
self.muted_users.remove(target)
|
||||
print(f"{Color.C}[系统提示] 已解禁{target}{Color.R}")
|
||||
try:
|
||||
self.clients[target].send(f"{Color.C}[系统提示] 你已被房主解禁{Color.R}".encode('utf-8'))
|
||||
except Exception:
|
||||
pass
|
||||
else:
|
||||
print(f"{Color.C}[系统提示] 该用户未被禁言或不在房间内{Color.R}")
|
||||
else:
|
||||
if client_socket:
|
||||
client_socket.send(f"{Color.C}[系统提示] 无效指令{Color.R}".encode('utf-8'))
|
||||
else:
|
||||
print(f"{Color.C}[系统提示] 无效指令{Color.R}")
|
||||
|
||||
# 房主菜单
|
||||
def show_host_menu(self):
|
||||
self.clear_screen()
|
||||
print(f"{Color.C}{'='*50}{Color.R}")
|
||||
print(f"{Color.C}{'[ 房主专属菜单 ]':^50}{Color.R}")
|
||||
print(f"{Color.C}{'='*50}{Color.R}")
|
||||
print(f"{Color.C}{'1. 踢人功能':<20}{'格式：/踢人 用户名':<30}{Color.R}")
|
||||
print(f"{Color.C}{'2. 禁言功能':<20}{'格式：/禁言 用户名/全体':<30}{Color.R}")
|
||||
print(f"{Color.C}{'3. 解禁功能':<20}{'格式：/解禁 用户名/全体':<30}{Color.R}")
|
||||
print(f"{Color.C}{'4. 房间锁管理':<20}{'开启/关闭房间锁':<30}{Color.R}")
|
||||
print(f"{Color.C}{'5. 查看在线用户':<20}{'显示当前房间所有成员':<30}{Color.R}")
|
||||
print(f"{Color.C}{'6. 解散房间':<20}{'强制关闭房间 全员退出':<30}{Color.R}")
|
||||
print(f"{Color.C}{'7. 关闭菜单':<20}{'返回聊天界面':<30}{Color.R}")
|
||||
print(f"{Color.C}{'='*50}{Color.R}")
|
||||
choice = input(f"{Color.C}[请选择功能(1-7)]：{Color.R}".strip())
|
||||
if choice == "1":
|
||||
target = input(f"{Color.C}[请输入要踢的用户名]：{Color.R}".strip())
|
||||
if not target:
|
||||
print(f"{Color.C}[系统提示] 用户名不能为空{Color.R}")
|
||||
elif target == self.username:
|
||||
print(f"{Color.C}[系统提示] 无法踢自己{Color.R}")
|
||||
elif target in self.clients:
|
||||
self.remove_client(target)
|
||||
print(f"{Color.C}[系统提示] 已将{target} 踢出房间{Color.R}")
|
||||
else:
|
||||
print(f"{Color.C}[系统提示] 该用户不在房间内{Color.R}")
|
||||
time.sleep(1)
|
||||
elif choice == "2":
|
||||
target = input(f"{Color.C}[请输入要禁言的用户名 输入全体开启全体禁言]：{Color.R}".strip())
|
||||
if not target:
|
||||
print(f"{Color.C}[系统提示] 请输入有效对象{Color.R}")
|
||||
elif target == "全体":
|
||||
self.global_mute = True
|
||||
print(f"{Color.C}[系统提示] 已开启全体禁言{Color.R}")
|
||||
self.broadcast(f"{Color.C}[系统提示] 房主已开启全体禁言{Color.R}", exclude=None)
|
||||
elif target == self.username:
|
||||
print(f"{Color.C}[系统提示] 无法禁言自己{Color.R}")
|
||||
elif target in self.clients:
|
||||
self.muted_users.add(target)
|
||||
print(f"{Color.C}[系统提示] 已禁言{target}{Color.R}")
|
||||
try:
|
||||
self.clients[target].send(f"{Color.C}[系统提示] 你已被房主禁言{Color.R}".encode('utf-8'))
|
||||
except Exception:
|
||||
pass
|
||||
else:
|
||||
print(f"{Color.C}[系统提示] 该用户不在房间内{Color.R}")
|
||||
time.sleep(1)
|
||||
elif choice == "3":
|
||||
target = input(f"{Color.C}[请输入要解禁的用户名 输入全体解除全体禁言]：{Color.R}".strip())
|
||||
if not target:
|
||||
print(f"{Color.C}[系统提示] 请输入有效对象{Color.R}")
|
||||
elif target == "全体":
|
||||
self.global_mute = False
|
||||
self.muted_users.clear()
|
||||
print(f"{Color.C}[系统提示] 已解除全体禁言{Color.R}")
|
||||
self.broadcast(f"{Color.C}[系统提示] 房主已解除全体禁言{Color.R}", exclude=None)
|
||||
elif target in self.muted_users:
|
||||
self.muted_users.remove(target)
|
||||
print(f"{Color.C}[系统提示] 已解禁{target}{Color.R}")
|
||||
try:
|
||||
self.clients[target].send(f"{Color.C}[系统提示] 你已被房主解禁{Color.R}".encode('utf-8'))
|
||||
except Exception:
|
||||
pass
|
||||
else:
|
||||
print(f"{Color.C}[系统提示] 该用户未被禁言或不在房间内{Color.R}")
|
||||
time.sleep(1)
|
||||
elif choice == "4":
|
||||
print(f"{Color.C}{'-'*50}{Color.R}")
|
||||
print(f"{Color.C}[当前房间锁状态] {'已开启' if self.room_locked else '已关闭'}{Color.R}")
|
||||
while True:
|
||||
lock_choice = input(f"{Color.C}[请选择 1=开启 0=关闭]：{Color.R}".strip())
|
||||
if lock_choice in ["0", "1"]:
|
||||
new_locked = (lock_choice == "1")
|
||||
break
|
||||
print(f"{Color.C}[系统提示] 无效选择 请输入0或1{Color.R}")
|
||||
if new_locked == self.room_locked:
|
||||
print(f"{Color.C}[系统提示] 房间锁状态未改变{Color.R}")
|
||||
else:
|
||||
if new_locked:
|
||||
while True:
|
||||
pwd = input(f"{Color.C}[请设置房间密码 4-8位字符]：{Color.R}".strip())
|
||||
if 4 <= len(pwd) <= 8 and not ' ' in pwd:
|
||||
self.room_password = pwd
|
||||
break
|
||||
print(f"{Color.C}[系统提示] 密码无效 需4-8位字符且不含空格{Color.R}")
|
||||
self.room_locked = new_locked
|
||||
print(f"{Color.C}[系统提示] 房间锁状态已更新为{'开启' if new_locked else '关闭'}{Color.R}")
|
||||
self.broadcast(f"{Color.C}[系统提示] 房主已{'开启' if new_locked else '关闭'}房间锁{Color.R}", exclude=None)
|
||||
print(f"{Color.C}{'-'*50}{Color.R}")
|
||||
input(f"{Color.C}[系统提示] 按回车键返回{Color.R}")
|
||||
elif choice == "5":
|
||||
print(f"{Color.C}{'-'*50}{Color.R}")
|
||||
print(f"{Color.C}[在线用户统计] 共{len(self.clients)+1}人{Color.R}")
|
||||
print(f"{Color.C}[房主] {self.username}{Color.R}")
|
||||
if self.clients:
|
||||
muted_list = [user for user in self.clients if user in self.muted_users]
|
||||
normal_list = [user for user in self.clients if user not in self.muted_users]
|
||||
print(f"{Color.C}[成员] {' | '.join(normal_list)}{Color.R}")
|
||||
if muted_list:
|
||||
print(f"{Color.C}[禁言成员] {' | '.join(muted_list)}{Color.R}")
|
||||
else:
|
||||
print(f"{Color.C}[成员] 暂无其他成员{Color.R}")
|
||||
print(f"{Color.C}[全体禁言状态] {'开启' if self.global_mute else '关闭'}{Color.R}")
|
||||
print(f"{Color.C}{'-'*50}{Color.R}")
|
||||
input(f"{Color.C}[系统提示] 按回车键返回{Color.R}")
|
||||
elif choice == "6":
|
||||
print(f"{Color.C}{'-'*50}{Color.R}")
|
||||
confirm = input(f"{Color.C}[确认解散房间 y/n]：{Color.R}".strip())
|
||||
if confirm.lower() == "y":
|
||||
self.dismiss_room()
|
||||
elif choice != "7":
|
||||
print(f"{Color.C}[系统提示] 无效选择{Color.R}")
|
||||
time.sleep(1)
|
||||
|
||||
# 更换聊天颜色
|
||||
def change_color(self):
|
||||
print(f"{Color.C}{'='*40}{Color.R}")
|
||||
print(f"{Color.C}{'[ 更换聊天颜色 ]':^40}{Color.R}")
|
||||
print(f"{Color.C}{'='*40}{Color.R}")
|
||||
colors = [Color.Y, '\033[91m', '\033[92m', '\033[95m', '\033[94m']
|
||||
color_names = ['黄色', '红色', '绿色', '紫色', '蓝色']
|
||||
for i, name in enumerate(color_names, 1):
|
||||
print(f"{Color.C}{i}. {colors[i-1]}{name}{Color.R}")
|
||||
print(f"{Color.C}{'-'*40}{Color.R}")
|
||||
choice = input(f"{Color.C}[请选择颜色(1-5)]：{Color.R}".strip())
|
||||
if choice.isdigit() and 1 <= int(choice) <= 5:
|
||||
self.user_color = colors[int(choice)-1]
|
||||
print(f"{Color.C}[系统提示] 聊天颜色已更换为{color_names[int(choice)-1]}{Color.R}")
|
||||
else:
|
||||
print(f"{Color.C}[系统提示] 无效选择 颜色未更改{Color.R}")
|
||||
time.sleep(1)
|
||||
|
||||
# 解散房间
|
||||
def dismiss_room(self):
|
||||
print(f"{Color.C}[系统提示] 正在解散房间{Color.R}")
|
||||
self.connected = False
|
||||
for sock in self.clients.values():
|
||||
try:
|
||||
sock.send(f"{Color.C}[系统提示] 房主已解散房间{Color.R}".encode('utf-8'))
|
||||
sock.close()
|
||||
except Exception:
|
||||
pass
|
||||
self.clients.clear()
|
||||
self.muted_users.clear()
|
||||
self.global_mute = False
|
||||
print(f"{Color.Y}[系统提示] 房间已解散{Color.R}")
|
||||
time.sleep(2)
|
||||
self.is_host = False
|
||||
self.room_ip = ""
|
||||
self.room_port = 0
|
||||
self.room_locked = False
|
||||
self.room_password = ""
|
||||
self.show_main_menu()
|
||||
|
||||
# 退出房间
|
||||
def exit_room(self):
|
||||
print(f"{Color.C}[系统提示] 正在退出房间{Color.R}")
|
||||
self.connected = False
|
||||
if self.is_host:
|
||||
for sock in self.clients.values():
|
||||
try:
|
||||
sock.send(f"{Color.C}[系统提示] 房主已退出房间{Color.R}".encode('utf-8'))
|
||||
sock.close()
|
||||
except Exception:
|
||||
pass
|
||||
self.clients.clear()
|
||||
self.muted_users.clear()
|
||||
self.global_mute = False
|
||||
else:
|
||||
try:
|
||||
self.client_socket.send("/退出房间".encode('utf-8'))
|
||||
self.client_socket.close()
|
||||
except Exception:
|
||||
pass
|
||||
print(f"{Color.Y}[系统提示] 已退出房间{Color.R}")
|
||||
self.is_host = False
|
||||
self.room_ip = ""
|
||||
self.room_port = 0
|
||||
self.room_locked = False
|
||||
self.room_password = ""
|
||||
time.sleep(1)
|
||||
self.show_main_menu()
|
||||
|
||||
# 退出脚本
|
||||
def exit_script(self):
|
||||
print(f"{Color.C}[系统提示] 正在退出脚本{Color.R}")
|
||||
self.running = False
|
||||
self.connected = False
|
||||
if self.client_socket:
|
||||
try:
|
||||
self.client_socket.send("/退出脚本".encode('utf-8'))
|
||||
self.client_socket.close()
|
||||
except Exception:
|
||||
pass
|
||||
if self.is_host:
|
||||
for sock in self.clients.values():
|
||||
try:
|
||||
sock.send(f"{Color.C}[系统提示] 房主已退出 脚本关闭{Color.R}".encode('utf-8'))
|
||||
sock.close()
|
||||
except Exception:
|
||||
pass
|
||||
self.clients.clear()
|
||||
self.muted_users.clear()
|
||||
self.global_mute = False
|
||||
sys.exit(0)
|
||||
|
||||
if __name__ == "__main__":
|
||||
chat = ChatRoom()
|
||||
chat.register()
|
||||
chat.show_main_menu()
|
||||
Reference in New Issue
Block a user