Allow users to specify pygments style

honza · Apr 5, 2012 · 05f9ff3 · 05f9ff3
1 parent aa8f13e
commit 05f9ff3
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -29,6 +29,7 @@ Env settings
   mongo`)
 * `SPRUNGE_MONGO_DB_NAME` - the string after the last slash on
   `SPRUNGE_MONGO_URI`
+* `SPRUNGE_STYLE` - Pygments style (defaults to `default`)
 
 
 Set these with:

diff --git a/sprunge/app.py b/sprunge/app.py
@@ -72,7 +72,7 @@ def snip(uid):
         except:
             lexer = pygments.lexers.TextLexer()
         formatter = HtmlFormatter(full=True,
-                style='borland', lineanchors='n', linenos='inline',
+                style=STYLE, linenos='inline',
                 encoding='utf-8')
         return highlight(data['content'], lexer, formatter)
     else:

diff --git a/sprunge/settings.py b/sprunge/settings.py
@@ -15,3 +15,8 @@ def get_value(key):
 NAME = get_value('NAME')  # the POST payload key
 MONGO_URI = get_value('MONGO_URI')
 MONGO_DB_NAME = get_value('MONGO_DB_NAME')
+
+try:
+    STYLE = os.environ['SPRUNGE_STYLE']
+except:
+    STYLE = 'default'