mirror of https://github.com/jumpserver/jumpserver
56 lines
1021 B
Python
56 lines
1021 B
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
|
|
from functools import partial
|
|
|
|
from common.utils import LocalProxy
|
|
from .models import Organization
|
|
|
|
try:
|
|
from threading import local
|
|
except ImportError:
|
|
from django.utils._threading_local import local
|
|
|
|
_thread_locals = local()
|
|
|
|
|
|
def get_org_from_request(request):
|
|
oid = request.session.get("oid")
|
|
org = Organization.get_instance(oid)
|
|
return org
|
|
|
|
|
|
def get_current_request():
|
|
return getattr(_thread_locals, 'request', None)
|
|
|
|
|
|
def get_current_org():
|
|
org = getattr(_thread_locals, 'current_org', None)
|
|
return org
|
|
|
|
|
|
def get_current_user():
|
|
return getattr(_thread_locals, 'user', None)
|
|
|
|
|
|
def set_current_org(org):
|
|
setattr(_thread_locals, 'current_org', org)
|
|
|
|
|
|
def set_to_default_org():
|
|
set_current_org(Organization.default())
|
|
|
|
|
|
def set_to_root_org():
|
|
set_current_org(Organization.root())
|
|
|
|
|
|
def _find(attr):
|
|
if hasattr(_thread_locals, attr):
|
|
return getattr(_thread_locals, attr)
|
|
return None
|
|
|
|
|
|
current_org = LocalProxy(get_current_org)
|
|
|