From caa71fa3b08278cc95413c451728f55a02a81f09 Mon Sep 17 00:00:00 2001
From: AdamSelene <AdamSelene@users.noreply.github.com>
Date: Wed, 21 Jun 2017 10:39:13 +0200
Subject: [PATCH] Add 'shortener_keys' to configuration

---
 README.md           | 25 +++++++++++++++++++++++++
 rollbar/__init__.py |  5 ++++-
 2 files changed, 29 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 79f5affd..a0680ee8 100644
--- a/README.md
+++ b/README.md
@@ -490,6 +490,31 @@ Default: `True`
 <dt>verify_https
 </dt>
 <dd>If `True`, network requests will fail unless encountering a valid certificate. Default `True`.
+</dd>
+<dt>shortener_keys
+</dt>
+<dd>A list of key prefixes (as tuple) to apply our shortener transform to. Added to built-in list:
+
+```
+[
+    ('body', 'request', 'POST'),
+    ('body', 'request', 'json')
+]
+```
+
+If `locals.enabled` is `True`, extra keys are also automatically added:
+
+```
+[
+    ('body', 'trace', 'frames', '*', 'code'),
+    ('body', 'trace', 'frames', '*', 'args', '*'),
+    ('body', 'trace', 'frames', '*', 'kwargs', '*'),
+    ('body', 'trace', 'frames', '*', 'locals', '*')
+]
+```
+
+Default: `[]`
+
 </dd>
 </dd>
 </dl>
diff --git a/rollbar/__init__.py b/rollbar/__init__.py
index 3767c62f..651b7a2f 100644
--- a/rollbar/__init__.py
+++ b/rollbar/__init__.py
@@ -236,7 +236,8 @@ def _get_pylons_request():
         'sizes': DEFAULT_LOCALS_SIZES,
         'whitelisted_types': []
     },
-    'verify_https': True
+    'verify_https': True,
+    'shortener_keys': []
 }
 
 # Set in init()
@@ -320,6 +321,8 @@ def init(access_token, environment='production', **kw):
         shortener_keys.append(('body', 'trace', 'frames', '*', 'kwargs', '*'))
         shortener_keys.append(('body', 'trace', 'frames', '*', 'locals', '*'))
 
+    shortener_keys.extend(SETTINGS['shortener_keys'])
+
     shortener = ShortenerTransform(safe_repr=SETTINGS['locals']['safe_repr'],
                                    keys=shortener_keys,
                                    **SETTINGS['locals']['sizes'])