Merge "pep8 middleware"
This commit is contained in:
commit
2fc9716ec9
@ -125,7 +125,8 @@ class CNAMELookupMiddleware(object):
|
||||
break
|
||||
else:
|
||||
# try one more deep in the chain
|
||||
self.logger.debug(_('Following CNAME chain for ' \
|
||||
self.logger.debug(
|
||||
_('Following CNAME chain for '
|
||||
'%(given_domain)s to %(found_domain)s') %
|
||||
{'given_domain': given_domain,
|
||||
'found_domain': found_domain})
|
||||
|
@ -245,8 +245,8 @@ class KeystoneAuth(object):
|
||||
"""
|
||||
# Allow container sync.
|
||||
if (req.environ.get('swift_sync_key')
|
||||
and req.environ['swift_sync_key'] ==
|
||||
req.headers.get('x-container-sync-key', None)
|
||||
and (req.environ['swift_sync_key'] ==
|
||||
req.headers.get('x-container-sync-key', None))
|
||||
and 'x-timestamp' in req.headers):
|
||||
log_msg = 'allowing proxy %s for container-sync' % req.remote_addr
|
||||
self.logger.debug(log_msg)
|
||||
|
@ -114,15 +114,18 @@ class NameCheckMiddleware(object):
|
||||
req = Request(env)
|
||||
|
||||
if self.check_character(req):
|
||||
return HTTPBadRequest(request=req,
|
||||
return HTTPBadRequest(
|
||||
request=req,
|
||||
body=("Object/Container name contains forbidden chars from %s"
|
||||
% self.forbidden_chars))(env, start_response)
|
||||
elif self.check_length(req):
|
||||
return HTTPBadRequest(request=req,
|
||||
body=("Object/Container name longer than the allowed maximum %s"
|
||||
% self.maximum_length))(env, start_response)
|
||||
return HTTPBadRequest(
|
||||
request=req,
|
||||
body=("Object/Container name longer than the allowed maximum "
|
||||
"%s" % self.maximum_length))(env, start_response)
|
||||
elif self.check_regexp(req):
|
||||
return HTTPBadRequest(request=req,
|
||||
return HTTPBadRequest(
|
||||
request=req,
|
||||
body=("Object/Container name contains a forbidden substring "
|
||||
"from regular expression %s"
|
||||
% self.forbidden_regexp))(env, start_response)
|
||||
|
@ -47,9 +47,11 @@ class RateLimitMiddleware(object):
|
||||
float(conf.get('log_sleep_time_seconds', 0))
|
||||
self.clock_accuracy = int(conf.get('clock_accuracy', 1000))
|
||||
self.rate_buffer_seconds = int(conf.get('rate_buffer_seconds', 5))
|
||||
self.ratelimit_whitelist = [acc.strip() for acc in
|
||||
self.ratelimit_whitelist = \
|
||||
[acc.strip() for acc in
|
||||
conf.get('account_whitelist', '').split(',') if acc.strip()]
|
||||
self.ratelimit_blacklist = [acc.strip() for acc in
|
||||
self.ratelimit_blacklist = \
|
||||
[acc.strip() for acc in
|
||||
conf.get('account_blacklist', '').split(',') if acc.strip()]
|
||||
self.memcache_client = None
|
||||
conf_limits = []
|
||||
@ -139,8 +141,8 @@ class RateLimitMiddleware(object):
|
||||
try:
|
||||
now_m = int(round(time.time() * self.clock_accuracy))
|
||||
time_per_request_m = int(round(self.clock_accuracy / max_rate))
|
||||
running_time_m = self.memcache_client.incr(key,
|
||||
delta=time_per_request_m)
|
||||
running_time_m = self.memcache_client.incr(
|
||||
key, delta=time_per_request_m)
|
||||
need_to_sleep_m = 0
|
||||
if (now_m - running_time_m >
|
||||
self.rate_buffer_seconds * self.clock_accuracy):
|
||||
@ -155,7 +157,8 @@ class RateLimitMiddleware(object):
|
||||
if max_sleep_m - need_to_sleep_m <= self.clock_accuracy * 0.01:
|
||||
# treat as no-op decrement time
|
||||
self.memcache_client.decr(key, delta=time_per_request_m)
|
||||
raise MaxSleepTimeHitError("Max Sleep Time Exceeded: %.2f" %
|
||||
raise MaxSleepTimeHitError(
|
||||
"Max Sleep Time Exceeded: %.2f" %
|
||||
(float(need_to_sleep_m) / self.clock_accuracy))
|
||||
|
||||
return float(need_to_sleep_m) / self.clock_accuracy
|
||||
@ -176,7 +179,8 @@ class RateLimitMiddleware(object):
|
||||
account_name)
|
||||
eventlet.sleep(self.BLACK_LIST_SLEEP)
|
||||
return Response(status='497 Blacklisted',
|
||||
body='Your account has been blacklisted', request=req)
|
||||
body='Your account has been blacklisted',
|
||||
request=req)
|
||||
if account_name in self.ratelimit_whitelist:
|
||||
return None
|
||||
for key, max_rate in self.get_ratelimitable_key_tuples(
|
||||
@ -186,15 +190,17 @@ class RateLimitMiddleware(object):
|
||||
need_to_sleep = self._get_sleep_time(key, max_rate)
|
||||
if self.log_sleep_time_seconds and \
|
||||
need_to_sleep > self.log_sleep_time_seconds:
|
||||
self.logger.warning(_("Ratelimit sleep log: %(sleep)s for "
|
||||
self.logger.warning(
|
||||
_("Ratelimit sleep log: %(sleep)s for "
|
||||
"%(account)s/%(container)s/%(object)s"),
|
||||
{'sleep': need_to_sleep, 'account': account_name,
|
||||
'container': container_name, 'object': obj_name})
|
||||
if need_to_sleep > 0:
|
||||
eventlet.sleep(need_to_sleep)
|
||||
except MaxSleepTimeHitError, e:
|
||||
self.logger.error(_('Returning 498 for %(meth)s to '
|
||||
'%(acc)s/%(cont)s/%(obj)s . Ratelimit (Max Sleep) %(e)s'),
|
||||
self.logger.error(
|
||||
_('Returning 498 for %(meth)s to %(acc)s/%(cont)s/%(obj)s '
|
||||
'. Ratelimit (Max Sleep) %(e)s'),
|
||||
{'meth': req.method, 'acc': account_name,
|
||||
'cont': container_name, 'obj': obj_name, 'e': str(e)})
|
||||
error_resp = Response(status='498 Rate Limited',
|
||||
|
@ -256,10 +256,10 @@ class TempAuth(object):
|
||||
# account DELETE or PUT...
|
||||
req.environ['swift_owner'] = True
|
||||
return None
|
||||
if (req.environ.get('swift_sync_key') and
|
||||
req.environ['swift_sync_key'] ==
|
||||
req.headers.get('x-container-sync-key', None) and
|
||||
'x-timestamp' in req.headers):
|
||||
if (req.environ.get('swift_sync_key')
|
||||
and (req.environ['swift_sync_key'] ==
|
||||
req.headers.get('x-container-sync-key', None))
|
||||
and 'x-timestamp' in req.headers):
|
||||
return None
|
||||
if req.method == 'OPTIONS':
|
||||
#allow OPTIONS requests to proceed as normal
|
||||
|
@ -261,8 +261,8 @@ class TempURL(object):
|
||||
already = True
|
||||
break
|
||||
if not already:
|
||||
headers.append(('Content-Disposition',
|
||||
'attachment; filename=%s' %
|
||||
headers.append(
|
||||
('Content-Disposition', 'attachment; filename=%s' %
|
||||
(quote(basename(env['PATH_INFO'])))))
|
||||
return start_response(status, headers, exc_info)
|
||||
|
||||
@ -365,7 +365,8 @@ class TempURL(object):
|
||||
"""
|
||||
if not request_method:
|
||||
request_method = env['REQUEST_METHOD']
|
||||
return hmac.new(key, '%s\n%s\n%s' % (request_method, expires,
|
||||
return hmac.new(
|
||||
key, '%s\n%s\n%s' % (request_method, expires,
|
||||
env['PATH_INFO']), sha1).hexdigest()
|
||||
|
||||
def _invalid(self, env, start_response):
|
||||
|
Loading…
Reference in New Issue
Block a user