New agent edit form

hitier
1 parent 7bca61b9
Showing 4 changed files with 114 additions and 46 deletions Show diff stats
app/main/routes.py
app/main/templates/agent.html
app/main/templates/agent_edit.html
app/models.py
-
-from flask import render_template, make_response, current_app, redirect, url_for, request
+from flask import render_template, make_response, current_app, redirect, url_for, request, flash
 from flask_login import login_required, current_user
  
 from . import bp
  
-from app.models import Agent, Project, Service, Capacity, Period
+from app.models import Agent, Project, Service, Capacity, Period, db
 from app import db_mgr
 from app.auth.routes import role_required
  
@@ -112,7 +111,42 @@ def agent(agent_id):
     return render_template('agent.html',
                            agent=this_agent,
                            charges=charges_table,
-                           subtitle="{} {}".format(this_agent.firstname, this_agent.secondname))
+                           subtitle=f"{this_agent.fullname}")
+
+
+@bp.route('/agent/create', methods=('POST', 'GET'))
+@bp.route('/agent/<agent_id>/edit', methods=('POST', 'GET'))
+@role_required('service')
+def agent_edit(agent_id=None):
+    # Make the form, filled with existing agent if updating
+    #
+    if request.method == 'GET':
+        if agent_id:
+            this_agent = Agent.query.get(int(agent_id))
+        else:
+            this_agent = Agent()
+        # export to structure for jinja display
+        agent_struct = this_agent.to_struct()
+        return render_template('agent_edit.html', agent=agent_struct)
+    # Or submit for db writing
+    #
+    elif request.method == 'POST':
+        agent_id = request.form['agent_id']
+        if agent_id:
+            # then update existing
+            this_agent = Agent.query.get(int(agent_id))
+            done_string = "mis à jour."
+        else:
+            # or create from scratch
+            this_agent = Agent()
+            done_string = "ajouté."
+        # fill orm with form and write to db
+        this_agent.from_request(request)
+        db.session.add(this_agent)
+        db.session.commit()
+        # we're done
+        flash(f"Agent {this_agent.fullname} "+done_string)
+        return redirect(url_for('main.agent', agent_id=this_agent.id))
  
  
 # - - - - - - - - - - - - - - - - - - - -  REST API - - - - - - - - - - - - - - - - - - - -
@@ -141,4 +175,3 @@ def charge_agent_csv(agent_id):
     resp = make_response("\n".join(csv_table))
     resp.headers['Content-Type'] = 'text/plain;charset=utf8'
     return resp
-
@@ -64,8 +64,7 @@
       <dt class="col-sm-2 text-right">Besoin en formation :</dt>
       <dd class="col-sm-10 text-left"></dd>
     </dl>
-
-    <a class="card-link disabled" href="#">Modifier</a>
+    <a class="card-link" href="{{url_for('main.agent_edit', agent_id=agent.id)}}">Modifier</a>
   </div>
 </div>
  
@@ -103,7 +102,7 @@
 <script>
   build_chart("#projects_chart",
       "{{url_for('main.charge_agent_csv', agent_id=agent.id)}}",
-      "{{agent.secondname}}" + " {{agent.firstname}}",
+      "{{agent.fullname}}",
       "project");
 </script>
 {% endblock %}
 {% extends "base_page.html" %}
-{% block content %}
-{% if agent %}
-Modifier l'agent {{ agent.name}}.
+{#  Set the title that will be used in base_page #}
+{% if agent['agent_id'] != '' %}
+  {% set subtitle = "Modifier l'agent"+ agent['fullname'] +":" %}
 {% else %}
-Ajouter un nouvel agent.
+  {% set subtitle = "Ajouter un nouvel agent:" %}
 {% endif %}
-<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.id}}">
-  {% endif %}
-  <div class="form-group">
-    <label for="firstname">Nom</label>
-    <input class="form-control" id="firstname" name="firstname" type="text" value="{{agent.firstname}}">
-  </div>
-  <div class="form-group">
-    <label for="secondname">Prénom</label>
-    <input class="form-control" id="secondname" name="secondname" type="text" value="{{agent.secondname}}">
-  </div>
-  <div class="form-group">
-    <label for="virtual">Virtuel</label>
-    <input class="form-control" id="virtual" name="virtual" type="text" value="{{agent.virtual}}">
-  </div>
-  <div class="form-group">
-    <label for="company">Structure</label>
-    <input class="form-control" id="company" name="company" type="text" value="{{agent.company.name}}">
-  </div>
-  <div class="form-group">
-    <label for="grade">Grade</label>
-    <input class="form-control" id="grade" name="grade" type="text" value="{{agent.grade.name}}">
-  </div>
-  <div class="form-group">
-    <label for="status">Statut</label>
-    <input class="form-control" id="status" name="status" type="text" value="{{agent.status.name}}">
-  </div>
-  <div class="form-group">
-    <label for="bap">Bap</label>
-    <input class="form-control" id="bap" name="bap" type="text" value="{{agent.bap.name}}">
-  </div>
-  <input class="btn btn-dark" type="submit" value="Valider">
-</form>
+{% block content %}
+  <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'] }}">
+    {% endif %}
+    <div class="form-group">
+      <label for="firstname">Nom</label>
+      <input class="form-control" id="firstname" name="firstname" type="text" value="{{ agent['firstname'] }}">
+    </div>
+    <div class="form-group">
+      <label for="secondname">Prénom</label>
+      <input class="form-control" id="secondname" name="secondname" type="text" value="{{ agent['secondname'] }}">
+    </div>
+    <div class="form-group">
+      <label for="virtual">Virtuel</label>
+      <input class="form-control" id="virtual" name="virtual" type="text" value="{{ agent['virtual'] }}">
+    </div>
+    <div class="form-group">
+      <label for="company">Structure</label>
+      <input class="form-control" id="company_id" name="company_id" type="text" value="{{ agent['company_id'] }}">
+    </div>
+    <div class="form-group">
+      <label for="grade">Grade</label>
+      <input class="form-control" id="grade_id" name="grade_id" type="text" value="{{ agent['grade_id'] }}">
+    </div>
+    <div class="form-group">
+      <label for="status">Statut</label>
+      <input class="form-control" id="status_id" name="status_id" type="text" value="{{ agent['status_id'] }}">
+    </div>
+    <div class="form-group">
+      <label for="bap">Bap</label>
+      <input class="form-control" id="bap_id" name="bap_id" type="text" value="{{ agent['bap_id'] }}">
+    </div>
+    <div class="form-group">
+      <label for="permanent">Permanent</label>
+      <input class="form-control" id="permanent" name="permanent" type="text" value="{{ agent['permanent'] }}">
+    </div>
+    <input class="btn btn-dark" type="submit" value="Valider">
+  </form>
 {% endblock %}
@@ -82,6 +82,37 @@ class Agent(db.Model):
     status = relationship("AgentStatus", back_populates="agents")
     company = relationship("Company", back_populates="agents")
  
+    @property
+    def export_keys(self):
+        return ['firstname', 'secondname', 'virtual', 'permanent', 'company_id', 'status_id', 'grade_id', 'bap_id']
+
+    @property
+    def fullname(self):
+        return f"{self.secondname} {self.firstname}"
+
+    def to_struct(self):
+        """
+        Export the orm object to a structure easily used in jinja
+
+        :return:  nothing
+        """
+        _struct = {'agent_id': self.id if self.id else '',
+                   'fullname': self.fullname}
+        for key in self.export_keys:
+            _value = getattr(self, key)
+            _struct[key] = '' if _value is None else _value
+        return _struct
+
+    def from_request(self, form_request):
+        """
+        Get a form request structure and fill in our fields
+
+        :param form_request:
+        :return:
+        """
+        for key in self.export_keys:
+            setattr(self, key, form_request.form[key])
+
  
 class Service(db.Model):
     id = db.Column(db.Integer, primary_key=True)