Merge branch 'master' into partials-update

Conflicts:
	realms/modules/wiki/views.py

realms/modules/__init__.py

@@ -0,0 +1 @@
+from __future__ import absolute_import

realms/modules/auth/__init__.py

@@ -1,7 +1,10 @@
-from realms import login_manager
+from __future__ import absolute_import
+
 from flask import request, flash, redirect
 from flask_login import login_url
 
+from realms import login_manager
+
 modules = set()
 
 @login_manager.unauthorized_handler

realms/modules/auth/ldap/__init__.py

@@ -1,3 +1,5 @@
-from ..models import Auth
+from __future__ import absolute_import
+
+from realms.modules.auth.models import Auth
 
 Auth.register('ldap')

realms/modules/auth/ldap/forms.py

@@ -1,7 +1,9 @@
+from __future__ import absolute_import
+
 from flask_wtf import Form
 from wtforms import StringField, PasswordField, validators
 
 
 class LoginForm(Form):
     login = StringField('Username', [validators.DataRequired()])
-    password = PasswordField('Password', [validators.DataRequired()])
+    password = PasswordField('Password', [validators.DataRequired()])

realms/modules/auth/ldap/models.py

@@ -1,8 +1,11 @@
+from __future__ import absolute_import
+
 from flask import render_template
 from flask_login import login_user
-from realms import ldap
 from flask_ldap_login import LDAPLoginForm
-from ..models import BaseUser
+
+from realms import ldap
+from realms.modules.auth.models import BaseUser
 
 
 users = {}

realms/modules/auth/ldap/views.py

@@ -1,7 +1,11 @@
-from flask import current_app, request, redirect, Blueprint, flash, url_for
-from ..ldap.models import User
+from __future__ import absolute_import
+
+from flask import current_app, request, redirect, Blueprint, flash, url_for, session
 from flask_ldap_login import LDAPLoginForm
 
+from .models import User
+
+
 blueprint = Blueprint('auth.ldap', __name__)
 
 
@@ -14,6 +18,6 @@ def login():
         return redirect(url_for('auth.login'))
 
     if User.auth(form.user, request.form['password']):
-        return redirect(request.args.get("next") or url_for(current_app.config['ROOT_ENDPOINT']))
+        return redirect(session.get("next_url") or url_for(current_app.config['ROOT_ENDPOINT']))
     else:
         return redirect(url_for('auth.login'))

realms/modules/auth/local/__init__.py

@@ -1,3 +1,5 @@
-from ..models import Auth
+from __future__ import absolute_import
+
+from realms.modules.auth.models import Auth
 
 Auth.register('local')

realms/modules/auth/local/commands.py

@@ -1,8 +1,11 @@
+from __future__ import absolute_import
+
 import click
-from realms.lib.util import random_string
-from realms.modules.auth.local.models import User
-from realms.lib.util import green, red, yellow
+
 from realms import cli_group
+from realms.lib.util import random_string
+from realms.lib.util import green, red, yellow
+from .models import User
 
 
 @cli_group(short_help="Auth Module")

realms/modules/auth/local/forms.py

@@ -1,3 +1,5 @@
+from __future__ import absolute_import
+
 from flask_wtf import Form
 from wtforms import StringField, PasswordField, validators
 

realms/modules/auth/local/hooks.py

@@ -1,5 +1,8 @@
+from __future__ import absolute_import
+
 from flask import current_app
 from flask_wtf import RecaptchaField
+
 from .forms import RegistrationForm
 
 

realms/modules/auth/local/models.py

@@ -1,11 +1,15 @@
+from __future__ import absolute_import
+
+from hashlib import sha256
+
 from flask import current_app, render_template
 from flask_login import logout_user, login_user
+from itsdangerous import URLSafeSerializer, BadSignature
+
 from realms import login_manager, db
 from realms.lib.model import Model
-from ..models import BaseUser
+from realms.modules.auth.models import BaseUser
 from .forms import LoginForm
-from itsdangerous import URLSafeSerializer, BadSignature
-from hashlib import sha256
 
 
 @login_manager.token_loader

realms/modules/auth/local/views.py

@@ -1,6 +1,10 @@
-from flask import current_app, render_template, request, redirect, Blueprint, flash, url_for
-from realms.modules.auth.local.models import User
-from realms.modules.auth.local.forms import LoginForm, RegistrationForm
+from __future__ import absolute_import
+
+from flask import current_app, render_template, request, redirect, Blueprint, flash, url_for, session
+
+from .models import User
+from .forms import LoginForm, RegistrationForm
+
 
 blueprint = Blueprint('auth.local', __name__)
 
@@ -46,6 +50,6 @@ def register():
         User.create(request.form['username'], request.form['email'], request.form['password'])
         User.auth(request.form['email'], request.form['password'])
 
-        return redirect(request.args.get("next") or url_for(current_app.config['ROOT_ENDPOINT']))
+        return redirect(session.get("next_url") or url_for(current_app.config['ROOT_ENDPOINT']))
 
     return render_template("auth/register.html", form=form)

realms/modules/auth/models.py

@@ -1,12 +1,16 @@
+from __future__ import absolute_import
+
+import importlib
+from hashlib import sha256
+
 from flask import current_app
 from flask_login import UserMixin, logout_user, AnonymousUserMixin
+from itsdangerous import URLSafeSerializer, BadSignature
+import bcrypt
+
 from realms import login_manager
 from realms.lib.util import gravatar_url
-from itsdangerous import URLSafeSerializer, BadSignature
-from hashlib import sha256
 from . import modules
-import bcrypt
-import importlib
 
 
 @login_manager.user_loader

realms/modules/auth/oauth/__init__.py

@@ -1,3 +1,5 @@
-from ..models import Auth
+from __future__ import absolute_import
+
+from realms.modules.auth.models import Auth
 
 Auth.register('oauth')

realms/modules/auth/oauth/models.py

@@ -1,9 +1,11 @@
+from __future__ import absolute_import
+
 from flask import session
 from flask_login import login_user
 from flask_oauthlib.client import OAuth
 
 from realms import config
-from ..models import BaseUser
+from realms.modules.auth.models import BaseUser
 
 config = config.conf
 

realms/modules/auth/oauth/views.py

@@ -1,3 +1,5 @@
+from __future__ import absolute_import
+
 from flask import Blueprint, url_for, request, flash, redirect, session, current_app
 from .models import User
 
@@ -16,7 +18,7 @@ def login(provider):
 
 @blueprint.route('/login/oauth/<provider>/callback')
 def callback(provider):
-    next_url = request.args.get('next') or url_for(current_app.config['ROOT_ENDPOINT'])
+    next_url = session.get('next_url') or url_for(current_app.config['ROOT_ENDPOINT'])
     try:
         remote_app = User.get_app(provider)
         resp = remote_app.authorized_response()

realms/modules/auth/tests.py

@@ -0,0 +1 @@
+from __future__ import absolute_import

realms/modules/auth/views.py

@@ -1,12 +1,18 @@
-from flask import current_app, render_template, request, redirect, Blueprint, flash, url_for
+from __future__ import absolute_import
+
+from flask import current_app, render_template, request, redirect, Blueprint, flash, url_for, session
 from flask_login import logout_user
-from realms.modules.auth.models import Auth
+
+from .models import Auth
+
 
 blueprint = Blueprint('auth', __name__, template_folder='templates')
 
 
 @blueprint.route("/login", methods=['GET', 'POST'])
 def login():
+    next_url = request.args.get('next') or url_for(current_app.config['ROOT_ENDPOINT'])
+    session['next_url'] = next_url
     return render_template("auth/login.html", forms=Auth.login_forms())
 
 

realms/modules/search/__init__.py

@@ -0,0 +1 @@
+from __future__ import absolute_import

realms/modules/search/commands.py

@@ -1,5 +1,8 @@
+from __future__ import absolute_import
+
 import click
 from flask import current_app
+
 from realms import search, cli_group
 from realms.modules.wiki.models import Wiki
 

realms/modules/search/hooks.py

@@ -1,5 +1,7 @@
-from realms.modules.wiki.models import WikiPage
+from __future__ import absolute_import
+
 from realms import search
+from realms.modules.wiki.models import WikiPage
 
 
 @WikiPage.after('write')

realms/modules/search/models.py

@@ -1,7 +1,10 @@
+from __future__ import absolute_import
+
 import re
 import sys
 
 from flask import g, current_app
+
 from realms.lib.util import filename_to_cname
 
 

realms/modules/search/views.py

@@ -1,7 +1,11 @@
+from __future__ import absolute_import
+
 from flask import render_template, request, Blueprint, current_app
-from flask.ext.login import current_user
+from flask_login import current_user
+
 from realms import search as search_engine
 
+
 blueprint = Blueprint('search', __name__, template_folder='templates')
 
 

realms/modules/wiki/__init__.py

@@ -1,6 +1,9 @@
+from __future__ import absolute_import
+
 import os
 import sys
-from realms.modules.wiki.models import Wiki
+
+from .models import Wiki
 
 
 def init(app):

realms/modules/wiki/assets.py

@@ -1,3 +1,5 @@
+from __future__ import absolute_import
+
 from realms import assets
 
 assets.register('editor.js',

realms/modules/wiki/hooks.py

@@ -1,4 +1,7 @@
+from __future__ import absolute_import
+
 from flask import g, current_app
+
 from .models import Wiki
 
 

realms/modules/wiki/models.py

@@ -1,13 +1,17 @@
+from __future__ import absolute_import
+
 import os
 import posixpath
 import re
+
 import ghdiff
 import yaml
 from dulwich.object_store import tree_lookup_path
 from dulwich.repo import Repo, NotGitRepository
-from realms.lib.util import cname_to_filename, filename_to_cname
+
 from realms import cache
 from realms.lib.hook import HookMixin
+from realms.lib.util import cname_to_filename, filename_to_cname
 
 
 class PageNotFound(Exception):

realms/modules/wiki/tests.py

@@ -1,6 +1,10 @@
+from __future__ import absolute_import
+
 import json
+
 from nose.tools import *
 from flask import url_for
+
 from realms.lib.util import cname_to_filename, filename_to_cname
 from realms.lib.test import BaseTest
 

realms/modules/wiki/views.py

@@ -1,12 +1,17 @@
+from __future__ import absolute_import
+
 import collections
 import itertools
 import sys
 from datetime import datetime
+
 from flask import abort, g, render_template, request, redirect, Blueprint, flash, url_for, current_app
 from flask_login import login_required, current_user
+
 from realms.lib.util import to_canonical, remove_ext, gravatar_url
 from .models import PageNotFound
 
+
 blueprint = Blueprint('wiki', __name__, template_folder='templates',
                       static_folder='static', static_url_path='/static/wiki')