Add ability to restore partial snapshots

Closes #5742