# 服務(wù)器的地址
server_address = ('127.0.0.1', 8000)

def send(dir_name, data_format, file_name):
 # 與接收端建立socket通信
 sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  # AF_INET（TCP/IP – IPv4）協(xié)議
 sock.connect(server_address)
 # 每次通信都帶一個(gè)通信頭，表明數(shù)據(jù)源的類(lèi)型(紅外還是可見(jiàn)光)，要保存數(shù)據(jù)幀的文件夾名file_name
 # 你可以不要數(shù)據(jù)格式，這里可以定義成你自己的形式，也算是一種安全機(jī)制
 sock.send('{}|{}'.format(data_format, file_name).encode())  # 默認(rèn)編碼 utf-8,發(fā)送文件長(zhǎng)度和文件名
 reply = sock.recv(1024)
 # 按照文件名排序，0.png,1.png
 file_list = os.listdir(dir_name)
 file_list.sort(key=lambda x: int(x[:-4]))
 if 'ok' == reply.decode():  # 確認(rèn)一下服務(wù)器get到文件長(zhǎng)度和文件名數(shù)據(jù)
  i = 0
  print(len(file_list))
  for file_name in file_list:
data = file_deal(os.path.join(dir_name, file_name))
sock.send('{}|{}'.format(len(data), file_name).encode())
sock.recv(1024)
go = 0
total = len(data)
while go < total:  # 發(fā)送文件
 data_to_send = data[go:go + total//2]
 sock.send(data_to_send)
 go += len(data_to_send)
sock.recv(1024).decode()
i += 1
if i < len(file_list):
 sock.send(b'continue')
  sock.send(b'over')
  sock.close()
  sys.exit(0)

def file_deal(file_path):  # 讀取文件的方法
 mes = b''
 try:
  file = open(file_path, 'rb')
  mes = file.read()
 except:
  print('error{}'.format(file_path))
 else:
  file.close()
  return mes

LOCAL_IP = '127.0.0.1'  # 本機(jī)測(cè)試使用ip，局域網(wǎng)中使用需更換ip
PORT = 8000  # 隨意指定一個(gè)端口
def server():
 sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  # socket.AF_INET 指ipv4  socket.SOCK_STREAM 使用tcp協(xié)議
 sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)  # 設(shè)置端口
 sock.bind((LOCAL_IP, PORT))  # 綁定端口
 sock.listen(3)  # 監(jiān)聽(tīng)端口
 while True:
  sc, sc_name = sock.accept()  # 當(dāng)有請(qǐng)求到指定端口是 accept()會(huì)返回一個(gè)新的socket和對(duì)方主機(jī)的（ip,port）
  print('收到{}機(jī)器請(qǐng)求'.format(sc_name))
  info = sc.recv(1024)  # 接受客戶端發(fā)來(lái)的協(xié)議頭，區(qū)分?jǐn)?shù)據(jù)源
  # 安全處理：如果不是以這個(gè)協(xié)議頭開(kāi)始，認(rèn)為是非法接入，就直接斷掉。這里可以自己定義一些安全消息機(jī)制
  print(info)
  try:
data_format, directory_name = info.decode().split("|")
sc.send(b'ok')  # 表示收到文件長(zhǎng)度和文件名
  except:
print('協(xié)議頭不對(duì)，自動(dòng)斷開(kāi)連接')
sc.close()
continue
  if not os.path.exists(directory_name):
os.mkdir(directory_name)
  # 協(xié)議頭正確之后，不斷接收發(fā)來(lái)的數(shù)據(jù)幀
  while True:
head_info = sc.recv(1024)
# print(data_info)
length, file_name = head_info.decode().split('|')
sc.send(b'ok')
if length and file_name:
 print(file_name)
 newfile = open(os.path.join(directory_name, file_name), 'wb')  # 這里可以使用從客戶端解析出來(lái)的文件名
 file = b''
 total = int(length)
 get = 0
 while get < total:  # 接收文件
  data = sc.recv(total//2)
  file += data
  get = get + len(data)
 sc.send(b'ok')
 print('應(yīng)該接收{(diào)},實(shí)際接收{(diào)}'.format(length, len(file)))
 if file:
  print('actually length:{}'.format(len(file)))
  newfile.write(file[:])
  newfile.close()
reply = sc.recv(1024)
if reply.decode() == "over":
 break
server()