common.py
124 lines
| 4.1 KiB
| text/x-python
|
PythonLexer
|
r2391 | # hgweb/common.py - Utility functions needed by hgweb_mod and hgwebdir_mod | ||
|
r2356 | # | ||
# Copyright 21 May 2005 - (c) 2005 Jake Edge <jake@edge2.net> | ||||
|
r2859 | # Copyright 2005, 2006 Matt Mackall <mpm@selenic.com> | ||
|
r2356 | # | ||
# This software may be used and distributed according to the terms | ||||
# of the GNU General Public License, incorporated herein by reference. | ||||
|
r5561 | import errno, mimetypes, os | ||
|
r5993 | HTTP_OK = 200 | ||
HTTP_BAD_REQUEST = 400 | ||||
|
r6926 | HTTP_UNAUTHORIZED = 401 | ||
|
r7029 | HTTP_FORBIDDEN = 403 | ||
|
r5993 | HTTP_NOT_FOUND = 404 | ||
|
r6926 | HTTP_METHOD_NOT_ALLOWED = 405 | ||
|
r5993 | HTTP_SERVER_ERROR = 500 | ||
|
r5561 | class ErrorResponse(Exception): | ||
def __init__(self, code, message=None): | ||||
Exception.__init__(self) | ||||
self.code = code | ||||
|
r6924 | if message is not None: | ||
|
r5563 | self.message = message | ||
|
r5561 | else: | ||
|
r5563 | self.message = _statusmessage(code) | ||
def _statusmessage(code): | ||||
from BaseHTTPServer import BaseHTTPRequestHandler | ||||
responses = BaseHTTPRequestHandler.responses | ||||
return responses.get(code, ('Error', 'Unknown error'))[0] | ||||
|
r5760 | |||
|
r5563 | def statusmessage(code): | ||
return '%d %s' % (code, _statusmessage(code)) | ||||
|
r2356 | |||
def get_mtime(repo_path): | ||||
|
r3853 | store_path = os.path.join(repo_path, ".hg") | ||
if not os.path.isdir(os.path.join(store_path, "data")): | ||||
store_path = os.path.join(store_path, "store") | ||||
cl_path = os.path.join(store_path, "00changelog.i") | ||||
if os.path.exists(cl_path): | ||||
|
r2356 | return os.stat(cl_path).st_mtime | ||
else: | ||||
|
r3853 | return os.stat(store_path).st_mtime | ||
|
r2356 | |||
|
r2514 | def staticfile(directory, fname, req): | ||
|
r5930 | """return a file inside directory with guessed Content-Type header | ||
|
r2356 | |||
fname always uses '/' as directory separator and isn't allowed to | ||||
contain unusual path components. | ||||
|
r5930 | Content-Type is guessed using the mimetypes module. | ||
|
r2356 | Return an empty string if fname is illegal or file not found. | ||
""" | ||||
parts = fname.split('/') | ||||
path = directory | ||||
for part in parts: | ||||
if (part in ('', os.curdir, os.pardir) or | ||||
os.sep in part or os.altsep is not None and os.altsep in part): | ||||
return "" | ||||
path = os.path.join(path, part) | ||||
try: | ||||
os.stat(path) | ||||
ct = mimetypes.guess_type(path)[0] or "text/plain" | ||||
|
r5993 | req.respond(HTTP_OK, ct, length = os.path.getsize(path)) | ||
|
r3244 | return file(path, 'rb').read() | ||
|
r5561 | except TypeError: | ||
|
r5993 | raise ErrorResponse(HTTP_SERVER_ERROR, 'illegal file name') | ||
|
r5561 | except OSError, err: | ||
if err.errno == errno.ENOENT: | ||||
|
r5993 | raise ErrorResponse(HTTP_NOT_FOUND) | ||
|
r5561 | else: | ||
|
r5993 | raise ErrorResponse(HTTP_SERVER_ERROR, err.strerror) | ||
|
r3276 | |||
def style_map(templatepath, style): | ||||
"""Return path to mapfile for a given style. | ||||
Searches mapfile in the following locations: | ||||
1. templatepath/style/map | ||||
2. templatepath/map-style | ||||
3. templatepath/map | ||||
""" | ||||
locations = style and [os.path.join(style, "map"), "map-"+style] or [] | ||||
locations.append("map") | ||||
|
r7107 | if isinstance(templatepath, str): | ||
templatepath = [templatepath] | ||||
for path in templatepath: | ||||
for location in locations: | ||||
mapfile = os.path.join(path, location) | ||||
if os.path.isfile(mapfile): | ||||
return mapfile | ||||
|
r3276 | raise RuntimeError("No hgweb templates found in %r" % templatepath) | ||
|
r4462 | def paritygen(stripecount, offset=0): | ||
"""count parity of horizontal stripes for easier reading""" | ||||
if stripecount and offset: | ||||
# account for offset, e.g. due to building the list in reverse | ||||
count = (stripecount + offset) % stripecount | ||||
parity = (stripecount + offset) / stripecount & 1 | ||||
else: | ||||
count = 0 | ||||
parity = 0 | ||||
while True: | ||||
yield parity | ||||
count += 1 | ||||
if stripecount and count >= stripecount: | ||||
parity = 1 - parity | ||||
count = 0 | ||||
|
r6122 | def countgen(start=0, step=1): | ||
"""count forever -- useful for line numbers""" | ||||
while True: | ||||
yield start | ||||
start += step | ||||
|
r5779 | def get_contact(config): | ||
"""Return repo contact information or empty string. | ||||
web.contact is the primary source, but if that is not set, try | ||||
ui.username or $EMAIL as a fallback to display something useful. | ||||
""" | ||||
return (config("web", "contact") or | ||||
config("ui", "username") or | ||||
os.environ.get("EMAIL") or "") | ||||