Add image restore action

This action provides the image restoring functionality on a specific
client.

1 files changed, 8 insertions, 0 deletions

diff --git a/ogcp/forms/action_forms.py b/ogcp/forms/action_forms.py
index 4dfaa21..eb8d309 100644
--- a/ogcp/forms/action_forms.py
+++ b/ogcp/forms/action_forms.py
@@ -38,6 +38,14 @@ class SessionForm(FlaskForm):
     os = RadioField(label=_('Session'), choices=[])
     run = SubmitField(label=_('Run'))
 
+class ImageRestoreForm(FlaskForm):
+    ips = HiddenField()
+    partition = SelectField(label=_('Partition'), choices=[])
+    image = SelectField(label=_('Image'), choices=[])
+    method = SelectField(label=_('Method'),
+                         choices=[('UNICAST', 'Unicast')])
+    restore = SubmitField(label=_('Restore'))
+
 class ClientDetailsForm(FlaskForm):
     name = StringField(label=_('Name'))
     ip = StringField(label=_('IP'))