本實例有文件傳輸相關功能，包括：文件校驗、進度條打印、斷點續傳
客戶端示例：
import
socket
import
json
import
os
import
hashlib
CODE
=
{
'
1001
'
:
'
重新上傳文件
'
}
def
file_md5(file_path):
obj
= open(file_path,
'
rb
'
)
m
=
hashlib.md5()
for
line
in
obj:
m.update(line)
obj.close()
return
m.hexdigest()
def
jdt(size,total_size):
'''
顯示進度條
'''
val
= int(size/total_size * 100
)
print
(
'
\r%s%%|%s
'
%(val,
'
#
'
*val,),end=
''
)
def
send_file(exist_size,file_total_size):
'''
發送文件
:param exist_size: 開始讀取字節的位置
:param file_total_size: 文件大小
:return:
'''
f
= open(file_path,
'
rb
'
)
f.seek(exist_size)
send_size
=
exist_size
while
send_size <
file_total_size:
data
= f.read(1024
)
sk.sendall(data)
send_size
+=
len(data)
jdt(send_size,file_total_size)
f.close()
print
(
'
上傳成功
'
)
def
upload(file_path):
'''
文件上傳（含斷點）
'''
file_path
=
'
111.jpg
'
file_md5_val
=
file_md5(file_path)
file_name
=
os.path.basename(file_path)
file_size
=
os.stat(file_path).st_size
cmd_dict
= {
'
cmd
'
:
'
upload
'
,
'
file_name
'
: file_name,
'
size
'
: file_size,
'
md5
'
: file_md5_val}
upload_cmd_bytes
= json.dumps(cmd_dict).encode(
'
utf8
'
)
sk.sendall(upload_cmd_bytes)
#
2.等待服務端相應
response = json.loads(sk.recv(8096).decode(
'
utf8
'
))
if
response[
'
code
'
] == 1001
:
send_file(0,file_size)
else
:
#
斷點續傳
exist_size = response(
'
size
'
)
send_file(exist_size,file_size)
sk
=
socket.socket()
sk.connect((
'
127.0.0.1
'
,8001
))
while
1
:
#
upload|文件路徑
user_input = input(
'
命令>>>
'
)
#
1、自定義協議{'cmd':'upload','file_name':'...'}
cmd,file_path = user_input.split(
'
|
'
,maxsplit=1
)
if
cmd ==
'
upload
'
:
upload(file_path)
elif
cmd ==
'
download
'
:
pass
?
服務端示例：
import
socketserver
import
json
import
os
import
shutil
CODE
=
{
'
1001
'
:
'
重新上傳文件
'
}
def
upload(cmd_dict,conn,username):
'''
服務端完成上傳文件（含斷點續傳）
'''
#
2.獲取文件信息
file_md5 = cmd_dict[
'
md5
'
]
file_name
= cmd_dict[
'
file_name
'
]
file_md5_path
= os.path.join(
'
home
'
, username, file_md5)
file_name_path
= os.path.join(
'
home
'
, username, file_name)
upload_file_size
= cmd_dict[
'
size
'
]
#
3.判斷文件是否存在
exist =
os.path.exists(file_md5_path)
if
not
exist:
#
不續傳
#
3.1.1通知可以開始上傳，已經準備好
response = {
'
code
'
: 1001
}
conn.sendall(json.dumps(response).encode(
'
utf8
'
))
#
3.1.2接收上傳的文件內容
f = open(file_md5_path,
'
wb
'
)
recv_size
=
0
while
recv_size <
upload_file_size:
data
= conn.recv(1024
)
f.write(data)
#
寫到內存
f.flush()
#
刷到硬盤
recv_size +=
len(data)
f.close()
#
3.1.3改文件名
shutil.move(file_md5_path, file_name_path)
else
:
#
存在就續傳
#
3.2續傳 + 大小
exist_size =
os.stat(file_md5_path).st_size
response
= {
'
code
'
: 1002,
'
size
'
: exist_size}
conn.sendall(json.dumps(response).encode(
'
utf8
'
))
f
= open(file_md5_path,
'
ab
'
)
recv_size
=
exist_size
while
recv_size <
upload_file_size:
data
= conn.recv(1024
)
f.write(data)
#
寫到內存
f.flush()
#
刷到硬盤
recv_size +=
len(data)
f.close()
#
3.1.3改文件名
shutil.move(file_md5_path, file_name_path)
class
NBServer(socketserver.BaseRequestHandler):
def
handle(self):
'''
self.request是請求客戶端的socket對象
'''
#
1.接受命令
upload_cmd_bytes = self.request.recv(8096
)
cmd_dict
= json.loads(upload_cmd_bytes.decode(
'
utf8
'
))
if
cmd_dict[
'
cmd
'
] ==
'
upload
'
:
upload(cmd_dict,self.request,
'
zhangzijian
'
)
elif
cmd_dict[
'
cmd
'
] ==
'
download
'
:
pass
if
__name__
==
'
__main__
'
:
server
= socketserver.ThreadingTCPServer((
'
127.0.0.1
'
,8001
),NBServer)
server.serve_forever()
?
更多文章、技術交流、商務合作、聯系博主
微信掃碼或搜索：z360901061
微信掃一掃加我為好友
QQ號聯系： 360901061
您的支持是博主寫作最大的動力，如果您喜歡我的文章，感覺我的文章對您有幫助，請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧，狠狠點擊下面給點支持吧，站長非常感激您！手機微信長按不能支付解決辦法：請將微信支付二維碼保存到相冊，切換到微信，然后點擊微信右上角掃一掃功能，選擇支付二維碼完成支付。
【本文對您有幫助就好】元
