Now import/export Class.id not 'class_id'

hitier
1 parent 142e2e9d
Showing 3 changed files with 8 additions and 6 deletions Show diff stats
app/main/routes.py
app/main/templates/agent_form.html
app/models.py
@@ -143,7 +143,10 @@ def agent_edit(agent_id=None):
     # Or submit for db writing
     #
     elif request.method == 'POST':
-        agent_id = request.form['agent_id']
+        try:
+            agent_id = request.form['agent_id']
+        except KeyError:
+            agent_id = None
         if agent_id:
             # then update existing
             this_agent = Agent.query.get(int(agent_id))
 {% extends "base_page.html" %}
 {#  Set the title that will be used in base_page #}
-{% if agent['agent_id'] != '' %}
+{% if agent and agent['id'] not in ['', None] %}
   {% set subtitle = "Modifier l'agent "+ agent['fullname']  %}
 {% else %}
   {% set subtitle = "Ajouter un nouvel agent" %}
@@ -13,8 +13,8 @@
   <span id="nav_actived" style="display: none">cds,agent/create</span>
   <form id="agent_form" class="pdc-form" action="{{ url_for('main.agent_edit') }}" method="post">
-    {% if agent %}
-      <input class="form-control" id="agent_id" name="agent_id" type="hidden" value="{{ agent['agent_id'] }}">
+    {% if agent and agent['id'] not in ['', None] %}
+      <input class="form-control" id="agent_id" name="agent_id" type="hidden" value="{{ agent['id'] }}">
     {% endif %}
     <div class="form-group">
       <label for="firstname">Nom</label>
@@ -31,7 +31,7 @@ class Formable:
         :return:  nothing
         """
-        _struct = {}
+        _struct = {'id': self.id}
         for key in self.export_keys:
             _value = getattr(self, key)
             _struct[key] = '' if _value is None else _value
@@ -135,7 +135,6 @@ class Agent(db.Model, Formable):
         """
         _struct = super(Agent, self).to_struct()
         _struct['fullname'] = self.fullname
-        _struct['agent_id'] = self.id if self.id else ''
         return _struct