Merge pull request #5 from samiuelson/KG/flaskAdmin

added basic auth to admin panel

Author: samiuelson

dbms/rdb.py

@@ -1,8 +1,37 @@
+from flask import redirect, Response
 from flask_sqlalchemy import SQLAlchemy
-from flask_admin.contrib.sqla import ModelView
+from flask_admin.contrib import sqla
+from flask_basicauth import BasicAuth
+
+from werkzeug.exceptions import HTTPException
+
+class AuthException(HTTPException):
+    def __init__(self, message):
+        super(AuthException, self).__init__(message, Response(
+            "You could not be authenticated. Please refresh the page.", 401,
+            {'WWW-Authenticate': 'Basic realm="Login Required"'}
+        ))
+
+
+class ModelView(sqla.ModelView):
+    @classmethod
+    def setup_basic_auth(cls, basic_auth):
+        cls.basic_auth = basic_auth
+    def is_accessible(self):
+        if not self.basic_auth.authenticate():
+            raise AuthException('Not authenticated.')
+        else:
+            return True
+
+    def inaccessible_callback(self, name, **kwargs):
+        return redirect(self.basic_auth.challenge())
 
 db = SQLAlchemy()
 
-def register_admin(admin):
-    from .models import Intent
+def register_admin(admin, app):
+    from .models import Intent, BibleVerse
+    basic_auth = BasicAuth(app)
+    ModelView.setup_basic_auth(basic_auth)
     admin.add_view(ModelView(Intent, db.session))
+    admin.add_view(ModelView(BibleVerse, db.session))
+

requirements.txt

@@ -38,3 +38,4 @@ SQLAlchemy
 Flask-SQLAlchemy
 psycopg2==2.6.2
 Flask-Admin==1.4.2
+Flask-BasicAuth==0.2.0

web.py

@@ -3,21 +3,21 @@
 from flask import Flask, request
 from flask.views import MethodView
 from flask_admin import Admin
-from flask_admin.contrib.sqla import ModelView
 from flask_babel import Babel
 from facebook.api import FacebookApi
 from prayer import PrayerWebhook as webhook
 from dbms.rdb import db, register_admin
 
-
 def create_app():
     app = Flask(__name__)
     app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///intent.db'
     app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
+    app.config['BASIC_AUTH_USERNAME'] = 'john'
+    app.config['BASIC_AUTH_PASSWORD'] = 'matrix'
     babel = Babel(app)
     admin = Admin(app, name='PrayerBot', template_mode='bootstrap3')
     db.init_app(app)
-    register_admin(admin)
+    register_admin(admin, app)
     return app
 
 ###