From d7239d665e6781d7b84ecfde3fd5d7d1831de284 Mon Sep 17 00:00:00 2001
From: Chris Czub <chris.czub@gmail.com>
Date: Mon, 9 Nov 2015 15:07:08 -0500
Subject: Add new contentviews' content type mappings

---
 libmproxy/contentviews.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'libmproxy')

diff --git a/libmproxy/contentviews.py b/libmproxy/contentviews.py
index 8d5aadfc..aa2082d1 100644
--- a/libmproxy/contentviews.py
+++ b/libmproxy/contentviews.py
@@ -520,8 +520,18 @@ def add(obj):
         if i.name == obj.name:
             raise ContentViewException("Duplicate view: " + obj.name)
 
+    for prompt in view_prompts:
+        if prompt[1] == obj.prompt[1]:
+            raise ContentViewException("Duplicate view shortcut: " + obj.prompt[1])
+
     views.append(obj)
 
+    for ct in obj.content_types:
+        l = content_types_map.setdefault(ct, [])
+        l.append(obj)
+
+    view_prompts.append(obj.prompt)
+
 
 def get(name):
     for i in views:
-- 
cgit v1.2.3