Add an (unsecured!) admin backend.

Antoine Goutenoir
1 parent 0daf0dc8
Showing 6 changed files with 48 additions and 9 deletions Show diff stats
flaskr/__init__.py
flaskr/controllers/admin_controller.py
flaskr/controllers/main_controller.py
flaskr/extensions.py
flaskr/models.py
requirements.txt
@@ -4,11 +4,13 @@ from flask import Flask
 from flask.cli import ScriptInfo
 from webassets.loaders import PythonLoader as PythonAssetsLoader
  
+
 from flaskr import assets
-from flaskr.models import db
+from flaskr.models import db, Estimation, EstimationView
 from flaskr.controllers.main_controller import main
  
 from flaskr.extensions import (
+    admin,
     cache,
     assets_env,
     debug_toolbar,
@@ -39,16 +41,13 @@ def create_app(object_name):
     # Load configuration
     app.config.from_object(object_name)
  
-    # initialize the cache
+    # Initialize
     cache.init_app(app)
-
-    # initialize the debug tool bar
     debug_toolbar.init_app(app)
-
-    # initialize SQLAlchemy
     db.init_app(app)
-
     login_manager.init_app(app)
+    admin.init_app(app)
+    admin.add_view(EstimationView(Estimation, db.session))
  
     # Import and register the different asset bundles
     assets_env.init_app(app)
@@ -0,0 +1,11 @@
+from flask import Blueprint, render_template, flash, request, redirect, url_for
+from flask_login import login_user, logout_user, login_required
+
+from flaskr.forms import LoginForm
+from flaskr.models import db, User
+
+from flaskr.controllers.main_controller import main
+
+
+# from flask.ext.admin import Admin
+# from flask.ext.admin.contrib.sqla import ModelView
@@ -50,6 +50,7 @@ def estimate():
         # estimation.email = form.email.data
         estimation.first_name = form.first_name.data
         estimation.last_name = form.last_name.data
+        estimation.institution = form.institution.data
         estimation.status = StatusEnum.pending
         estimation.origin_addresses = form.origin_addresses.data
         estimation.destination_addresses = form.destination_addresses.data
+from flask_admin import Admin
 from flask_caching import Cache
 from flask_debugtoolbar import DebugToolbarExtension
 from flask_login import LoginManager
@@ -17,6 +18,8 @@ login_manager = LoginManager()
 login_manager.login_view = "main.login"
 login_manager.login_message_category = "warning"
  
+admin = Admin()
+
  
 @login_manager.user_loader
 def load_user(userid):
+from flask_admin.contrib.sqla import ModelView
+
 from flaskr.core import generate_unique_id
 from flask_sqlalchemy import SQLAlchemy
 from flask_login import UserMixin, AnonymousUserMixin
@@ -32,12 +34,13 @@ class Estimation(db.Model):
     email = db.Column(db.Unicode(1024))
     first_name = db.Column(db.Unicode(1024))  # Antoine
     last_name = db.Column(db.Unicode(1024))   # Goutenoir
+    institution = db.Column(db.Unicode(1024))   # IRAP
     status = db.Column(db.Enum(StatusEnum), default=StatusEnum.pending)
  
     # City, Country
     # One address per line
-    origin_addresses = db.Column(db.Unicode())
-    destination_addresses = db.Column(db.Unicode())
+    origin_addresses = db.Column(db.UnicodeText())
+    destination_addresses = db.Column(db.UnicodeText())
  
     compute_optimal_destination = db.Column(db.Boolean())
  
@@ -55,6 +58,27 @@ class Estimation(db.Model):
         return 'cities' in self.get_output_dict()
  
  
+class EstimationView(ModelView):
+    # Show only name and email columns in list view
+    column_list = (
+        'public_id',
+        'status',
+        'first_name',
+        'last_name',
+        'origin_addresses',
+        'destination_addresses',
+        'warnings',
+        'errors',
+    )
+
+    # Enable search functionality - it will search for terms in
+    # name and email fields
+    # column_searchable_list = ('name', 'email')
+
+    # Add filters for name and email columns
+    column_filters = ('first_name', 'last_name')
+
+
 # USERS #######################################################################
  
 class User(db.Model, UserMixin):
 Flask==1.1.1
  
 # Flask Extensions
+Flask-Admin==1.5.4
 Flask-Assets==0.12
 Flask-Caching==1.7.2
 Flask-DebugToolbar==0.10.0