[svn] Name change, round 4 (rename SVN root folder).

1 files changed, 73 insertions, 0 deletions

diff --git a/pygments/formatters/__init__.py b/pygments/formatters/__init__.py
new file mode 100644
index 00000000..b701efb4
--- /dev/null
+++ b/pygments/formatters/__init__.py
@@ -0,0 +1,73 @@
+# -*- coding: utf-8 -*-
+"""
+    pygments.formatters
+    ~~~~~~~~~~~~~~~~~~
+
+    Pygments formatters.
+
+    :copyright: 2006 by Georg Brandl, Armin Ronacher.
+    :license: GNU LGPL, see LICENSE for more details.
+"""
+import os.path
+from pygments.formatters.html import HtmlFormatter
+from pygments.formatters.terminal import TerminalFormatter
+from pygments.formatters.latex import LatexFormatter
+from pygments.formatters.bbcode import BBCodeFormatter
+from pygments.formatters.other import NullFormatter, RawTokenFormatter
+
+
+def _doc_desc(obj):
+    res = ''
+    for line in obj.__doc__.strip().splitlines():
+        if line.strip(): res += line.strip() + " "
+        else: break
+    return res
+
+
+#: Map formatter classes to ``(longname, names, file extensions, descr)``.
+FORMATTERS = {
+    HtmlFormatter:        ('HTML', ('html',), ('.htm', '.html'),
+                           _doc_desc(HtmlFormatter)),
+    TerminalFormatter:    ('Terminal', ('terminal', 'console'), (),
+                           _doc_desc(TerminalFormatter)),
+    LatexFormatter:       ('LaTeX', ('latex', 'tex'), ('.tex',),
+                           _doc_desc(LatexFormatter)),
+    RawTokenFormatter:    ('Raw tokens', ('raw', 'tokens'), ('.raw',),
+                           _doc_desc(RawTokenFormatter)),
+    NullFormatter:        ('Text only', ('text', 'null'), ('.txt',),
+                           _doc_desc(NullFormatter)),
+    BBCodeFormatter:      ('BBcode', ('bbcode', 'bb'), (),
+                           _doc_desc(BBCodeFormatter))
+}
+
+
+_formatter_cache = {}
+
+def _init_formatter_cache():
+    if _formatter_cache: return
+    for cls, info in FORMATTERS.iteritems():
+        for alias in info[1]:
+            _formatter_cache[alias] = cls
+        for ext in info[2]:
+            _formatter_cache["/"+ext] = cls
+
+
+def get_formatter_by_name(name, **options):
+    _init_formatter_cache()
+    cls = _formatter_cache.get(name, None)
+    if not cls:
+        raise ValueError("No formatter found for name %r" % name)
+    return cls(**options)
+
+
+def get_formatter_for_filename(fn, **options):
+    _init_formatter_cache()
+    # try by filename extension
+    cls = _formatter_cache.get("/"+os.path.splitext(fn)[1], None)
+    if cls:
+        return cls(**options)
+    # try by whole file name
+    cls = _formatter_cache.get("/"+os.path.basename(fn), None)
+    if not cls:
+        raise ValueError("No formatter found for file name %r" % fn)
+    return cls(**options)