mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-12-21 11:33:03 +00:00
Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
e9652d56ff | ||
|
|
97262645be |
@@ -2,6 +2,7 @@
|
|||||||
#
|
#
|
||||||
import json
|
import json
|
||||||
import os
|
import os
|
||||||
|
import uuid
|
||||||
|
|
||||||
import redis_lock
|
import redis_lock
|
||||||
import redis
|
import redis
|
||||||
@@ -103,6 +104,10 @@ def get_celery_periodic_task(task_name):
|
|||||||
|
|
||||||
def get_celery_task_log_path(task_id):
|
def get_celery_task_log_path(task_id):
|
||||||
task_id = str(task_id)
|
task_id = str(task_id)
|
||||||
|
try:
|
||||||
|
uuid.UUID(task_id)
|
||||||
|
except:
|
||||||
|
return
|
||||||
rel_path = os.path.join(task_id[0], task_id[1], task_id + '.log')
|
rel_path = os.path.join(task_id[0], task_id[1], task_id + '.log')
|
||||||
path = os.path.join(settings.CELERY_LOG_DIR, rel_path)
|
path = os.path.join(settings.CELERY_LOG_DIR, rel_path)
|
||||||
os.makedirs(os.path.dirname(path), exist_ok=True)
|
os.makedirs(os.path.dirname(path), exist_ok=True)
|
||||||
|
|||||||
@@ -16,7 +16,7 @@ class CeleryLogWebsocket(JsonWebsocketConsumer):
|
|||||||
|
|
||||||
def connect(self):
|
def connect(self):
|
||||||
user = self.scope["user"]
|
user = self.scope["user"]
|
||||||
if user.is_authenticated and user.is_org_admin:
|
if user.is_authenticated:
|
||||||
self.accept()
|
self.accept()
|
||||||
else:
|
else:
|
||||||
self.close()
|
self.close()
|
||||||
|
|||||||
@@ -84,7 +84,10 @@ def set_org_mapper(org_mapper):
|
|||||||
|
|
||||||
def get_org_mapper():
|
def get_org_mapper():
|
||||||
org_mapper = _find('org_mapper')
|
org_mapper = _find('org_mapper')
|
||||||
if org_mapper is None:
|
return org_mapper
|
||||||
|
|
||||||
|
|
||||||
|
def update_org_mapper():
|
||||||
org_mapper = construct_org_mapper()
|
org_mapper = construct_org_mapper()
|
||||||
set_org_mapper(org_mapper)
|
set_org_mapper(org_mapper)
|
||||||
return org_mapper
|
return org_mapper
|
||||||
@@ -93,6 +96,13 @@ def get_org_mapper():
|
|||||||
def get_org_by_id(org_id):
|
def get_org_by_id(org_id):
|
||||||
org_id = str(org_id)
|
org_id = str(org_id)
|
||||||
org_mapper = get_org_mapper()
|
org_mapper = get_org_mapper()
|
||||||
|
if not org_mapper:
|
||||||
|
org_mapper = update_org_mapper()
|
||||||
|
|
||||||
|
org = org_mapper.get(org_id)
|
||||||
|
if not org:
|
||||||
|
# 解决celery缓存org_mapper问题
|
||||||
|
org_mapper = update_org_mapper()
|
||||||
org = org_mapper.get(org_id)
|
org = org_mapper.get(org_id)
|
||||||
return org
|
return org
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user