ps: 下面两个版本全部是 基于 tcp协议完成的.
简单版(无注释):
服务器(server)代码:
from socket import socket
sk = socket()
sk.bind(('127.0.0.1', 8585))
sk.listen()
def display(): return '1, 上传
2, 下载'
def upload(word,chose_path):
# chose_path = input('>>>请输入要上传的文件路径:')
with open(chose_path.decode('utf-8'), 'wb') as f:
f.write(word)
def download():
name = '__init__.py'
with open(name, 'rb') as f:
word = f.read()
return name.encode('utf-8'), word
def main():
while 1:
conn, addr = sk.accept()
while 1:
conn.send(display().encode('utf-8'))
msg_r = conn.recv(1024).decode('utf-8')
if msg_r == 'q':continue
if msg_r == '1':
word = conn.recv(1024*10)
chose_path = conn.recv(1024 * 10)
upload(word, chose_path)
conn.send('上传成功'.encode('utf-8'))
elif msg_r == '2':
name, word = download()
conn.send(name)
conn.send(word)
main()
客户端(client) 代码:
from socket import socket
sk = socket()
sk.connect(('127.0.0.1', 8585))
flag = True
def upload():
chose_path = input('>>>请输入要上传的文件路径:')
with open(chose_path, 'rb') as f:
word = f.read()
return chose_path.encode('utf-8'), word
def download(name, word):
with open(name,'wb') as f:
f.write(word)
return '下载成功'
def main():
global flag
msg_r = sk.recv(1024).decode('utf-8')
print(msg_r)
choise = input('>>>请选择功能ID(输入"q"退出程序):').strip().encode('utf-8')
sk.send(choise)
if choise == 'q'.encode('utf-8'): flag = False
if choise == '1'.encode('utf-8'):
chose_path, word = upload()
sk.send(word)
sk.send(chose_path)
print(sk.recv(1024).decode('utf-8'))
elif choise == '2'.encode('utf-8'):
name = sk.recv(1024).decode('utf-8')
word = sk.recv(1024)
print(download(name, word))
while flag:
main()
sk.close()
升级版 (有注释,此程序会出现个别粘包现象):
服务器(server):
from socket import socket
import os
import pickle
sk = socket()
sk.bind(('127.0.0.1', 8589))
sk.listen()
def display(): return '1, 上传
2, 下载' # 此函数用来发送第一个选择
def update(name): # 更名
chose = conn.recv(1024).decode('utf-8') # 接收客户的选择
if chose == '1':pass # 如果选择 1 覆盖 则不用操作
elif chose == '2': # 如果选择 2 更改 接收客户端的消息
name = conn.recv(1024).decode('utf-8') # 新文件的名字
return name
def upload(word,chose_path): # 上传函数
chose_path = chose_path.decode('utf-8') # 把接收的文件名 解码
# 判断 该文件 是否已经存在与 服务器 当前工作目录下:
if chose_path in os.listdir('E:/老男孩工具/13期课后作业及视频/day33/myself_test/new_upload_download/updata_server_path'): # 路径 按照自己的文件夹来自定义
# 如果在, 向客户端发消息, 并且询问 用户的选择
conn.send('服务器已存在该文件:选择覆盖还是更名:
1, 覆盖
2, 更名'.encode('utf-8'))
name = update(chose_path) # 调用 update() 更改函数 获得新文件的名字 或 老文件的名字
if not name:chose_path = name # 如果 name 不为None 给文件进行改名:
with open(chose_path, 'wb') as f: # 打开文件 进行写入
f.write(word)
def download(): # 下载
# 把服务器工作目录下 所有文件 装到列表里
path_list = os.listdir(r'E:老男孩工具13期课后作业及视频day33myself_test
ew_upload_downloadupdata_server_path')
lis = [{x:y} for x,y in enumerate(path_list,1)] # 生成一个 有序的元素为字典的 内容列表
pic = pickle.dumps(lis) # 序列化一下
conn.send(pic) # 发送给 客户端, 然后询问 客户要下载那个文件的id
file_id = conn.recv(1024).decode('utf-8') # 接收到 id
name = lis[int(file_id)-1][int(file_id)] # 取得 相对应的文件名
with open(name, 'rb') as f: # 打开文件, 读到内容
word = f.read()
return name.encode('utf-8'), word # 返回文件名字, 和 内容
def main(): ## 服务器
while 1:
conn.send(display().encode('utf-8')) # 发送 第一条消息: '1, 上传
2, 下载'
msg_r = conn.recv(1024).decode('utf-8') # 接收 客户端的选择信息
if msg_r == 'q':break # 如果是q 退出程序
if msg_r == '1': # 如果 选择 1 此时需要在客户端 向用户索要 文件信息.
word = conn.recv(1024*10) # 接收 文件内容
chose_path = conn.recv(1024 * 10) # 接收文件名字
upload(word, chose_path) # 调用上传 函数 并且把文件内容 和名字 传进去
conn.send('上传成功'.encode('utf-8')) # 如果程序成功运行到了这里 就把消息发过去.
elif msg_r == '2': # 如果选择2 下载
name, word = download() # 调用 下载函数 download() 把返回值 分别 赋值 name, word
conn.send(name) # 发送 文件名字.
conn.send(word) # 最后一条消息: 文件内容
return
if __name__ == '__main__':
while 1:
conn, addr = sk.accept()
main()
conn.close()
客户端(client) 代码:
import pickle
from socket import socket
import os
sk = socket()
sk.connect(('127.0.0.1', 8589))
flag = True
def upload():
chose_path = input('>>>请输入要上传的文件路径:')
with open(chose_path, 'rb') as f:
word = f.read() # 读取文件内容
chose_path = os.path.basename(chose_path) # 取得 路径的 文件名
return chose_path.encode('utf-8'), word # 把word name 返回
def download(name, word):
path = input('>>>请选择工作路径:')
new_path = os.path.join(path, name)
with open(new_path,'wb') as f:
f.write(word)
return '下载成功'
def update(): # 更名
chose = input('>>>请输入您的选择:').strip().encode('utf-8')
sk.send(chose) # 把选择发给 服务器, 让服务器做判断
if chose == '1'.encode('utf-8'):pass # 如果是1 则不作任何操作
elif chose == '2'.encode('utf-8'): # 如果是2 则更名
new_name = input('>>>请输入新的名字:').encode('utf-8')
sk.send(new_name) # 把新名字 给服务器 发过去
print(sk.recv(1024).decode('utf-8')) # 接收 服务器 发送的 成功的消息 并且打印出来
def main(): # 客户端
global flag
msg_r = sk.recv(1024).decode('utf-8') # 接收第一条消息, '1, 上传
2, 下载'
print(msg_r)
choise = input('>>>请选择功能ID(输入"q"退出程序):').strip().encode('utf-8')
sk.send(choise) # 把 客户选择的功能发送给 服务器
if choise == 'q'.encode('utf-8'): flag = False
if choise == '1'.encode('utf-8'):
chose_path, word = upload() # 调用upload() 上传函数 返回 word 和 文件名
sk.send(word) # 把 word发送给服务器
sk.send(chose_path) # 把文件名发送给 服务器
ret = sk.recv(1024).decode('utf-8') # 接收 服务器传输过来的 判断是否 存在的语句
print(ret)
if ret == '服务器已存在该文件:选择覆盖还是更名:
1, 覆盖
2, 更名': # 如果是这句话就 调用下面的函数, 如果不是这句话 会打印 上传成功
update() # 此函数的功能 和 服务器里面的 功能一样 :更名
elif choise == '2'.encode('utf-8'): # 当选择 2 下载 时
pic = pickle.loads(sk.recv(1024)) # 把服务器发来的 pickle 字典 反序列化了
print('下面是所有子文件:')
for i in pic:
print(i) # 一个一个打印出来
file_id = input('>>>请输入想要下载的文件的ID:').strip().encode('utf-8')
sk.send(file_id) # 把 id 发给 服务器
name = sk.recv(1024).decode('utf-8') # 接收 服务器发来的 文件的名字
word = sk.recv(1024*1000) # 最后一条消息: 接收文件名
download(name, word) # 此函数 是用来把内容写入新文件
print('下载成功')
if __name__ == '__main__':
while flag:
main()
sk.close()