Mercurial > hg > contenttransformer
view contenttransformer/app.py @ 13:9d51954e2e68
allow transformers to take kwargs
author | Jeff Hammel <jhammel@mozilla.com> |
---|---|
date | Wed, 30 Jun 2010 22:09:13 -0700 |
parents | a9ddcfc7c4e8 |
children | 7dae51af8db6 |
line wrap: on
line source
import os import sys from fnmatch import fnmatch from mimetypes import guess_type from paste.fileapp import FileApp from pkg_resources import iter_entry_points from transformers import ContentTypeChanger class FileTypeTransformer(object): def __init__(self, *types, **kwargs): """types is a list of two-tuples: glob pattern (string), transformer name (string, name of entry point)""" self.types = types # arguments to the xformers self.kwargs = kwargs self.transformers = transformers() for pattern, transformer_name in self.types: if '/' in transformer_name: continue assert transformer_name in self.transformers def __call__(self, path): """this should return something that is callable with (environ, start_response) to return a response; the transformer thing""" filename = os.path.basename(path) for pattern, transformer_name in self.types: if fnmatch(filename, pattern): content_type, _ = guess_type(filename) content = file(path).read() # transform content type # XXX hack: -> refactor if '/' in transformer_name: return ContentTypeChanger(content, content_type, transformer_name) return self.transformers[transformer_name](content, content_type, **self.kwargs.get(transformer_name, {})) return FileApp(path) def transformers(): transformers = {} # XXX could cache for entry_point in iter_entry_points('content_transformers'): try: transformers[entry_point.name] = entry_point.load() except: raise # XXX return transformers