def remove(user, database_name):
action = 'remove_user'
removed = False
t = time.time()
timestamp = datetime.datetime.fromtimestamp(t).strftime('%Y-%m-%d %H:%M:%S')
if add_user.findUser(user, database_name):
shutil.rmtree("databases/" + database_name + '/' + user)
removed = True
f = open('log.txt', 'a')
f.write(timestamp + '\n')
f.write(action + '\n')
f.write('user: '******'\n')
f.write('database name: ' + database_name + '\n' + '\n')
f.close()
return removed
def removeUserMenu(self):
# check if database selected
dbname = str(self.dbs_combo.currentText())
if dbname == '/':
QtGui.QMessageBox.warning(self, 'Error', 'Please choose a database.')
return
# check if user selected
username = str(self.users_combo.currentText())
if username == '/':
QtGui.QMessageBox.warning(self, 'Error', 'Please choose a user to remove.')
return
if add_user.findUser(username, dbname):
reply = QtGui.QMessageBox.question(self,"Are you sure?", "Are you sure you want to remove " + username + '?', QtGui.QMessageBox.Yes, QtGui.QMessageBox.No)
if reply == QtGui.QMessageBox.Yes:
success = remove_user.remove(username, dbname)
if success == True:
QtGui.QMessageBox.information(self, 'Success', 'User ' + username + " removed.")
else:
QtGui.QMessageBox.warning(self, 'Error', 'User not found.')
else:
QtGui.QMessageBox.warning(self, 'Error', 'User not found.')
