网络编程小练习1--登录上传,下载功能,类型网盘功能

    # 1.登录 + 文件下载
    # 用户必须登录才能下载
    # 用户是否登录应该记录在服务器
    # 并且用户可以自己选择 上传 还是 下载

server端

# _*_ coding : UTF-8 _*_
import socket
import  json
import  struct
import os

dic_key = {'alex':'123'}

dir_name = os.path.dirname(__file__)  #返回path的目录 E:/untitled3/test作业/day30/上传下载  显示当前路径
dowload_path = os.path.join(dir_name,'服务端的下载目录')  #将多个路径组合后返回E:/untitled3/test作业/day30/上传下载服务端的下载目录
dowload_file = os.listdir(dowload_path) #列出指定目录下的所有文件和子目录,以列表方式 获取下载的目录文件列表
# print(dowload_file)   #['下载测试文件1.txt', '下载测试文件2.txt']


sk=socket.socket()
sk.bind(('127.0.0.1',7001))
sk.listen()

conn,addr = sk.accept()

res = conn.recv(1024).decode('utf-8') #接收用户名,密码的字符串
recv_dic = json.loads(res) #转换

if recv_dic['username']  in dic_key and recv_dic['password'] == dic_key[recv_dic['username']]:
    print(f'{recv_dic["username"]}登录成功')
    conn.send(b'0') #给客户端发送回执

    new_content = conn.recv(1024).decode('utf-8')
    if new_content == '1':
        lens =conn.recv(4)
        length =struct.unpack('i',lens)[0]
        f_fun =conn.recv(length).decode('utf-8')
        file_dic = json.loads(f_fun)
        with open(file_dic['filename'],mode='wb') as f:
            if file_dic['filesize'] > 0:
                content =conn.recv(1024)
                file_dic['filesize'] -= len(content)
                f.write(content)
        conn.close()

    elif new_content == '2':
        seng_filelist=json.dumps(dowload_file).encode('utf-8')
        conn.send(seng_filelist)
        down_file = conn.recv(1024).decode('utf-8')  #接收要下载的文件名
        down_path = os.path.join(dowload_path,down_file) #下载路径拼接  取得文件
        down_size = os.path.getsize(down_path) #取得文件大小
        # dic = {'filename':down_path,'filesize':down_size}
        # new_dic = json.dumps(dic)
        # conn.send(new_dic.encode('utf-8'))
        conn.send(str(down_size).encode('utf-8'))  #这样可以不用json转成字符串
        with open(down_path,mode='rb') as f:
            while down_size >0:
                con = f.read(1024)
                down_size -= len(con)
                conn.send(con)
    else:
        print('登录失败')
        conn.close()

sk.close()

client端

# _*_ coding : UTF-8 _*_

import socket
import json
import  struct
import os
num = ['上传','下载']

sk= socket.socket()
sk.connect(('127.0.0.1',7001))


username = input('请输入用户名:').strip()
password = input('请输入密码:').strip()

dic = {'username':username,'password':password} #用户名,密码

srt_dic = json.dumps(dic) #转成字符串
sk.send(srt_dic.encode('utf-8'))  #发送

login_mag = sk.recv(1024).decode('utf-8')
if login_mag == '0':
    print('登录成功')
    for i,v in enumerate(num,1):
        print(i,v)

    no = input('请输入功能序号').strip()
    if no == '1': #上传
        sk.send(b'1')
        abs_path = r'E:msdia80.dll'
        filename = os.path.basename(abs_path) #获得文件名
        filesize =  os.path.getsize(abs_path)  #文件大小
        dic = {'filename':filename,'filesize':filesize}
        new_dic=json.dumps(dic).encode('utf-8')
        s_dic = struct.pack('i',len(new_dic))
        sk.send(s_dic)
        sk.send(new_dic)

        with open(abs_path,mode='rb') as f:
            if filesize>0:
                content=f.read(1024)
                filesize -= len(content)
                sk.send(content)
        sk.close()

    elif no == '2':#下载
        sk.send(b'2')
        ret = sk.recv(1024).decode('utf-8')
        down_list = json.loads(ret)
        while True:
            for v ,i in enumerate(down_list,1):
                print(v,i)
            no = input('请输入序号选择下载的文件:').strip()
            if no.isdecimal() and 0 < int(no) <= len(down_list):
               down_file = down_list[int(no)-1].encode('utf-8')  #选择的文件名
               sk.send(down_file)
               down_size = sk.recv(1024).decode('utf-8') #接收下载文件的大小
               size = int(down_size) #服务端用str转成字符串,客户端用int转成大小
               with open(down_file,mode='wb') as f:
                   while size > 0:
                       con = sk.recv(1024)
                       size -= len(con)
                       f.write(con)
               sk.close()
               break

            else:
                print('选择的序号有误,请重新输入')
                continue

2.用socoketservet模块

server端

# _*_ coding : UTF-8 _*_
import socket
import  json
import  struct
import os
import socketserver

dic_key = {'zeng':'123'}

dir_name = os.path.dirname(__file__)  #返回path的目录 E:/untitled3/test作业/day30/上传下载  显示当前路径
dowload_path = os.path.join(dir_name,'服务端的下载目录')  #将多个路径组合后返回E:/untitled3/test作业/day30/上传下载服务端的下载目录
dowload_file = os.listdir(dowload_path) #列出指定目录下的所有文件和子目录,以列表方式 获取下载的目录文件列表
# print(dowload_file)   #['下载测试文件1.txt', '下载测试文件2.txt']


class Myserver(socketserver.BaseRequestHandler):
    def handle(self):
        conn = self.request



# sk=socket.socket()
# sk.bind(('127.0.0.1',7001))
# sk.listen()

# conn,addr = sk.accept()

        res = conn.recv(1024).decode('utf-8') #接收用户名,密码的字符串
        recv_dic = json.loads(res) #转换

        if recv_dic['username']  in dic_key and recv_dic['password'] == dic_key[recv_dic['username']]:
            print(f'{recv_dic["username"]}登录成功')
            conn.send(b'0') #给客户端发送回执

            new_content = conn.recv(1024).decode('utf-8')
            if new_content == '1':
                lens =conn.recv(4)
                length =struct.unpack('i',lens)[0]
                f_fun =conn.recv(length).decode('utf-8')
                file_dic = json.loads(f_fun)
                with open(file_dic['filename'],mode='wb') as f:
                    if file_dic['filesize'] > 0:
                        content =conn.recv(1024)
                        file_dic['filesize'] -= len(content)
                        f.write(content)
                conn.close()

            elif new_content == '2':
                seng_filelist=json.dumps(dowload_file).encode('utf-8')
                conn.send(seng_filelist)
                down_file = conn.recv(1024).decode('utf-8')  #接收要下载的文件名
                down_path = os.path.join(dowload_path,down_file) #下载路径拼接  取得文件
                down_size = os.path.getsize(down_path) #取得文件大小
                # dic = {'filename':down_path,'filesize':down_size}
                # new_dic = json.dumps(dic)
                # conn.send(new_dic.encode('utf-8'))
                conn.send(str(down_size).encode('utf-8'))  #这样可以不用json转成字符串
                with open(down_path,mode='rb') as f:
                    while down_size >0:
                        con = f.read(1024)
                        down_size -= len(con)
                        conn.send(con)
            else:
                print('登录失败')



server = socketserver.ThreadingTCPServer(('127.0.0.1',7002),Myserver)
server.serve_forever()

client 端

# _*_ coding : UTF-8 _*_

import socket
import json
import  struct
import os
num = ['上传','下载']

sk= socket.socket()
sk.connect(('127.0.0.1',7002))


username = input('请输入用户名:').strip()
password = input('请输入密码:').strip()

dic = {'username':username,'password':password} #用户名,密码

srt_dic = json.dumps(dic) #转成字符串
sk.send(srt_dic.encode('utf-8'))  #发送

login_mag = sk.recv(1024).decode('utf-8')
if login_mag == '0':
    print('登录成功')
    for i,v in enumerate(num,1):
        print(i,v)

    no = input('请输入功能序号').strip()
    if no == '1': #上传
        sk.send(b'1')
        abs_path = r'E:msdia80.dll'
        filename = os.path.basename(abs_path) #获得文件名
        filesize =  os.path.getsize(abs_path)  #文件大小
        dic = {'filename':filename,'filesize':filesize}
        new_dic=json.dumps(dic).encode('utf-8')
        s_dic = struct.pack('i',len(new_dic))
        sk.send(s_dic)
        sk.send(new_dic)

        with open(abs_path,mode='rb') as f:
            if filesize>0:
                content=f.read(1024)
                filesize -= len(content)
                sk.send(content)
        sk.close()

    elif no == '2':#下载
        sk.send(b'2')
        ret = sk.recv(1024).decode('utf-8')
        down_list = json.loads(ret)
        while True:
            for v ,i in enumerate(down_list,1):
                print(v,i)
            no = input('请输入序号选择下载的文件:').strip()
            if no.isdecimal() and 0 < int(no) <= len(down_list):
               down_file = down_list[int(no)-1].encode('utf-8')  #选择的文件名
               sk.send(down_file)
               down_size = sk.recv(1024).decode('utf-8') #接收下载文件的大小
               size = int(down_size) #服务端用str转成字符串,客户端用int转成大小
               with open(down_file,mode='wb') as f:
                   while size > 0:
                       con = sk.recv(1024)
                       size -= len(con)
                       f.write(con)
               sk.close()
               break

            else:
                print('选择的序号有误,请重新输入')
                continue