client端-1130929-udp-receiver01.py
import socket
import threading
import time
import uuid
import os
import subprocess
def get_mac_address():
mac = ':'.join(['{:02x}'.format((uuid.getnode() >> elements) & 0xff) for elements in range(0, 2*6, 2)][::-1])
return mac
def get_computer_name():
return socket.gethostname()
def send_report():
client_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
client_socket.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
while True:
try:
report_msg = f"{socket.gethostbyname(socket.gethostname())},{get_mac_address()},{get_computer_name()}"
client_socket.sendto(report_msg.encode(), ('<broadcast>', 37020))
except Exception as e:
print(f"發送報到時發生錯誤: {e}")
time.sleep(5)
def receive_file_list():
client_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
client_socket.bind(("", 8500))
while True:
try:
data, addr = client_socket.recvfrom(1024)
file_list = data.decode().split(",")
print(f"接收到的檔案清單: {file_list}")
# 建立接收檔案的資料夾
os.makedirs("received_files", exist_ok=True)
# 使用 udp-receiver.exe 接收檔案
for file_name in file_list:
save_path = os.path.join("received_files", os.path.basename(file_name))
subprocess.run(['udp-receiver.exe', '--file', save_path, '--port', '8500'])
except Exception as e:
print(f"接收檔案清單時發生錯誤: {e}")
# 啟動客戶端
if __name__ == "__main__":
threading.Thread(target=send_report, daemon=True).start()
threading.Thread(target=receive_file_list, daemon=True).start()
while True:
time.sleep(1)
伺服端-
-有傳輸狀態,是抓本機的
-可以選多clinet,同時傳輸
-缺,還沒傳輸完,使用者可以再點選傳輸檔案!
-僅能一次傳一個檔案
-
import socket
import threading
import tkinter as tk
from tkinter import filedialog, ttk
import time
import os
import subprocess
class Server:
def __init__(self, root):
self.root = root
self.root.title("Server")
self.root.geometry("800x600")
# 新增標題 "UDP派送檔案"
self.title_label = tk.Label(self.root, text="UDP派送檔案", font=("Arial", 18), fg="deepskyblue")
self.title_label.pack(pady=10)
# 建立表格
self.tree = ttk.Treeview(root, columns=("IP", "MAC", "Name", "Status"), show='headings')
# 設定表格標題與資料置中
self.tree.heading("IP", text="IP 位置", anchor="center")
self.tree.heading("MAC", text="MAC 位置", anchor="center")
self.tree.heading("Name", text="電腦名稱", anchor="center")
self.tree.heading("Status", text="狀態", anchor="center")
# 設定每個欄位的資料置中
self.tree.column("IP", anchor="center")
self.tree.column("MAC", anchor="center")
self.tree.column("Name", anchor="center")
self.tree.column("Status", anchor="center")
self.tree.pack(fill=tk.BOTH, expand=True)
# 儲存客戶端資料的字典
self.client_data_dict = {}
# 建立廣播按鈕
self.broadcast_button = tk.Button(self.root, text="手動廣播伺服器", command=self.start_broadcast_manually)
self.broadcast_button.pack(pady=10)
# 建立傳輸檔案按鈕
self.transfer_button = tk.Button(self.root, text="傳送檔案", command=self.select_client_and_file)
self.transfer_button.pack(pady=10)
# 新增檔案傳輸狀態顯示欄位,初始化為 "等待中"
self.status_label = tk.Label(self.root, text="目前傳輸狀態:等待中", font=("Arial", 12), fg="black")
self.status_label.pack(pady=10)
# 新增備註區域,左右兩格
self.footer_frame = tk.Frame(self.root)
self.footer_frame.pack(pady=20, fill=tk.X)
# 左邊格 - 跑馬燈效果,文字顏色改為gold
self.marquee_text = "感謝ChatGPT 感謝UDPcast "
self.marquee_label = tk.Label(self.footer_frame, text=self.marquee_text, font=("Arial", 12), fg="gold")
self.marquee_label.pack(side=tk.LEFT, padx=10)
# 右邊格 - 固定文字,文字顏色改為gold
self.fixed_label = tk.Label(self.footer_frame, text="製作者:WCS 日期:2024.10.01", font=("Arial", 12), fg="gold")
self.fixed_label.pack(side=tk.RIGHT, padx=10)
# 啟動跑馬燈
self.move_text()
# 取得伺服端的本地 IP
self.server_ip = self.get_local_ip()
# 啟動伺服器
self.server_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
self.server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
self.server_socket.bind(("", 37020))
# 啟動接收客戶端報到的線程
threading.Thread(target=self.receive_clients, daemon=True).start()
# 啟動時廣播30秒
threading.Thread(target=self.broadcast_for_30_seconds, daemon=True).start()
def get_local_ip(self):
hostname = socket.gethostname()
local_ip = socket.gethostbyname(hostname)
print(f"伺服端 IP: {local_ip}")
return local_ip
def move_text(self):
# 跑馬燈效果,將字串每次往左移動一個字符
self.marquee_text = self.marquee_text[1:] + self.marquee_text[0]
self.marquee_label.config(text=self.marquee_text)
self.root.after(200, self.move_text) # 每200毫秒移動一次
def receive_clients(self):
while True:
try:
data, addr = self.server_socket.recvfrom(1024)
client_data = data.decode().split(',')
# 確認收到的資料是否完整
if len(client_data) == 3:
client_ip, client_mac, client_name = client_data
# 排除伺服端自己的 IP,不顯示在表格中
if client_ip != self.server_ip:
self.update_or_insert_client(client_ip, client_mac, client_name, "Online")
else:
print(f"接收到不完整的資料:{client_data}")
except Exception as e:
print(f"接收資料時發生錯誤: {e}")
def update_or_insert_client(self, ip, mac, name, status):
if ip in self.client_data_dict or mac in self.client_data_dict:
# 更新該行的資料
item_id = self.client_data_dict.get(ip) or self.client_data_dict.get(mac)
self.tree.item(item_id, values=(ip, mac, name, status))
else:
# 插入新行並保存該行的 ID
item_id = self.tree.insert("", "end", values=(ip, mac, name, status))
self.client_data_dict[ip] = item_id
self.client_data_dict[mac] = item_id
def broadcast_for_30_seconds(self):
start_time = time.time()
while time.time() - start_time < 30:
try:
self.server_socket.sendto(b"Server here", ('<broadcast>', 37020))
time.sleep(5)
except Exception as e:
print(f"廣播時發生錯誤: {e}")
print("廣播已停止")
def start_broadcast_manually(self):
threading.Thread(target=self.broadcast_for_30_seconds, daemon=True).start()
def select_client_and_file(self):
selected_items = self.tree.selection() # 可以選取多個客戶端
if selected_items:
client_ips = [self.tree.item(item, 'values')[0] for item in selected_items] # 獲取所有選中的客戶端 IP
# 選擇檔案或資料夾
file_paths = filedialog.askopenfilenames(title="選擇檔案")
if file_paths:
# 更新狀態為正在準備傳送
self.status_label.config(text="目前傳輸狀態:正在準備傳送檔案...")
# 傳送檔案清單給每一個選中的客戶端
for client_ip in client_ips:
self.prepare_file_transfer_list(client_ip, file_paths)
def prepare_file_transfer_list(self, client_ip, file_paths):
transfer_list = "\n".join(file_paths)
print(f"準備傳輸檔案到 {client_ip}:\n{transfer_list}")
# 傳送檔案清單給 client
threading.Thread(target=self.send_file_list, args=(client_ip, file_paths), daemon=True).start()
def send_file_list(self, client_ip, file_paths):
# 建立檔案傳輸指令
file_list_str = ",".join(file_paths)
try:
self.server_socket.sendto(file_list_str.encode(), (client_ip, 8500))
print(f"檔案傳輸清單已傳送至 {client_ip}")
# 等待2秒後再使用 udp-sender.exe 傳送檔案
time.sleep(2)
# 根據選取的客戶端數量,計算 min-receivers
selected_clients = self.tree.selection()
client_count = len(selected_clients)
# 使用 udp-sender.exe 傳送檔案
for file_path in file_paths:
if os.path.exists(file_path):
file_size = os.path.getsize(file_path) # 取得檔案大小
self.status_label.config(text=f"目前傳輸檔案:{os.path.basename(file_path)},大小:{file_size // (1024 * 1024)} MB")
# 開始執行 udp-sender.exe
command = f"udp-sender.exe --file \"{file_path}\" --port 8500 --min-receivers {client_count} --nokbd --autostart 3 "
subprocess.run(command, shell=True)
# 完成傳輸,更新狀態
self.status_label.config(text="目前傳輸狀態:傳輸完成!")
except Exception as e:
print(f"傳送檔案清單時發生錯誤: {e}")
self.status_label.config(text="傳輸失敗!")
if __name__ == "__main__":
root = tk.Tk()
server_app = Server(root)
root.mainloop()
-待測
要讓你的程式能夠在選擇多個檔案後,逐一傳輸每個檔案,可以透過使用遞迴或者在傳輸完成後自動觸發下一個檔案的傳送。以下是修改後的範例程式碼,這樣可以確保每次傳輸完成後,才會開始傳送下一個檔案。
import os
import subprocess
import threading
import time
from tkinter import filedialog
def choose_files(self):
file_paths = filedialog.askopenfilenames(title="選擇檔案")
if file_paths:
self.status_label.config(text="目前傳輸狀態:正在準備傳送檔案...")
for client_ip in client_ips:
self.prepare_file_transfer_list(client_ip, file_paths)
def prepare_file_transfer_list(self, client_ip, file_paths):
transfer_list = "\n".join(file_paths)
print(f"準備傳輸檔案到 {client_ip}:\n{transfer_list}")
threading.Thread(target=self.send_file_list, args=(client_ip, file_paths), daemon=True).start()
def send_file_list(self, client_ip, file_paths):
file_list_str = ",".join(file_paths)
try:
self.server_socket.sendto(file_list_str.encode(), (client_ip, 8500))
print(f"檔案傳輸清單已傳送至 {client_ip}")
time.sleep(2)
selected_clients = self.tree.selection()
client_count = len(selected_clients)
# 開始逐一傳送檔案
for file_path in file_paths:
if os.path.exists(file_path):
file_size = os.path.getsize(file_path)
self.status_label.config(text=f"目前傳輸檔案:{os.path.basename(file_path)},大小:{file_size // (1024 * 1024)} MB")
command = f"udp-sender.exe --file \"{file_path}\" --port 8500 --min-receivers {client_count} --nokbd --autostart 3 "
subprocess.run(command, shell=True)
# 等待檔案傳送完成的時間,可以根據需要調整
time.sleep(1)
# 完成所有傳輸,更新狀態
self.status_label.config(text="目前傳輸狀態:傳輸完成!")
except Exception as e:
print(f"傳輸過程中發生錯誤:{e}")
self.status_label.config(text="目前傳輸狀態:傳輸失敗!")
逐一傳輸檔案:在 send_file_list 方法中,對於每個檔案,我們檢查檔案是否存在,然後進行傳輸,這樣每個檔案的傳輸是在前一個檔案完成後才開始。
狀態更新:在每次檔案傳輸開始和結束時更新狀態,以便使用者了解當前進度。
異常處理:加了一些基本的異常處理,這樣如果在傳輸過程中發生錯誤,程式不會崩潰,並會顯示適當的狀態。
這樣的改動可以讓你的程式支持多檔案傳輸,並且在每次傳輸之間確保每個檔案都已經處理完成。
待測2-
import os
import subprocess
import threading
import time
from tkinter import filedialog
class YourClass: # 假設這是你的類別名稱
def select_client_and_file(self):
selected_items = self.tree.selection() # 可以選取多個客戶端
if selected_items:
client_ips = [self.tree.item(item, 'values')[0] for item in selected_items] # 獲取所有選中的客戶端 IP
# 選擇檔案或資料夾
file_paths = filedialog.askopenfilenames(title="選擇檔案")
if file_paths:
self.status_label.config(text="目前傳輸狀態:正在準備傳送檔案...")
# 對每個客戶端發送檔案
for client_ip in client_ips:
self.prepare_file_transfer_list(client_ip, file_paths)
def prepare_file_transfer_list(self, client_ip, file_paths):
print(f"準備傳輸檔案到 {client_ip}:\n{file_paths}")
# 開始傳送檔案
threading.Thread(target=self.send_file_list, args=(client_ip, list(file_paths)), daemon=True).start()
def send_file_list(self, client_ip, file_paths):
try:
for file_path in file_paths:
# 建立檔案傳輸指令
if os.path.exists(file_path):
file_size = os.path.getsize(file_path)
self.status_label.config(text=f"目前傳輸檔案:{os.path.basename(file_path)},大小:{file_size // (1024 * 1024)} MB")
# 開始執行 udp-sender.exe
command = f"udp-sender.exe --file \"{file_path}\" --port 8500 --min-receivers {len(client_ips)} --nokbd --autostart 3"
subprocess.run(command, shell=True)
# 等待檔案傳輸完成,可以根據需要調整時間
time.sleep(1)
# 所有檔案傳輸完成,更新狀態
self.status_label.config(text="目前傳輸狀態:傳輸完成!")
except Exception as e:
print(f"傳送檔案時發生錯誤: {e}")
self.status_label.config(text="傳輸失敗!")
待測3-要在使用者按下傳送檔案按鈕時彈出提示,讓他們知道正在進行檔案傳輸,並請他們等待完成,可以在 select_client_and_file 方法中加入檢查。如果正在傳輸檔案,則顯示提示框。這裡的做法是增加一個標誌位元來追蹤檔案傳輸的狀態。
import socket
import threading
import tkinter as tk
from tkinter import filedialog, ttk, messagebox
import time
import os
import subprocess
class Server:
def __init__(self, root):
self.root = root
self.root.title("Server")
self.root.geometry("800x600")
# 新增標題 "UDP派送檔案"
self.title_label = tk.Label(self.root, text="UDP派送檔案", font=("Arial", 18), fg="deepskyblue")
self.title_label.pack(pady=10)
# 建立表格
self.tree = ttk.Treeview(root, columns=("IP", "MAC", "Name", "Status"), show='headings')
self.tree.heading("IP", text="IP 位置", anchor="center")
self.tree.heading("MAC", text="MAC 位置", anchor="center")
self.tree.heading("Name", text="電腦名稱", anchor="center")
self.tree.heading("Status", text="狀態", anchor="center")
self.tree.column("IP", anchor="center")
self.tree.column("MAC", anchor="center")
self.tree.column("Name", anchor="center")
self.tree.column("Status", anchor="center")
self.tree.pack(fill=tk.BOTH, expand=True)
# 儲存客戶端資料的字典
self.client_data_dict = {}
# 追蹤檔案傳輸狀態
self.is_transferring = False
# 建立廣播按鈕
self.broadcast_button = tk.Button(self.root, text="手動廣播伺服器", command=self.start_broadcast_manually)
self.broadcast_button.pack(pady=10)
# 建立傳輸檔案按鈕
self.transfer_button = tk.Button(self.root, text="傳送檔案", command=self.select_client_and_file)
self.transfer_button.pack(pady=10)
# 新增檔案傳輸狀態顯示欄位,初始化為 "等待中"
self.status_label = tk.Label(self.root, text="目前傳輸狀態:等待中", font=("Arial", 12), fg="black")
self.status_label.pack(pady=10)
# 新增備註區域,左右兩格
self.footer_frame = tk.Frame(self.root)
self.footer_frame.pack(pady=20, fill=tk.X)
# 跑馬燈效果,文字顏色改為gold
self.marquee_text = "感謝ChatGPT 感謝UDPcast "
self.marquee_label = tk.Label(self.footer_frame, text=self.marquee_text, font=("Arial", 12), fg="gold")
self.marquee_label.pack(side=tk.LEFT, padx=10)
# 固定文字,文字顏色改為gold
self.fixed_label = tk.Label(self.footer_frame, text="製作者:WCS 日期:2024.10.01", font=("Arial", 12), fg="gold")
self.fixed_label.pack(side=tk.RIGHT, padx=10)
# 啟動跑馬燈
self.move_text()
# 取得伺服端的本地 IP
self.server_ip = self.get_local_ip()
# 啟動伺服器
self.server_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
self.server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
self.server_socket.bind(("", 37020))
# 啟動接收客戶端報到的線程
threading.Thread(target=self.receive_clients, daemon=True).start()
# 啟動時廣播30秒
threading.Thread(target=self.broadcast_for_30_seconds, daemon=True).start()
def get_local_ip(self):
hostname = socket.gethostname()
local_ip = socket.gethostbyname(hostname)
print(f"伺服端 IP: {local_ip}")
return local_ip
def move_text(self):
# 跑馬燈效果,將字串每次往左移動一個字符
self.marquee_text = self.marquee_text[1:] + self.marquee_text[0]
self.marquee_label.config(text=self.marquee_text)
self.root.after(200, self.move_text) # 每200毫秒移動一次
def receive_clients(self):
while True:
try:
data, addr = self.server_socket.recvfrom(1024)
client_data = data.decode().split(',')
if len(client_data) == 3:
client_ip, client_mac, client_name = client_data
if client_ip != self.server_ip:
self.update_or_insert_client(client_ip, client_mac, client_name, "Online")
else:
print(f"接收到不完整的資料:{client_data}")
except Exception as e:
print(f"接收資料時發生錯誤: {e}")
def update_or_insert_client(self, ip, mac, name, status):
if ip in self.client_data_dict or mac in self.client_data_dict:
item_id = self.client_data_dict.get(ip) or self.client_data_dict.get(mac)
self.tree.item(item_id, values=(ip, mac, name, status))
else:
item_id = self.tree.insert("", "end", values=(ip, mac, name, status))
self.client_data_dict[ip] = item_id
self.client_data_dict[mac] = item_id
def broadcast_for_30_seconds(self):
start_time = time.time()
while time.time() - start_time < 30:
try:
self.server_socket.sendto(b"Server here", ('<broadcast>', 37020))
time.sleep(5)
except Exception as e:
print(f"廣播時發生錯誤: {e}")
print("廣播已停止")
def start_broadcast_manually(self):
threading.Thread(target=self.broadcast_for_30_seconds, daemon=True).start()
def select_client_and_file(self):
if self.is_transferring:
messagebox.showinfo("注意", "請等待傳輸完成")
return
selected_items = self.tree.selection() # 可以選取多個客戶端
if selected_items:
client_ips = [self.tree.item(item, 'values')[0] for item in selected_items] # 獲取所有選中的客戶端 IP
file_paths = filedialog.askopenfilenames(title="選擇檔案")
if file_paths:
self.status_label.config(text="目前傳輸狀態:正在準備傳送檔案...")
self.is_transferring = True # 開始傳輸
for client_ip in client_ips:
self.prepare_file_transfer_list(client_ip, file_paths)
def prepare_file_transfer_list(self, client_ip, file_paths):
transfer_list = "\n".join(file_paths)
print(f"準備傳輸檔案到 {client_ip}:\n{transfer_list}")
threading.Thread(target=self.send_file_list, args=(client_ip, file_paths), daemon=True).start()
def send_file_list(self, client_ip, file_paths):
file_list_str = ",".join(file_paths)
try:
self.server_socket.sendto(file_list_str.encode(), (client_ip, 8500))
print(f"檔案傳輸清單已傳送至 {client_ip}")
time.sleep(2)
selected_clients = self.tree.selection()
client_count = len(selected_clients)
for file_path in file_paths:
if os.path.exists(file_path):
file_size = os.path.getsize(file_path)
self.status_label.config(text=f"目前傳輸檔案:{os.path.basename(file_path)},大小:{file_size // (1024 * 1024)} MB")
command = f"udp-sender.exe --file \"{file_path}\" --port 8500 --min-receivers {client_count} --nokbd --autostart 3"
subprocess.run(command, shell=True)
self.status_label.config(text="目前傳輸狀態:傳輸完成!")
self.is_transferring = False # 傳輸完成
except Exception as e:
print(f"傳送檔案清單時發生錯誤: {e}")
self.status_label.config(text="傳輸失敗!")
self.is_transferring = False # 傳輸失敗
if __name__ == "__main__":
root = tk.Tk()
server_app = Server(root)
root.mainloop()