dict_client.py

from socket import *
import sys 
import getpass 

#创建网路链接
def main():
    if len(sys.argv) < 3:
        print("argv is error")
        return 
    HOST = sys.argv[1]
    PORT = int(sys.argv[2])
    s = socket()
    try:
        s.connect((HOST,PORT))
    except Exception as e:
        print(e)
        return 
    
    while True:
        print('''
        ============Welcome==========
        --1.注册    2.登录     3.退出--
        =============================
        ''')

        try:
            cmd = int(input("输入选项:"))
        except Exception as e:
            print("命令错误!")
            continue 
        if cmd not in [1,2,3]:
            print("请输入正确选项")
            continue 
        elif cmd == 1:
            do_register(s)
        elif cmd == 2:
            do_login(s)
        elif cmd == 3:
            s.send(b'E')
            sys.exit("谢谢使用")

def do_register(s):
    while True:
        name = input("User:")
        passwd = getpass.getpass()
        passwd1 = getpass.getpass("Again:")

        if (' ' in name) or (' ' in passwd):
            print("用户名密码不许有空格")
            continue 
        if passwd != passwd1:
            print("两次密码不一致")
            continue 
        
        msg = "R %s %s"%(name,passwd)
        #发送请求
        s.send(msg.encode())
        #等待回复
        data = s.recv(128).decode()
        if data == 'OK':
            print("注册成功")
        elif data == 'EXISTS':
            print("用户已存在")
        else:
            print("注册失败")
        return

def do_login(s):
    name = input("User:")
    passwd = getpass.getpass()
    msg = "L %s %s"%(name,passwd)
    s.send(msg.encode())
    data = s.recv(128).decode()
    if data == 'OK':
        print("登录成功")
    else:
        print("登录失败")

if __name__ == "__main__":
    main()