Skip to content

Commit 84e1bae

Browse files
committed
Add ability to clear database internally
1 parent 913aea8 commit 84e1bae

File tree

1 file changed

+11
-6
lines changed

1 file changed

+11
-6
lines changed

usrcheck.py

+11-6
Original file line numberDiff line numberDiff line change
@@ -26,11 +26,14 @@ def save_users(usrlist,overWrite):
2626
saveFile = open("usrlist", "w")
2727
else:
2828
saveFile = open("usrlist", "a")
29-
for x in range(len(usrlist)):
30-
if (x-2)<len(usrlist):
31-
saveFile.write(str(usrlist[x][0])+","+str(usrlist[x][1])+",")
32-
else:
33-
saveFile.write(str(usrlist[x][0])+","+str(usrlist[x][1]))
29+
if len(usrlist)>0:
30+
for x in range(len(usrlist)):
31+
if (x-2)<len(usrlist):
32+
saveFile.write(str(usrlist[x][0])+","+str(usrlist[x][1])+",")
33+
else:
34+
saveFile.write(str(usrlist[x][0])+","+str(usrlist[x][1]))
35+
else:
36+
saveFile.write('')
3437
saveFile.close()
3538

3639
def usr_check(usrn,pswd,usrlist):
@@ -78,7 +81,7 @@ def login_init(usrname,pswd,overWrite,quiet):
7881
else:
7982
newFile=False
8083
while True:
81-
nUsr=input("Enter the number of users to initialize: ")
84+
nUsr=input("Enter the number of users to initialize ('0' Clears the database): ")
8285
try:
8386
nUsr=int(nUsr)
8487
break
@@ -87,6 +90,8 @@ def login_init(usrname,pswd,overWrite,quiet):
8790
if nUsr>0:
8891
usrlist=init_usrs(nUsr)
8992
save_users(usrlist,newFile)
93+
elif nUsr==0:
94+
save_users([],True)
9095
else:
9196
usrlist = load_users()
9297
Ufound,Pfound=usr_check(usrname,pswd,usrlist)

0 commit comments

Comments
 (0)