61 lines
1.7 KiB
Python
61 lines
1.7 KiB
Python
|
import socket
|
||
|
|
import struct
|
||
|
|
import json
|
||
|
|
import os
|
||
|
|
from datetime import datetime
|
||
|
|
|
||
|
|
|
||
|
|
def recvdata(conn, path):
|
||
|
|
"""
|
||
|
|
接受文件
|
||
|
|
:param conn:
|
||
|
|
:param path:
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
header_size = struct.unpack('i', conn.recv(4))[0]
|
||
|
|
header_bytes = conn.recv(header_size)
|
||
|
|
header_json = header_bytes.decode('utf-8')
|
||
|
|
header_dic = json.loads(header_json)
|
||
|
|
content_len = header_dic['contentlen']
|
||
|
|
content_name = header_dic['contentname']
|
||
|
|
recv_len = 0
|
||
|
|
fielpath = os.path.join(path, '%s_%s' %(datetime.now().strftime("%m%d%H%M"), content_name))
|
||
|
|
file = open(fielpath, 'wb')
|
||
|
|
while recv_len < content_len:
|
||
|
|
correntrecv = conn.recv(1024 * 1000)
|
||
|
|
file.write(correntrecv)
|
||
|
|
recv_len += len(correntrecv)
|
||
|
|
file.close()
|
||
|
|
return fielpath
|
||
|
|
|
||
|
|
|
||
|
|
def senddata(conn, path, message=None):
|
||
|
|
name = os.path.basename(os.path.realpath(path))
|
||
|
|
if not message:
|
||
|
|
with open(path, 'rb') as file:
|
||
|
|
content = file.read()
|
||
|
|
headerdic = dict(
|
||
|
|
contentlen=len(content),
|
||
|
|
contentname=name
|
||
|
|
)
|
||
|
|
headerjson = json.dumps(headerdic)
|
||
|
|
headerbytes = headerjson.encode('utf-8')
|
||
|
|
headersize = len(headerbytes)
|
||
|
|
conn.send(struct.pack('i', headersize))
|
||
|
|
conn.send(headerbytes)
|
||
|
|
conn.sendall(content)
|
||
|
|
else:
|
||
|
|
headerdic = dict(
|
||
|
|
contentlen=len(path),
|
||
|
|
contentname='message'
|
||
|
|
)
|
||
|
|
headerjson = json.dumps(headerdic)
|
||
|
|
headerbytes = headerjson.encode('utf-8')
|
||
|
|
headersize = len(headerbytes)
|
||
|
|
conn.send(struct.pack('i', headersize))
|
||
|
|
conn.send(headerbytes)
|
||
|
|
conn.sendall(path.encode('utf-8'))
|
||
|
|
|
||
|
|
|
||
|
|
|