网络编程: 1, 简单版的 上传 与 下载; 2, 简单升级版的 上传 与 下载( 有详细注释)

ps: 下面两个版本全部是 基于 tcp协议完成的.

 

简单版:

  服务器(server)代码:

    因为是简单版的所以 无注释

 1 from socket import socket
 2 sk = socket()
 3 sk.bind(('127.0.0.1', 8585))
 4 sk.listen()
 5 
 6 def display(): return '1, 上传\n2, 下载'
 7 
 8 
 9 def upload(word,chose_path):
10     # chose_path = input('>>>请输入要上传的文件路径:')
11     with open(chose_path.decode('utf-8'), 'wb') as f:
12         f.write(word)
13 
14 
15 def download():
16     name = '__init__.py'
17     with open(name, 'rb') as f:
18         word = f.read()
19     return name.encode('utf-8'), word
20 
21 
22 def main():
23     while 1:
24         conn, addr = sk.accept()
25         while 1:
26             conn.send(display().encode('utf-8'))
27             msg_r = conn.recv(1024).decode('utf-8')
28             if msg_r == 'q':continue
29             if msg_r == '1':
30                 word = conn.recv(1024*10)
31                 chose_path = conn.recv(1024 * 10)
32                 upload(word, chose_path)
33                 conn.send('上传成功'.encode('utf-8'))
34             elif msg_r == '2':
35                 name, word = download()
36                 conn.send(name)
37                 conn.send(word)
38 
39 main()

 

  客户端(client) 代码:

 1 from socket import socket
 2 
 3 sk = socket()
 4 sk.connect(('127.0.0.1', 8585))
 5 flag = True
 6 
 7 def upload():
 8     chose_path = input('>>>请输入要上传的文件路径:')
 9     with open(chose_path, 'rb') as f:
10         word = f.read()
11         return chose_path.encode('utf-8'), word
12 
13 
14 def download(name, word):
15     with open(name,'wb') as f:
16         f.write(word)
17     return '下载成功'
18 
19 
20 def main():
21     global flag
22     msg_r = sk.recv(1024).decode('utf-8')
23     print(msg_r)
24     choise = input('>>>请选择功能ID(输入"q"退出程序):').strip().encode('utf-8')
25     sk.send(choise)
26     if choise == 'q'.encode('utf-8'): flag = False
27     if choise == '1'.encode('utf-8'):
28         chose_path, word = upload()
29         sk.send(word)
30         sk.send(chose_path)
31         print(sk.recv(1024).decode('utf-8'))
32     elif choise == '2'.encode('utf-8'):
33         name = sk.recv(1024).decode('utf-8')
34         word = sk.recv(1024)
35         print(download(name, word))
36 
37 while flag:
38     main()
39 sk.close()

 

 

 

 

 


 

 

 

 

 

 

简单的 升级版 的上传与下载:

  服务器(server):

    有详细注释:           此程序会出现个别粘包现象

 1 from socket import socket
 2 import os
 3 import pickle
 4 
 5 sk = socket()
 6 sk.bind(('127.0.0.1', 8589))
 7 sk.listen()
 8 
 9 def display(): return '1, 上传\n2, 下载'     #  此函数用来发送第一个选择
10 
11 
12 def update(name): # 更名
13     chose = conn.recv(1024).decode('utf-8')  # 接收客户的选择
14     if chose == '1':pass   # 如果选择 1 覆盖 则不用操作
15     elif chose == '2':      # 如果选择 2 更改 接收客户端的消息
16         name = conn.recv(1024).decode('utf-8')      #  新文件的名字
17     return name
18 
19 
20 def upload(word,chose_path):    # 上传函数
21     chose_path = chose_path.decode('utf-8')  # 把接收的文件名 解码
22 
23     # 判断 该文件 是否已经存在与 服务器 当前工作目录下:
24     if chose_path in os.listdir('E:/老男孩工具/13期课后作业及视频/day33/myself_test/new_upload_download/updata_server_path'):    # 路径 按照自己的文件夹来自定义
25         # 如果在, 向客户端发消息, 并且询问 用户的选择
26         conn.send('服务器已存在该文件:选择覆盖还是更名:\n1, 覆盖\n2, 更名'.encode('utf-8'))
27         name = update(chose_path)     # 调用 update() 更改函数 获得新文件的名字 或 老文件的名字
28         if not name:chose_path = name  # 如果 name 不为None  给文件进行改名:
29     with open(chose_path, 'wb') as f:  # 打开文件 进行写入
30         f.write(word)
31 
32 
33 def download(): # 下载
34     # 把服务器工作目录下 所有文件 装到列表里
35     path_list = os.listdir(r'E:\老男孩工具\13期课后作业及视频\day33\myself_test\new_upload_download\updata_server_path')
36     lis = [{x:y} for x,y in enumerate(path_list,1)] # 生成一个 有序的元素为字典的 内容列表
37     pic = pickle.dumps(lis)    # 序列化一下
38     conn.send(pic)          # 发送给 客户端, 然后询问 客户要下载那个文件的id
39     file_id = conn.recv(1024).decode('utf-8')  #   接收到 id
40     name = lis[int(file_id)-1][int(file_id)]    # 取得 相对应的文件名
41     with open(name, 'rb') as f:         # 打开文件, 读到内容
42         word = f.read()
43     return name.encode('utf-8'), word    # 返回文件名字, 和 内容
44 
45 
46 def main(): ## 服务器
47     while 1:
48         conn.send(display().encode('utf-8'))  # 发送 第一条消息:  '1, 上传\n2, 下载'
49         msg_r = conn.recv(1024).decode('utf-8') #  接收 客户端的选择信息
50         if msg_r == 'q':break               # 如果是q 退出程序
51         if msg_r == '1':                # 如果 选择 1  此时需要在客户端 向用户索要 文件信息.
52             word = conn.recv(1024*10)    # 接收 文件内容
53             chose_path = conn.recv(1024 * 10)   #   接收文件名字
54             upload(word, chose_path)        # 调用上传 函数 并且把文件内容 和名字 传进去
55             conn.send('上传成功'.encode('utf-8'))   # 如果程序成功运行到了这里 就把消息发过去.
56         elif msg_r == '2':              # 如果选择2 下载
57             name, word = download()     # 调用 下载函数 download() 把返回值 分别 赋值 name, word
58             conn.send(name)                # 发送 文件名字.
59             conn.send(word)         # 最后一条消息: 文件内容
60     return
61 
62 
63 if __name__ == '__main__':
64     while 1:
65         conn, addr = sk.accept()
66         main()
67         conn.close()

 

    客户端(client):

      同样 有详细的注释:

 1 import pickle
 2 from socket import socket
 3 import os
 4 
 5 sk = socket()
 6 sk.connect(('127.0.0.1', 8589))
 7 flag = True
 8 
 9 def upload():
10     chose_path = input('>>>请输入要上传的文件路径:')
11     with open(chose_path, 'rb') as f:
12         word = f.read()                         #  读取文件内容
13     chose_path = os.path.basename(chose_path)   # 取得 路径的 文件名
14     return chose_path.encode('utf-8'), word    # 把word  name  返回
15 
16 
17 def download(name, word):
18     path = input('>>>请选择工作路径:')
19     new_path = os.path.join(path, name)
20     with open(new_path,'wb') as f:
21         f.write(word)
22     return '下载成功'
23 
24 
25 def update(): # 更名
26     chose = input('>>>请输入您的选择:').strip().encode('utf-8')
27     sk.send(chose) # 把选择发给 服务器, 让服务器做判断
28     if chose == '1'.encode('utf-8'):pass   # 如果是1 则不作任何操作
29     elif chose == '2'.encode('utf-8'):      # 如果是2 则更名
30         new_name = input('>>>请输入新的名字:').encode('utf-8')
31         sk.send(new_name)        # 把新名字 给服务器 发过去
32         print(sk.recv(1024).decode('utf-8')) # 接收 服务器 发送的 成功的消息 并且打印出来
33 
34 
35 def main():   # 客户端
36     global flag
37     msg_r = sk.recv(1024).decode('utf-8')   #  接收第一条消息,   '1, 上传\n2, 下载'
38     print(msg_r)
39     choise = input('>>>请选择功能ID(输入"q"退出程序):').strip().encode('utf-8')
40     sk.send(choise)  # 把 客户选择的功能发送给 服务器
41     if choise == 'q'.encode('utf-8'): flag = False
42     if choise == '1'.encode('utf-8'):
43         chose_path, word = upload()  # 调用upload() 上传函数  返回 word 和 文件名
44         sk.send(word) #         把 word发送给服务器
45         sk.send(chose_path) #       把文件名发送给 服务器
46         ret = sk.recv(1024).decode('utf-8')     # 接收 服务器传输过来的 判断是否 存在的语句
47         print(ret)
48         if ret == '服务器已存在该文件:选择覆盖还是更名:\n1, 覆盖\n2, 更名':      # 如果是这句话就 调用下面的函数, 如果不是这句话 会打印 上传成功
49             update()   # 此函数的功能 和 服务器里面的 功能一样 :更名
50 
51     elif choise == '2'.encode('utf-8'):    # 当选择 2 下载 时
52         pic = pickle.loads(sk.recv(1024))     #  把服务器发来的 pickle 字典 反序列化了
53         print('下面是所有子文件:')
54         for i in pic:
55             print(i)                        #  一个一个打印出来
56         file_id = input('>>>请输入想要下载的文件的ID:').strip().encode('utf-8')
57         sk.send(file_id)                    # 把 id 发给 服务器
58         name = sk.recv(1024).decode('utf-8')   # 接收 服务器发来的 文件的名字
59         word = sk.recv(1024*1000)   #  最后一条消息: 接收文件名
60         download(name, word)         # 此函数 是用来把内容写入新文件
61         print('下载成功')
62 
63 
64 if __name__ == '__main__':
65     while flag:
66         main()
67     sk.close()

 

posted @ 2018-07-18 20:47  浮生凉年  阅读(257)  评论(0编辑  收藏  举报