Python语言实现两台计算机用TCP协议跨局域网通信 - ToB企服应用市场:ToB评测及商务社交产业平台

HOST = '你的服务器IP地址'
PORT = 21567
ADDR = (HOST, PORT)
BUFSIZ = 65536
tcpCli = socket(AF_INET, SOCK_STREAM)
tcpCli.connect(ADDR)
root = tk.Tk()
chat_window = ChatWindow(root)
threading.Thread(target=recv, args=(tcpCli, BUFSIZ, chat_window)).start() # 创建接收消息的线程并启动
root.mainloop()
tcpCli.close()

复制代码

def send_messages(self):
while True:
message = self.message_queue.get() # 从队列中获取消息
if message.startswith("#sendFile#"):
_, __, filepath = message.split("#")
path_parts = filepath.split("/")
# 获取最后一个部分
last_part = path_parts[-1]
toFile = "/home/liyishui/" #发送到服务器上的存储路径
toFile+=last_part
with open(filepath, "rb") as f:
data = f.read()
tcpCli.sendall(("#fileData#%s#" % toFile).encode() + data)
toFile= "#sendFile#"+last_part
tcpCli.send(toFile.encode())
else:
tcpCli.send(message.encode())

复制代码

def recv(sock, BUFSIZ, chat_window):
try:
while True:
data = sock.recv(BUFSIZ)
print(data)
print(data.decode())
if not data:
break
if data.startswith(b"#fileData#"):
_, filename, filepath, filedata = data.decode().split("#", 3)
toFile = "C:\\Users\\33245\\Desktop\\ttssmm\\animal\\backend\\user2\" + filepath.strip() #本地文件存储路径
with open(toFile, "wb") as f:
f.write(filedata.encode())
time_tag = '[%s] ' % ctime()
tag_config = ('sent_time_tag', 'green')
chat_window.info_text.config(state='normal')
chat_window.info_text.insert(tk.END, time_tag, tag_config)
chat_window.info_text.insert(tk.END, '[Download] File download complete: {}\n'.format(filepath))
chat_window.info_text.config(state='disabled')
chat_window.info_text.see(tk.END)
continue
elif data.startswith(b"#sendFile#"):
_, __, filename = data.split(b"#", 2)
chat_window.info_text.config(state='normal')
filename=filename.decode()
chat_window.info_text.insert(tk.END, '[Newfile]: %s\n' % filename)
chat_window.info_text.config(state='disabled')
chat_window.info_text.see(tk.END)
continue
elif data.decode() == '[CHAT]BEGIN':
chat_window.add_log('[对方] ', data.decode())
print(data.decode()) # 打印到控制台
elif data.decode() == '[CHAT]END':
sock.close()
break
else:
chat_window.add_log('对方 ', data.decode())
print(data.decode()) # 打印到控制台
except OSError:
pass

复制代码

HOST = ''
PORT = 21567
ADDR = (HOST, PORT)
tcp = socket(AF_INET, SOCK_STREAM)
tcp.bind(ADDR)
tcp.listen(3)
Users = []
Addrs = []

复制代码

def accept_connections():
user1_files_queue = queue.Queue() # 新建用户1的文件队列
user2_files_queue = queue.Queue() # 新建用户2的文件队列
while len(Users) != 2:
tcpCli, addr = tcp.accept()
Users.append(tcpCli)
Addrs.append(addr)
user_box.insert(tk.END, "User" + str(len(Users)) + ": " + str(addr) + "\n")

复制代码

trans1 = threading.Thread(target=trans, args=(Users[0], Users[1], 1024, user1_files_queue, user2_files_queue))
trans1.start()
trans2 = threading.Thread(target=trans, args=(Users[1], Users[0], 1024, user2_files_queue, user1_files_queue))
trans2.start()

复制代码

def trans(sock1, sock2, BUFSIZ, user1_files_queue, user2_files_queue):
while True:
try:
data = sock1.recv(BUFSIZ)
except OSError:
break
if not data:
sock1.close()
else:
# 判断是否是文件传输标识
if data.startswith(b"#fileData#"):
_, filename, filepath, filedata = data.split(b"#", 3)
save_file(filepath.decode(), filedata)
message_display.insert(tk.END, f"File received: {filepath.decode()}\n")
user2_files_queue.put(filepath.decode())
elif data == b"#receiveAll#":
send_all_files(sock1, user1_files_queue)
else:
try:
sock2.send(data)
except OSError:
sock1.close()
break

复制代码

import os.path
from socket import *
import threading
from time import ctime
from queue import Queue
import tkinter as tk
from tkinter.filedialog import askopenfilename
class ChatWindow:
def __init__(self, master):
self.master = master
master.title('501密聊')
master.geometry('620x400') # 设置窗口大小
self.paned_window = tk.PanedWindow(master, orient=tk.HORIZONTAL, sashwidth=4, sashrelief=tk.RAISED)
self.paned_window.pack(expand=True, fill='both')
# 左侧面板
self.left_panel = tk.Frame(self.paned_window, width=200, height=400)
self.left_panel.grid_propagate(False) # 禁止自动调整大小
self.paned_window.add(self.left_panel)
# 在左侧面板中添加显示信息的Text组件
self.info_text = tk.Text(self.left_panel, state='disabled', width=25, height=20, bg='light yellow')
self.info_text.pack(expand=True, fill='both')
self.info_text.tag_configure('tag_me', foreground='green')
self.info_text.tag_configure('tag_user', foreground='blue')
#在左侧面板中添加"reiceive"按钮
self.receive_button = tk.Button(self.left_panel, text="Receive", command=self.receive_all_files)
self.receive_button.pack(side=tk.BOTTOM, pady=5)
# 右侧面板
self.right_panel = tk.Frame(self.paned_window)
self.right_panel.grid_propagate(False) # 禁止自动调整大小
self.paned_window.add(self.right_panel)
# 在右侧面板中添加聊天窗口的组件
self.log_text = tk.Text(self.right_panel, state='disabled', width=50, height=20, bg='light blue')
self.log_text.grid(row=0, column=0, columnspan=2)
self.send_entry = tk.Text(self.right_panel, width=40, height=5)
self.send_entry.grid(row=1, column=0)
self.send_entry.bind('<KeyPress-Return>', self.send_message_on_enter) # 绑定回车键事件
self.send_button = tk.Button(self.right_panel, text='Send Text [or Enter]', command=self.send_message)
self.send_button.grid(row=1, column=1,columnspan=2)
self.scrollbar = tk.Scrollbar(self.right_panel, orient='vertical', command=self.log_text.yview)
self.scrollbar.grid(row=0, column=2, sticky='ns')
self.log_text.config(yscrollcommand=self.scrollbar.set)
self.message_queue = Queue() # 创建队列用于存储待发送的消息
self.send_thread = threading.Thread(target=self.send_messages) # 创建发送消息的线程
self.send_thread.start() # 启动发送消息的线程
# 设置标签样式
self.log_text.tag_configure('sent_time_tag', foreground='green')
self.log_text.tag_configure('recv_time_tag', foreground='blue')
self.log_text.tag_configure('sent_message', foreground='green')
self.log_text.tag_configure('recv_message', foreground='blue')
self.file_entry = tk.Entry(self.right_panel, width=40)
self.file_entry.grid(row=2, column=0)
self.select_file_button = tk.Button(self.right_panel, text='Select File', command=self.select_file)
self.select_file_button.grid(row=2, column=1)
self.send_file_button = tk.Button(self.right_panel, text='Send File', command=self.send_file)
self.send_file_button.grid(row=2, column=2)
def receive_all_files(self):
self.message_queue.put("#receiveAll#")
def save_file(filename, filedata):
with open(filename, "wb") as f:
f.write(filedata)
# 在客户端的 process_received_messages 方法中
def select_file(self):
filepath = askopenfilename()
if filepath:
self.file_entry.delete(0, tk.END)
self.file_entry.insert(tk.END, filepath)
def send_file(self):
filepath = self.file_entry.get()
if filepath:
self.message_queue.put("#sendFile#" + filepath)
self.info_text.config(state='normal')
self.info_text.insert(tk.END, 'File sent complete: %s\n' % os.path.basename(filepath))
self.info_text.config(state='disabled')
self.info_text.see(tk.END)
self.file_entry.delete(0, tk.END)
else:
tk.messagebox.showinfo('Warning', 'Please enter a file path.')
def send_message(self):
message = self.send_entry.get('1.0', tk.END)
self.send_entry.delete('1.0', tk.END)
if message.strip():
self.message_queue.put(message.strip()) # 将消息放入队列中
self.add_log('[Me]', message.strip(), is_sent=True) # 将消息显示在聊天消息列表框中
print('[Me]', message.strip()) # 打印到控制台
def send_message_on_enter(self, event): # 回车键事件处理函数
self.send_message()
def add_log(self, user, message, is_sent=False):
self.log_text.config(state='normal')
if is_sent:
time_tag = '[%s] ' % ctime()
tag_config = ('sent_time_tag', 'green')
message_tag = 'sent_message'
else:
time_tag = '[%s] ' % ctime()
tag_config = ('recv_time_tag', 'blue')
message_tag = 'recv_message'
self.log_text.insert(tk.END, time_tag, tag_config)
self.log_text.insert(tk.END, '\n') # 插入换行符
if is_sent:
self.log_text.insert(tk.END, user, message_tag)
else:
self.log_text.insert(tk.END, user, message_tag)
self.log_text.insert(tk.END, '%s\n' % message)
self.log_text.config(state='disabled')
self.log_text.see(tk.END)
print(message) # 打印到控制台
def send_messages(self):
while True:
message = self.message_queue.get() # 从队列中获取消息
if message.startswith("#sendFile#"):
_, __, filepath = message.split("#")
path_parts = filepath.split("/")
# 获取最后一个部分
last_part = path_parts[-1]
toFile = "你的服务器上的存储路径" #
toFile+=last_part
with open(filepath, "rb") as f:
data = f.read()
tcpCli.sendall(("#fileData#%s#" % toFile).encode() + data)
toFile= "#sendFile#"+last_part
tcpCli.send(toFile.encode())
else:
tcpCli.send(message.encode())
def recv(sock, BUFSIZ, chat_window):
try:
while True:
data = sock.recv(BUFSIZ)
print(data)
print(data.decode())
if not data:
break
if data.startswith(b"#fileData#"):
_, filename, filepath, filedata = data.decode().split("#", 3)
toFile = "你的本地文件存储路径" + filepath.strip()
with open(toFile, "wb") as f:
f.write(filedata.encode())
time_tag = '[%s] ' % ctime()
tag_config = ('sent_time_tag', 'green')
chat_window.info_text.config(state='normal')
chat_window.info_text.insert(tk.END, time_tag, tag_config)
chat_window.info_text.insert(tk.END, '[Download] File download complete: {}\n'.format(filepath))
chat_window.info_text.config(state='disabled')
chat_window.info_text.see(tk.END)
continue
elif data.startswith(b"#sendFile#"):
_, __, filename = data.split(b"#", 2)
chat_window.info_text.config(state='normal')
filename=filename.decode()
chat_window.info_text.insert(tk.END, '[Newfile]: %s\n' % filename)
chat_window.info_text.config(state='disabled')
chat_window.info_text.see(tk.END)
continue
elif data.decode() == '[CHAT]BEGIN':
chat_window.add_log('[对方] ', data.decode())
print(data.decode()) # 打印到控制台
elif data.decode() == '[CHAT]END':
sock.close()
break
else:
chat_window.add_log('对方 ', data.decode())
print(data.decode()) # 打印到控制台
except OSError:
pass
if __name__ == '__main__':
HOST = '你的ip地址'
PORT = 21567
ADDR = (HOST, PORT)
BUFSIZ = 65536
tcpCli = socket(AF_INET, SOCK_STREAM)
tcpCli.connect(ADDR)
root = tk.Tk()
chat_window = ChatWindow(root)
threading.Thread(target=recv, args=(tcpCli, BUFSIZ, chat_window)).start() # 创建接收消息的线程并启动
root.mainloop()
tcpCli.close()

复制代码

from socket import *import tkinter as tkimport threadingimport osimport queue # 添加队列模块def trans(sock1, sock2, BUFSIZ, user1_files_queue, user2_files_queue):
while True:
try:
data = sock1.recv(BUFSIZ)
except OSError:
break
if not data:
sock1.close()
else:
# 判断是否是文件传输标识
if data.startswith(b"#fileData#"):
_, filename, filepath, filedata = data.split(b"#", 3)
save_file(filepath.decode(), filedata)
message_display.insert(tk.END, f"File received: {filepath.decode()}\n")
user2_files_queue.put(filepath.decode())
elif data == b"#receiveAll#":
send_all_files(sock1, user1_files_queue)
else:
try:
sock2.send(data)
except OSError:
sock1.close()
breakdef save_file(filename, filedata): with open(filename, "wb") as f: f.write(filedata)def send_all_files(sock, files_queue): while not files_queue.empty(): filepath = files_queue.get() filename = os.path.basename(filepath) print(filename) with open(filepath, "rb") as f: data = f.read() sock.sendall(("#fileData#%s#" % filename).encode() + data)def accept_connections():
user1_files_queue = queue.Queue() # 新建用户1的文件队列
user2_files_queue = queue.Queue() # 新建用户2的文件队列
while len(Users) != 2:
tcpCli, addr = tcp.accept()
Users.append(tcpCli)
Addrs.append(addr)
user_box.insert(tk.END, "User" + str(len(Users)) + ": " + str(addr) + "\n") trans1 = threading.Thread(target=trans, args=(Users[0], Users[1], 1024, user1_files_queue, user2_files_queue))
trans1.start()
trans2 = threading.Thread(target=trans, args=(Users[1], Users[0], 1024, user2_files_queue, user1_files_queue))
trans2.start()if __name__ == '__main__': HOST = ''
PORT = 21567
ADDR = (HOST, PORT)
tcp = socket(AF_INET, SOCK_STREAM)
tcp.bind(ADDR)
tcp.listen(3)
Users = []
Addrs = [] root = tk.Tk() root.title("Chat Server") root.geometry("400x300") # 创建用户框 user_box = tk.Listbox(root, width=20) user_box.pack(side=tk.LEFT, fill=tk.BOTH, expand=True) # 创建用于显示消息的文本框 message_display = tk.Text(root) message_display.pack(side=tk.LEFT, fill=tk.BOTH, expand=True) # 创建一个线程用于接受连接 accept_thread = threading.Thread(target=accept_connections) accept_thread.start() root.mainloop()

复制代码