haiwen/seahub
1
0
Fork 0
mirror of https://github.com/haiwen/seahub.git synced 2025-08-16 22:17:59 +00:00
Code Issues Releases Wiki Activity

fix shibboleth update_user_quota (#5242)

Browse Source
This commit is contained in:
王健辉 2022-08-15 11:37:15 +08:00 committed by GitHub
parent a3eb82f3fb
commit 0cf73131f5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
thirdpart/shibboleth/middleware.py
View File

@ -18,6 +18,7 @@ from seahub.base.accounts import User
from seahub.base.sudo_mode import update_sudo_mode_ts from seahub.base.sudo_mode import update_sudo_mode_ts
from seahub.profile.models import Profile from seahub.profile.models import Profile
from seahub.utils.file_size import get_quota_from_string from seahub.utils.file_size import get_quota_from_string
from seahub.role_permissions.utils import get_enabled_role_permissions_by_role
# Get an instance of a logger # Get an instance of a logger
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -224,8 +225,9 @@ class ShibbolethRemoteUserMiddleware(RemoteUserMiddleware):
return role return role
def update_user_quota(self, user, user_role): def update_user_quota(self, user, user_role):
if user.permissions.role_quota(): role_quota = get_enabled_role_permissions_by_role(user_role)['role_quota']
quota = get_quota_from_string(user.permissions.role_quota()) if role_quota:
quota = get_quota_from_string(role_quota)
logger.info('Set quota[%d] for user: %s, role[%s]' % (quota, user.username, user_role)) logger.info('Set quota[%d] for user: %s, role[%s]' % (quota, user.username, user_role))
seafile_api.set_role_quota(user_role, quota) seafile_api.set_role_quota(user_role, quota)
else: else: