-
Notifications
You must be signed in to change notification settings - Fork 0
/
user_management.py
74 lines (66 loc) · 2.33 KB
/
user_management.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
from login import read_passwords
def read_users():
users = []
try:
with open("users.txt", "r") as file:
for line in file:
username, role = line.strip().split(",")
users.append({'username': username, 'role': role})
except FileNotFoundError:
print("Error: Users file not found.")
return users
def write_users(users):
try:
with open("users.txt", "w") as file:
for user in users:
file.write(f"{user['username']},{user['role']}\n")
print("User records updated successfully.")
except Exception as e:
print(f"Error: {e}")
def write_passwords(passwords):
with open("passwords.txt", "w") as file:
for item in passwords:
file.write(f"{item['username']},{item['password']}\n")
print("Passwords updated successfully.")
def register_user():
users = read_users()
passwords = read_passwords()
username = input("Enter username for new user: ")
role = 'student'
password = input('Enter password: ')
users.append({'username':username, 'role':role})
passwords.append({'username': username, 'password': password})
write_users(users)
write_passwords(passwords)
def modify_user():
users = read_users()
username = input("Enter username of user to modify: ")
for user in users:
if user['username'] == username:
new_role = input("Enter new role for the user (admin/student): ")
user['role'] = new_role
write_users(users)
print(f"Role for user '{username}' updated successfully.")
return
print("User not found.")
def delete_user():
users = read_users()
passwords = read_passwords()
username = input("Enter username of user to delete: ")
# Remove user from the users list
for user in users:
if user['username'] == username:
users.remove(user)
break
else:
print("User not found.")
return
# Remove password from the passwords dictionary
for user in passwords:
if user['username'] == username:
passwords.remove(user)
write_passwords(passwords)
write_users(users)
print(f"User '{username}' deleted successfully.")
else:
print("Password not found.")