Move the auth context methods into their own file so that we don't have auth trying to import itself
This commit is contained in:
parent
a120f6c64a
commit
61ca29de04
5 changed files with 12 additions and 14 deletions
11
auth/auth.py
11
auth/auth.py
|
@ -1,7 +1,7 @@
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
from functools import wraps
|
from functools import wraps
|
||||||
from flask import request, _request_ctx_stack, abort as flask_abort, session
|
from flask import request, _request_ctx_stack, session
|
||||||
from flask.ext.principal import identity_changed, Identity
|
from flask.ext.principal import identity_changed, Identity
|
||||||
from base64 import b64decode
|
from base64 import b64decode
|
||||||
|
|
||||||
|
@ -15,15 +15,6 @@ from util.http import abort
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
def get_authenticated_user():
|
|
||||||
return getattr(_request_ctx_stack.top, 'authenticated_user', None)
|
|
||||||
|
|
||||||
|
|
||||||
def get_validated_token():
|
|
||||||
return getattr(_request_ctx_stack.top, 'validated_token', None)
|
|
||||||
|
|
||||||
|
|
||||||
def process_basic_auth(auth):
|
def process_basic_auth(auth):
|
||||||
normalized = [part.strip() for part in auth.split(' ') if part]
|
normalized = [part.strip() for part in auth.split(' ') if part]
|
||||||
if normalized[0].lower() != 'basic' or len(normalized) != 2:
|
if normalized[0].lower() != 'basic' or len(normalized) != 2:
|
||||||
|
|
7
auth/auth_context.py
Normal file
7
auth/auth_context.py
Normal file
|
@ -0,0 +1,7 @@
|
||||||
|
from flask import _request_ctx_stack
|
||||||
|
|
||||||
|
def get_authenticated_user():
|
||||||
|
return getattr(_request_ctx_stack.top, 'authenticated_user', None)
|
||||||
|
|
||||||
|
def get_validated_token():
|
||||||
|
return getattr(_request_ctx_stack.top, 'validated_token', None)
|
|
@ -9,8 +9,8 @@ from collections import OrderedDict
|
||||||
from data import model, userevent
|
from data import model, userevent
|
||||||
from data.queue import webhook_queue
|
from data.queue import webhook_queue
|
||||||
from app import mixpanel, app
|
from app import mixpanel, app
|
||||||
from auth.auth import (process_auth, get_authenticated_user,
|
from auth.auth import process_auth
|
||||||
get_validated_token)
|
from auth.auth_context import get_authenticated_user, get_validated_token
|
||||||
from util.names import parse_repository_name
|
from util.names import parse_repository_name
|
||||||
from util.email import send_confirmation_email
|
from util.email import send_confirmation_email
|
||||||
from auth.permissions import (ModifyRepositoryPermission, UserPermission,
|
from auth.permissions import (ModifyRepositoryPermission, UserPermission,
|
||||||
|
|
|
@ -2,7 +2,7 @@ import logging
|
||||||
import json
|
import json
|
||||||
|
|
||||||
from flask import (make_response, request, session, Response, redirect,
|
from flask import (make_response, request, session, Response, redirect,
|
||||||
Blueprint, abort as flask_abort)
|
Blueprint)
|
||||||
from functools import wraps
|
from functools import wraps
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
from time import time
|
from time import time
|
||||||
|
|
|
@ -2,6 +2,7 @@ import logging
|
||||||
|
|
||||||
from app import mixpanel
|
from app import mixpanel
|
||||||
from flask import request, abort as flask_abort, jsonify
|
from flask import request, abort as flask_abort, jsonify
|
||||||
|
from auth.auth_context import get_authenticated_user, get_validated_token
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
@ -15,7 +16,6 @@ DEFAULT_MESSAGE[409] = 'Conflict'
|
||||||
DEFAULT_MESSAGE[501] = 'Not Implemented'
|
DEFAULT_MESSAGE[501] = 'Not Implemented'
|
||||||
|
|
||||||
def abort(status_code, message=None, issue=None, **kwargs):
|
def abort(status_code, message=None, issue=None, **kwargs):
|
||||||
from auth.auth import get_authenticated_user, get_validated_token
|
|
||||||
|
|
||||||
message = (str(message) % kwargs if message else
|
message = (str(message) % kwargs if message else
|
||||||
DEFAULT_MESSAGE.get(status_code, ''))
|
DEFAULT_MESSAGE.get(status_code, ''))
|
||||||
|
|
Reference in a new issue