diff --git a/src/view/SnapshotsDialog.cpp b/src/view/SnapshotsDialog.cpp index 31a1aec..995b93a 100644 --- a/src/view/SnapshotsDialog.cpp +++ b/src/view/SnapshotsDialog.cpp @@ -152,6 +152,8 @@ void SnapshotsDialog::OnBackTo() QString filename, defaultFilename ; QListWidgetItem * item = _snapshots->currentItem(); QString user; + std::list users; + std::list::iterator it; if (!item) return; @@ -180,11 +182,26 @@ void SnapshotsDialog::OnBackTo() user = _kiss->GetUser()->_name; _kiss->ChangeDatabase(Database::GetDatabaseHome() + defaultFilename); - _kiss->LoadUser(user); - QMessageBox::information(0, "KissCount", _("Welcome back to ") + filename); + users = _kiss->GetUsers(); + + for(it=users.begin(); it!=users.end(); it++) + { + if (*it == user) + { + _kiss->LoadUser(user); + QMessageBox::information(0, "KissCount", _("Welcome back to ") + filename); + } + } close(); + + if (it == users.end()) + { + _wxUI->KillMe(); // Clear all panels + QMessageBox::information(0, "KissCount", _("Welcome back to ") + filename); + _wxUI->ChangeUser(); + } } void SnapshotsDialog::OnOK()