gunicorn/gunicorn/util.py
2010-01-23 15:04:52 +01:00

101 lines
2.5 KiB
Python

# -*- coding: utf-8 -
#
# This file is part of gunicorn released under the MIT license.
# See the NOTICE for more information.
import errno
import fcntl
import select
import socket
import time
timeout_default = object()
CHUNK_SIZE = (16 * 1024)
MAX_BODY = 1024 * (80 + 32)
weekdayname = ['Mon', 'Tue', 'Wed', 'Thu', 'Fri', 'Sat', 'Sun']
monthname = [None,
'Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun',
'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec']
def close_on_exec(fd):
flags = fcntl.fcntl(fd, fcntl.F_GETFD) | fcntl.FD_CLOEXEC
fcntl.fcntl(fd, fcntl.F_SETFL, flags)
def close(sock):
try:
sock.close()
except socket.error:
pass
def read_partial(sock, length):
while True:
try:
ret = select.select([sock.fileno()], [], [], 0)
if ret[0]: break
except select.error, e:
if e[0] == errno.EINTR:
continue
raise
data = sock.recv(length)
return data
def write(sock, data):
buf = ""
buf += data
i = 0
while buf:
try:
bytes = sock.send(buf)
if bytes < len(buf):
buf = buf[bytes:]
continue
return len(data)
except socket.error, e:
if e[0] in (errno.EWOULDBLOCK, errno.EAGAIN):
break
raise
i += 1
def writelines(sock, lines):
for line in list(lines):
write(sock, line)
def normalize_name(name):
return "-".join([w.lower().capitalize() for w in name.split("-")])
def import_app(module):
parts = module.rsplit(":", 1)
if len(parts) == 1:
module, obj = module, "application"
else:
module, obj = parts[0], parts[1]
mod = __import__(module)
parts = module.split(".")
for p in parts[1:]:
mod = getattr(mod, p, None)
if mod is None:
raise ImportError("Failed to import: %s" % module)
app = getattr(mod, obj, None)
if app is None:
raise ImportError("Failed to find application object: %r" % obj)
if not callable(app):
raise TypeError("Application object must be callable.")
return app
def http_date(timestamp=None):
"""Return the current date and time formatted for a message header."""
if timestamp is None:
timestamp = time.time()
year, month, day, hh, mm, ss, wd, y, z = time.gmtime(timestamp)
s = "%s, %02d %3s %4d %02d:%02d:%02d GMT" % (
weekdayname[wd],
day, monthname[month], year,
hh, mm, ss)
return s