服務器 參考代碼如下:
from socket import *
import sys
def get_file_content(file_name):
"""獲取文件的內容"""
try:
with open(file_name, "rb") as f:
content = f.read()
return content
except:
print("沒有下載的文件:%s" % file_name)
def main():
if len(sys.argv) != 2:
print("請按照如下方式運行：Python3 xxx.py 7890")
return
else:
# 運行方式為python3 xxx.py 7890
port = int(sys.argv[1])
# 創建socket
tcp_server_socket = socket(AF_INET, SOCK_STREAM)
# 本地信息
address = ('', port)
# 綁定本地信息
tcp_server_socket.bind(address)
# 將主動套接字變為被動套接字
tcp_server_socket.listen(128)
while True:
# 等待客戶端的鏈接，即為這個客戶端發送文件
client_socket, clientAddr = tcp_server_socket.accept()
# 接收對方發送過來的數據
recv_data = client_socket.recv(1024) # 接收1024個字節
file_name = recv_data.decode("utf-8")
print("對方請求下載的文件名為:%s" % file_name)
file_content = get_file_content(file_name)
# 發送文件的數據給客戶端
# 因為獲取打開文件時是以rb方式打開，所以file_content中的數據已經是二進制的格式，因此不需要encode編碼
if file_content:
client_socket.send(file_content)
# 關閉這個套接字
client_socket.close()
# 關閉監聽套接字
tcp_server_socket.close()
if __name__ == "__main__":
main()
客戶端 參考代碼如下:
from socket import *
def main():
# 創建socket
tcp_client_socket = socket(AF_INET, SOCK_STREAM)
# 目的信息
server_ip = input("請輸入服務器ip:")
server_port = int(input("請輸入服務器port:"))
# 鏈接服務器
tcp_client_socket.connect((server_ip, server_port))
# 輸入需要下載的文件名
file_name = input("請輸入要下載的文件名：")
# 發送文件下載請求
tcp_client_socket.send(file_name.encode("utf-8"))
# 接收對方發送過來的數據，最大接收1024個字節（1K）
recv_data = tcp_client_socket.recv(1024)
# print('接收到的數據為:', recv_data.decode('utf-8'))
# 如果接收到數據再創建文件，否則不創建
if recv_data:
with open("[接收]"+file_name, "wb") as f:
f.write(recv_data)
# 關閉套接字
tcp_client_socket.close()
if __name__ == "__main__":
main()
