socket网络编程

表示编程这条路真得好难走，后悔大学没有读信息专业啊。

第七周的作业：

1. 用户登陆
2. 上传/下载文件
3. 不同用户家目录不同
4. 查看当前目录下文件

刚开始没有一点思路，上网查了很多资料，还自学了FTP。吐槽一句，讲师真是太不负责了，作业要求不详细，还没有思路。

要实现通信，需要7个层次：物理层，数据链路（mac地址），网络层(IP)，传输层（有标准的数据类型， TCP/IP 三次握手四次断开， UDP），会话层（http,smtp），表示层，应用层。 传输过程中就需要做到的事情是接收和发送，在python中，这部分功能被封装在socket模块的socket类中。

要实现文件上传和下载，肯定是需要两个端口：一个是服务器端(server),另外一个是客户端（client）。大致的交互流程如下：

理清楚思路之后，就是开始写代码了。

server端的代码如下：

import socket
import os
import sys
sys.path.append('..')
from conf import setting
com_path = setting.DATABASE

def run():
    server = socket.socket()
    server.bind(('localhost',6969))
    server.listen()

    while True:
        print('等待客户端的连入')
        conn, addr = server.accept()
        print('收到来自%s,%s的请求'%(str(conn),str(addr)))
        data = conn.recv(1024).decode()
        server_path = '%s\%s\%s'%(com_path['path'],com_path['server'],data)
        print('当前目录下的文件有：')
        print(os.listdir(server_path))
        #检查服务器端是否已经有目录存在，要是没有的化，创建一个目录
        if os.path.isdir(server_path):
            print('该用户的目录已经存在')
        else:
            os.mkdir(server_path)
            print('成功在远程端创建%s的目录'%data)
        while True:
            try:
                data = conn.recv(1024)
                if len(data) == 0:
                    continue
                else:
                    cmd, filename = data.decode().split()
                    file_path = "%s\%s.txt"%(server_path,filename)
                    print(cmd,filename)
                if cmd == 'download':
                    print(file_path)
                    if os.path.isfile(file_path):
                        with open(file_path,'rb') as f:
                            file_size = os.stat(file_path).st_size
                            conn.send(bytes(str(file_size),encoding='utf-8'))
                            print('已经成功传送了文件大小%s'%str(file_size))
                            conn.recv(1024) #服务器端发送文件大小后，接受客户端的相应
                            for line in f:
                                conn.send(bytes(str(line),encoding='utf-8'))
                elif cmd == 'upload':
                    receive_file_size = conn.recv(1024).decode()
                    conn.send('准备好接收文件'.encode())
                    receive_size = 0
                    file_total_size = int(receive_file_size)
                    with open(file_path, 'wb') as f:
                        while receive_size < file_total_size:
                            data = conn.recv(1024)
                            receive_size += len(data)
                            f.write(data)
                        print('file recv done')
                        print(receive_size,file_total_size)
                print('更新后目录下的文件有：')
                print(os.listdir(server_path))
            except Exception:
                print('客户端链接断开')
                break

    server.close()

　　

client端的代码如下：

import socket
import sys
sys.path.append('..')
import os
from conf import setting
com_path = setting.DATABASE
current_user = None

def auth(func):
    def wrapper(*args,**kwargs):
        global current_user
        username, pwd = input('请输入您的用户名，密码：').strip().split(',')
        user_info_path = '%s\%s.txt'%(com_path['path'],com_path['user_info'])
        exist_flag = False
        not_user = False
        with open(user_info_path,'r') as f:
            while not exist_flag:
                for line in f:
                    if line.startswith(username):
                        if line.strip().split('|')[1] == pwd:
                            print('登录成功')
                            current_user = username
                            func()
                            break
                            exist_flag = True
                            not_user = True
                        else:
                            print('密码错误')
            if not not_user:
                print('用户不存在')
    return wrapper

@auth
def run():
    client = socket.socket()
    client.connect(('localhost',6969))
    client_path = '%s\%s\%s'%(com_path['path'],com_path['client'],current_user)

    client.send(current_user.encode())#将用户名发送给服务器端，看看是否有这个客户的目录

    while True:
        inp = input(">>:").strip()
        if len(inp) == 0:
            continue
        try:
            cmd,filename = inp.split()
            file_path = "%s\%s.txt"%(client_path,filename)
        except:
            print('输入有误，请重新输入')
            continue
        try:
            if cmd == 'download':
                client.send(inp.encode())
                receive_file_size = client.recv(1024).decode()
                print('server file size',receive_file_size)
                client.send('准备好接收文件'.encode())
                if os.path.isdir(client_path):
                    print('该用户的目录已经存在')
                else:
                    os.mkdir(client_path)
                    print('成功在远程端创建%s的目录'%data)
                receive_size = 0
                file_total_size = int(receive_file_size)
                with open(file_path,'wb')as f:
                    data = client.recv(1024)
                    receive_size += len(data)
                    f.write(data)
                    print('file recv done')
                    print(receive_size,file_total_size)

            elif cmd == 'upload':
                # try:
                if os.path.isfile(file_path):
                    client.send(bytes(inp,encoding='utf-8'))
                    with open(file_path,'rb') as f:
                        file_size = os.stat(file_path).st_size
                        print(file_size)
                        client.send(bytes(str(file_size),encoding='utf-8'))
                        print('已经成功传送了文件大小%s'%str(file_size))
                        client.recv(1024)
                        for line in f:
                            client.send(bytes(str(line), encoding='utf-8'))
                # except Exception:
                #     print('只能上传文件')
        except Exception:
            break
        client.close()

run()

本来想在写一个自己的类，然后继承socket类的，但是无奈，老是出现‘OSError: [WinError 10022] 提供了一个无效的参数’错误，这个BUG我也修改不了。所以就放弃了运用面向对象的方法。希望学到以后能回过头来找出今天没有能力解决的原因吧。