{每日一题}:tcp协议实现简单的文件下载器(单任务版)

文件下载器客户端

　　这个版本的只是为了方便回顾一下TCP客服端,服务端的创建流程,缺点就是  服务器一次只能让一个人访问下载,过两个写个使用面向对象写一个多线程版的强化一下.

 from socket import *


 def mian():
     # 1. 创建套接字
     tcp_client_scoket = socket(AF_INET,SOCK_STREAM)

     # 2.连接本地服务器
     tcp_client_scoket.connect(('192.168.0.117',8889))

     # 3.输入要下载的文件名
     file_name = input('请输入要下载的文件名：')
     tcp_client_scoket.send(file_name.encode())

     # 4.接收传过来的文件内容
     data = tcp_client_scoket.recv(1024*1024*1024)
     print(data)

     # 5.写入本地
     if data:
         with open('/home/python/Desktop/' + file_name,'wb') as f:
             f.write(data)

     tcp_client_scoket.close()

 if __name__ == '__main__':
     mian()

文件下载器服务端

 from socket import *

 def get_file_name(file_name):
     """获取文件内容"""
     try:
         with open('/home/python/Desktop/test/'+file_name,'rb') as f:
             content = f.read()
             return content
     except:
         print('无此文件')  # 会返回一个空值

 def main():
     # 1.创建服务器的tcp套接字
     tcp_serve_socket = socket(AF_INET,SOCK_STREAM)

     # 2.端口复用
     tcp_serve_socket.setsockopt(SOL_SOCKET,SO_REUSEADDR,True)

     # 3.绑定服务器端口
     tcp_serve_socket.bind(('',8889))

     # 4.开始监听
     tcp_serve_socket.listen(5)

     while True:
         new_socket,client_addr = tcp_serve_socket.accept()
         print("来自%s的下载请求" % str(client_addr))

         # 5.接收来自服务端的文件名
         file_data = new_socket.recv(1024)
         file_name = file_data.decode()

         # 6.查找是否有此文件并写出
         content = get_file_name(file_name)

         # 7. 发送文件内容
         if content:
             new_socket.send(content)
             print('发送成功')

         #8.关闭new_socket套接字
         new_socket.close()

     tcp_serve_socket.close()

 if __name__ == '__main__':
     main()

如果你和我有共同爱好,我们可以加个好友一起交流!