mirror of
https://github.com/haiwen/seahub.git
synced 2025-08-22 16:56:57 +00:00
e6e6e82b82
Conflicts: seahub/api2/endpoints/file.py seahub/urls.py static/scripts/app/views/share.js
98 lines
2.8 KiB
Python
98 lines
2.8 KiB
Python
# Copyright (c) 2012-2016 Seafile Ltd.
|
|
import re
|
|
|
|
from django.core.cache import cache
|
|
from django.core.urlresolvers import reverse
|
|
from django.http import HttpResponseRedirect
|
|
|
|
from seaserv import ccnet_api
|
|
|
|
from seahub.notifications.models import Notification
|
|
from seahub.notifications.utils import refresh_cache
|
|
try:
|
|
from seahub.settings import CLOUD_MODE
|
|
except ImportError:
|
|
CLOUD_MODE = False
|
|
try:
|
|
from seahub.settings import MULTI_TENANCY
|
|
except ImportError:
|
|
MULTI_TENANCY = False
|
|
from seahub.settings import SITE_ROOT
|
|
|
|
class BaseMiddleware(object):
|
|
"""
|
|
Middleware that add organization, group info to user.
|
|
"""
|
|
|
|
def process_request(self, request):
|
|
username = request.user.username
|
|
request.user.org = None
|
|
|
|
if CLOUD_MODE:
|
|
request.cloud_mode = True
|
|
|
|
if MULTI_TENANCY:
|
|
orgs = ccnet_api.get_orgs_by_user(username)
|
|
if orgs:
|
|
request.user.org = orgs[0]
|
|
else:
|
|
request.cloud_mode = False
|
|
|
|
return None
|
|
|
|
def process_response(self, request, response):
|
|
return response
|
|
|
|
class InfobarMiddleware(object):
|
|
"""Query info bar close status, and store into request."""
|
|
|
|
def get_from_db(self):
|
|
ret = Notification.objects.all().filter(primary=1)
|
|
refresh_cache()
|
|
return ret
|
|
|
|
def process_request(self, request):
|
|
|
|
# filter AJAX request out
|
|
if request.is_ajax():
|
|
return None
|
|
|
|
# filter API request out
|
|
if "api2/" in request.path or "api/v2.1/" in request.path:
|
|
return None
|
|
|
|
topinfo_close = request.COOKIES.get('info_id', '')
|
|
|
|
cur_note = cache.get('CUR_TOPINFO') if cache.get('CUR_TOPINFO') else \
|
|
self.get_from_db()
|
|
if not cur_note:
|
|
request.cur_note = None
|
|
else:
|
|
if str(cur_note[0].id) in topinfo_close.split('_'):
|
|
request.cur_note = None
|
|
else:
|
|
request.cur_note = cur_note[0]
|
|
|
|
return None
|
|
|
|
def process_response(self, request, response):
|
|
return response
|
|
|
|
|
|
class ForcePasswdChangeMiddleware(object):
|
|
def _request_in_black_list(self, request):
|
|
path = request.path
|
|
black_list = (r'^%s$' % SITE_ROOT, r'home/.+', r'repo/.+',
|
|
r'[f|d]/[a-f][0-9]+', r'group/\d+', r'groups/',
|
|
r'share/', r'profile/', r'notification/list/')
|
|
|
|
for patt in black_list:
|
|
if re.search(patt, path) is not None:
|
|
return True
|
|
return False
|
|
|
|
def process_request(self, request):
|
|
if request.session.get('force_passwd_change', False):
|
|
if self._request_in_black_list(request):
|
|
return HttpResponseRedirect(reverse('auth_password_change'))
|