From ea27d05c587fc8112ce52d7ded5b26ecb17ddd2e Mon Sep 17 00:00:00 2001
From: liuzheng712 <liuzheng712@gmail.com>
Date: Mon, 26 Feb 2018 18:05:41 +0800
Subject: [PATCH] feat: update app setting

---
 apps/terminal/forms.py                               | 12 +++++++++++-
 .../templates/terminal/terminal_modal_accept.html    |  1 +
 .../terminal/templates/terminal/terminal_update.html |  1 +
 3 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/apps/terminal/forms.py b/apps/terminal/forms.py
index 17526193e..a3672eeef 100644
--- a/apps/terminal/forms.py
+++ b/apps/terminal/forms.py
@@ -16,13 +16,23 @@ def get_all_command_storage():
         yield (k, k)
 
 
+def get_all_replay_storage():
+    # storage_choices = []
+    from common.models import Setting
+    Setting.refresh_all_settings()
+    for k, v in settings.TERMINAL_REPLAY_STORAGE.items():
+        yield (k, k)
+
+
 class TerminalForm(forms.ModelForm):
     command_storage = forms.ChoiceField(choices=get_all_command_storage(),
                                         label=_("Command storage"))
+    replay_storage = forms.ChoiceField(choices=get_all_replay_storage(),
+                                       label=_("Replay storage"))
 
     class Meta:
         model = Terminal
-        fields = ['name', 'remote_addr', 'ssh_port', 'http_port', 'comment', 'command_storage']
+        fields = ['name', 'remote_addr', 'ssh_port', 'http_port', 'comment', 'command_storage', 'replay_storage']
         help_texts = {
             'ssh_port': _("Coco ssh listen port"),
             'http_port': _("Coco http/ws listen port"),
diff --git a/apps/terminal/templates/terminal/terminal_modal_accept.html b/apps/terminal/templates/terminal/terminal_modal_accept.html
index fe70bb342..e846a63e1 100644
--- a/apps/terminal/templates/terminal/terminal_modal_accept.html
+++ b/apps/terminal/templates/terminal/terminal_modal_accept.html
@@ -13,6 +13,7 @@
     {% bootstrap_field form.ssh_port layout="horizontal" %}
     {% bootstrap_field form.http_port layout="horizontal" %}
     {% bootstrap_field form.command_storage layout="horizontal" %}
+    {% bootstrap_field form.replay_storage layout="horizontal" %}
     {% bootstrap_field form.comment layout="horizontal" %}
 </form>
 
diff --git a/apps/terminal/templates/terminal/terminal_update.html b/apps/terminal/templates/terminal/terminal_update.html
index cbf745608..a40e3c362 100644
--- a/apps/terminal/templates/terminal/terminal_update.html
+++ b/apps/terminal/templates/terminal/terminal_update.html
@@ -36,6 +36,7 @@
                             {% bootstrap_field form.ssh_port layout="horizontal" %}
                             {% bootstrap_field form.http_port layout="horizontal" %}
                             {% bootstrap_field form.command_storage layout="horizontal" %}
+                            {% bootstrap_field form.replay_storage layout="horizontal" %}
 
                             <div class="hr-line-dashed"></div>
                             <h3>{% trans 'Other' %}</h3>