mirror of
https://github.com/haiwen/seahub.git
synced 2025-09-12 21:30:39 +00:00
c5d80ee8e7
NOTE: Org feature is not enabled by default, in order to use org, need set USE_ORG to True in settings or local_settings.
38 lines
1.0 KiB
Python
38 lines
1.0 KiB
Python
from django.core.cache import cache
|
|
from django.http import HttpResponseRedirect
|
|
|
|
from seaserv import get_org_by_url_prefix, get_orgs_by_user
|
|
|
|
from settings import ORG_CACHE_PREFIX
|
|
try:
|
|
from seahub.settings import USE_ORG
|
|
except ImportError:
|
|
USE_ORG = False
|
|
|
|
class OrganizationMiddleware(object):
|
|
"""
|
|
Middleware that add organization info to request when user in organization
|
|
context.
|
|
"""
|
|
|
|
def process_request(self, request):
|
|
if USE_ORG:
|
|
request.use_org = True
|
|
|
|
# Get current org context
|
|
org = cache.get(ORG_CACHE_PREFIX + request.user.username)
|
|
request.user.org = org
|
|
|
|
# Get all orgs user created.
|
|
orgs = get_orgs_by_user(request.user.username)
|
|
request.user.orgs = orgs
|
|
else:
|
|
request.use_org = False
|
|
request.user.org = None
|
|
request.user.orgs = None
|
|
|
|
return None
|
|
|
|
def process_response(self, request, response):
|
|
return response
|