merged

.gitignore

@@ -2,6 +2,7 @@
 
 # C extensions
 *.so
+*.db
 
 # Packages
 *.egg

pytest_bdd_example/__init__.py

@@ -1,31 +0,0 @@
-from flask import Flask
-from flask.ext.admin import Admin
-from flask.ext.login import LoginManager
-from flask.ext.sqlalchemy import SQLAlchemy
-
-login_manager = LoginManager()
-
-app = Flask(__name__)
-app.config.from_object('settings')
-app.secret_key = 'A0Zr98j/3yX R~XHH!jmN]LWX/,?RT'
-
-db = SQLAlchemy(app)
-
-from flask.ext.admin.contrib.sqlamodel import ModelView
-
-from pytest_bdd_example import dashboard
-from pytest_bdd_example.dashboard.models import User
-
-
-app.register_blueprint(dashboard.bp, url_prefix='/dashboard')
-
-login_manager.init_app(app)
-
-
-@login_manager.user_loader
-def load_user(userid):
-    return User.get(userid)
-
-
-admin_panel = Admin(app)
-admin_panel.add_view(ModelView(User, db.session, endpoint='users'))

pytest_bdd_example/dashboard/__init__.py

@@ -1,3 +1,39 @@
-from flask import Blueprint
+from flask import Blueprint, Flask, url_for, redirect
+
+from flask.ext.admin import Admin
+from flask.ext.login import LoginManager
+from flask.ext.sqlalchemy import SQLAlchemy
+
+from pytest_bdd_example.dashboard import settings
+
+login_manager = LoginManager()
+
+app = Flask(__name__, static_folder=settings.MEDIA_ROOT, template_folder=settings.TEMPLATES_ROOT)
+app.config.from_object('pytest_bdd_example.dashboard.settings')
+app.secret_key = 'A0Zr98j/3yX R~XHH!jmN]LWX/,?RT'
+
+db = SQLAlchemy(app)
+
+from flask.ext.admin.contrib.sqlamodel import ModelView
+
+from pytest_bdd_example.dashboard.models import User
+
+from pytest_bdd_example.dashboard.views import *
+
+app.register_blueprint(bp, url_prefix='/dashboard')
+
+admin_panel = Admin(app)
+admin_panel.add_view(ModelView(User, db.session, endpoint='users'))
+
+login_manager.init_app(app)
+
+@login_manager.user_loader
+def load_user(userid):
+    return User.query.get(userid)
+
+@login_manager.unauthorized_handler
+def unauthorized():
+    return redirect(url_for('dashboard.login'))
+
+
 
-bp = Blueprint('dashboard', __name__)

pytest_bdd_example/dashboard/decorators.py

@@ -0,0 +1,14 @@
+from functools import wraps
+
+from flask import g, flash, redirect, url_for, request
+
+
+def requires_login(f):
+    @wraps(f)
+    def decorated_function(*args, **kwargs):
+        if g.user is None:
+            flash(u'You need to be signed in for this page.')
+            return redirect(url_for('dashboard.login', next=request.path))
+        return f(*args, **kwargs)
+    return decorated_function
+

pytest_bdd_example/dashboard/forms.py

@@ -0,0 +1,8 @@
+from flask.ext.wtf import Form
+from wtforms import TextField, PasswordField
+from wtforms.validators import Required, Email
+
+
+class LoginForm(Form):
+    email = TextField('Email address', [Required(), Email()])
+    password = PasswordField('Password', [Required()])

pytest_bdd_example/dashboard/models.py

@@ -1,27 +1,17 @@
-from pytest_bdd_example import db
+from flask.ext.security import UserMixin
+from pytest_bdd_example.dashboard import db
 
 ROLE_USER = 0
 ROLE_ADMIN = 1
 
 
-class User(db.Model):
+class User(db.Model, UserMixin):
-    uuid = db.Column(db.Integer, primary_key=True)
+    id = db.Column(db.Integer, primary_key=True)
     username = db.Column(db.String(64), unique=True)
     email = db.Column(db.String(120), unique=True)
     password = db.Column(db.String(20))
     role = db.Column(db.SmallInteger, default=ROLE_USER)
-
+    is_active = db.Column(db.Boolean, default=False)
-    def is_authenticated(self):
-        return True
-
-    def is_active(self):
-        return True
-
-    def is_anonymous(self):
-        return False
-
-    def get_id(self):
-        return unicode(self.id)
 
     def __repr__(self):
         return '<User %r>' % (self.username)

pytest_bdd_example/dashboard/settings.py

@@ -3,6 +3,6 @@ import os
 DEBUG = True
 BASE_PATH = os.path.dirname(os.path.abspath(__file__))
 MEDIA_ROOT = os.path.join(BASE_PATH, '../media')
-TEMPLATES_ROOT = os.path.join(BASE_PATH, '../templates', 'admin')
+TEMPLATES_ROOT = os.path.join(BASE_PATH, '../templates', 'dashboard')
 
-SQLALCHEMY_DATABASE_URI = 'sqlite:///' + os.path.join(BASE_PATH, '../pytest-bdd-example.db')
+SQLALCHEMY_DATABASE_URI = 'sqlite:///' + os.path.join(BASE_PATH, '../dashboard.db')

pytest_bdd_example/dashboard/views.py

@@ -1,12 +1,25 @@
-from flask import render_template
+from flask import render_template, Blueprint, url_for, redirect
-from flask.ext.login import current_user
+from flask.ext.login import login_required, login_user
 
-from pytest_bdd_example.dashboard import bp
+from pytest_bdd_example.dashboard import User
+from pytest_bdd_example.dashboard.forms import LoginForm
 
-# @bp.route('/')
+bp = Blueprint('dashboard', __name__)
-# def index(template='dashboatd.html'):
+
-#
+@bp.route('/')
-#     if not current_user.is_authenticated():
+@login_required
-#         template = "login.html"
+def index(template='dashboard.html'):
-#
+    return render_template(template)
-#     return render_template(template)
+
+@bp.route('/login/', methods=['GET', 'POST'])
+def login():
+    login_form = LoginForm()
+
+    if login_form.validate_on_submit():
+        user = User.query.filter_by(email=login_form.email.data).first()
+        login_user(user)
+        return redirect(url_for('dashboard.index'))
+
+    return render_template("login.html", **{
+        'form': login_form,
+    })

pytest_bdd_example/manage.py

@@ -1,11 +1,21 @@
 #!/usr/bin/env python
 
-from flask.ext.script import Manager, Shell, Server
+from flask.ext.script import Manager, Shell, Server, Command
 
-from pytest_bdd_example import app
+from pytest_bdd_example.dashboard import app as dashboard_app
+from pytest_bdd_example.shop import app as shop_app
 
-manager = Manager(app)
 
+def create_app(app=None):
+    APPS = {
+        'dashboard': dashboard_app,
+        'shop': shop_app,
+    }
+
+    return APPS[app]
+
+manager = Manager(create_app)
+manager.add_option("-app", "--application", dest="app", required=True)
 manager.add_command("runserver", Server("127.0.0.1"))
 manager.add_command("shell", Shell(use_ipython=True))
 manager.run()

pytest_bdd_example/templates/dashboard/login.html

@@ -4,14 +4,25 @@
 {% endblock %}
 
 {% block content %}
-<form class="form-signin">
+<form class="form-signin" method='POST'>
-        <h2 class="form-signin-heading">Please sign in</h2>
+    {{ form.csrf_token }}
-        <input type="text" class="input-block-level" placeholder="Email address">
+    <h2 class="form-signin-heading">Please sign in</h2>
-        <input type="password" class="input-block-level" placeholder="Password">
+
-        <label class="checkbox">
+    <div class="control-group">
-          <input type="checkbox" value="remember-me"> Remember me
+        <div class="controls">
-        </label>
+            {{ form.email(placeholder="Email address", class="input-block-level") }}
-        <button class="btn btn-large btn-primary" type="submit">Sign in</button>
+            <ul class="errors">{% for error in form.email.errors %}<li>{{ error|e }}</li>{% endfor %}</ul>
-      </form>
+        </div>
+    </div>
+
+    <div class="control-group">
+        <div class="controls">
+            {{ form.password(placeholder="Password", class="input-block-level") }}
+            <ul class="errors">{% for error in form.password.errors %}<li>{{ error|e }}</li>{% endfor %}</ul>
+        </div>
+    </div>
+
+    <button class="btn btn-large btn-primary" type="submit">Sign in</button>
+</form>
 
 {% endblock %}