def register_service(name):
if name == 'google':
return google.register_to(social)
if name == 'twitter':
return twitter.register_to(social)
if name == 'facebook':
return facebook.register_to(social)
if name == 'github':
return github.register_to(social)
if name == 'weibo':
return weibo.register_to(social)
"""
Views for the User model.
"""
import functools
from flask import Blueprint, render_template, url_for, redirect
from flask_oauthlib.client import OAuth
from flask_oauthlib.contrib.apps import github
from flask_login import login_required, current_user, login_user, logout_user
from codegolf import login_manager, app
from codegolf.database import db_session
from codegolf.models import User, Admin
user = Blueprint('user', __name__, url_prefix='/user')
oauth = OAuth(app)
git_auth = github.register_to(oauth)
@login_manager.user_loader
def user_loader(user_id):
return db_session.query(User).filter(User.id == user_id).one()
def admin_required(f):
"""
A decorator similar to login_required requiring an admin user
:param f: view to be made admin only
"""
@functools.wraps(f)
def wrapper(*a, **kw):
if db_session.query(Admin).filter(
"""
Views for the User model.
"""
import functools
from flask import Blueprint, render_template, url_for, redirect
from flask_oauthlib.client import OAuth
from flask_oauthlib.contrib.apps import github
from flask_login import login_required, current_user, login_user, logout_user
from codegolf import login_manager, app
from codegolf.database import db_session
from codegolf.models import User, Admin
user = Blueprint('user', __name__, url_prefix='/user')
oauth = OAuth(app)
git_auth = github.register_to(oauth)
@login_manager.user_loader
def user_loader(user_id):
return db_session.query(User).filter(User.id == user_id).one()
def admin_required(f):
"""
A decorator similar to login_required requiring an admin user
:param f: view to be made admin only
"""
@functools.wraps(f)
def wrapper(*a, **kw):
if db_session.query(Admin).filter(Admin.id == current_user.id).first() is None:
