1
# -*-coding: Utf-8 -*-
2
# @File : login .py
3
# author: Chimengmeng
4
# blog_url : https://www.cnblogs.com/dream-ze/
5
# Time：2023/6/3
6

7
import os
8
import hashlib
9
from verify_code import get_verify_code
10

11
# 声明数据库位置
12
file_path = 'Infos' + '\\' + 'user_pwd.txt'
13
if not os.path.exists(file_path):
14
    with open(file_path, 'a') as f:
15
        f.write('')
16

17

18
def encrypt_decrypt(data):
19
    # 转为二进制数据
20
    data = data.encode('utf-8')
21
    # 创建md5对象
22
    md5 = hashlib.md5()
23
    # md5进行加密
24
    md5.update(data)
25
    # 取出md5加密后的哈希值
26
    encrypt_result = md5.hexdigest()
27
    return encrypt_result
28

29

30
def write_read_data(data=None, cmd=0):
31
    if cmd == 0:
32
        with open(file_path, 'a+') as f:
33
            f.write(data)
34
    else:
35
        user_list = []
36
        user_info_data = []
37
        with open(file_path, 'r') as f:
38
            for line in f:
39
                user_data = {}
40
                line = line.strip().split('|')
41
                username, password, salt_code = line[0], line[1], line[2]
42
                user_data['username'] = username
43
                user_data['password'] = password
44
                user_data['salt_code'] = salt_code
45
                user_list.append(username)
46
                user_info_data.append(user_data)
47
        return [user_list, user_info_data]
48

49

50
def register(username, password):
51
    # 获得六位数的盐
52
    salt_code = get_verify_code(6)
53
    # 原始密码加盐
54
    password_str = password + salt_code
55
    # 加盐密码加密
56
    password_encrypted = encrypt_decrypt(password_str)
57
    # 拼接存储数据格式
58
    user_pwd_data = f'{username}|{password_encrypted}|{salt_code}\n'
59
    # 写入文件存储数据
60
    write_read_data(user_pwd_data, cmd=0)
61
    print(f'{username}注册成功,注册结束!')
62

63

64
def login():
65
    # 拿到用户名列表,用户名和密码及加盐后的列表
66
    user_list, user_data = write_read_data(data=None, cmd=1)
67
    username_input = input('校验Username:>>>')
68
    password_input = input('校验Password:>>>')
69
    # 判断用户名是否存在于用户名列表中
70
    # 存在则继续登录
71
    if username_input in user_list:
72
        # 循环所有用户名及信息
73
        for info in user_data:
74
            # 取用户名和加密后的密码
75
            username = info['username']
76
            password = info['password']
77
            # 取加盐后的密码
78
            salt_code = info['salt_code']
79
            # 当前密码加盐
80
            password_str = password_input + salt_code
81
            # 当前加盐密码加密
82
            password_encrypted = encrypt_decrypt(password_str)
83
            if username == username_input and password == password_encrypted:
84
                print('登陆成功!')
85
                return True
86
            else:
87
                print('用户名或密码错误,登陆失败!')
88
                main()
89
    else:
90
        print('用户名不存在,请注册')
91
        main()
92

93

94
def main():
95
    # 先校验用户名和密码是否存在
96
    username = input('Username:>>>')
97
    # 获取用户列表
98
    user_list = write_read_data(cmd=1)[0]
99
    # 不存在用户信息则进行注册
100
    if username not in user_list:
101
        print('当前用户未注册注册,注册操作开始!')
102
        # 注册函数
103
        password = input('Password:>>>')
104
        register(username, password)
105
        # 注册完成进行二次验证校验登陆
106
        main()
107
    else:
108
        password = input('Password:>>>')
109
        # 用户存在进行登陆校验
110
        print('进行登陆操作')
111
        # 拿到成功的结果
112
        res = login()
113
        # 成功则退出
114
        if res:
115
            print('欢迎使用')
116
            pass
117
        else:
118
            # 不成功二次校验
119
            login()
120

121

122
if __name__ == '__main__':
123
    main()

1
# -*-coding: Utf-8 -*-
2
# @File : verify_code .py
3
# author: Chimengmeng
4
# blog_url : https://www.cnblogs.com/dream-ze/
5
# Time：2023/6/3
6

7
import random
8

9
'''生成六位随机 （数字 + 大小写） 验证码'''
10

11

12
def get_verify_code(n):
13
    code = ''
14
    for i in range(n):
15
        random_int = str(random.randint(0, 9))  # 0-9之间的整数
16
        random_upper = chr(random.randint(65, 90))  # A-Z之间的字母
17
        random_lower = chr(random.randint(97, 122))  # a-z之间的字母
18
        temp = random.choice([random_int, random_upper, random_lower])
19
        code += temp
20
    return code
21

22

23
if __name__ == "__main__":
24
    res = get_verify_code(6)
25
    print(res)