From 651486621dd79025a1b0ea08abe786b50a9604ad Mon Sep 17 00:00:00 2001
From: Philipp Hagemeister <phihag@phihag.de>
Date: Tue, 1 Apr 2014 00:25:11 +0200
Subject: [PATCH] [comedycentral] Allow URLs with query parts (fixes #2661)

---
 test/test_all_urls.py                 | 3 +++
 youtube_dl/extractor/comedycentral.py | 5 +++--
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/test/test_all_urls.py b/test/test_all_urls.py
index dffe3f9585..ed041ffda0 100644
--- a/test/test_all_urls.py
+++ b/test/test_all_urls.py
@@ -150,6 +150,9 @@ def test_ComedyCentralShows(self):
         self.assertMatch(
             'http://thecolbertreport.cc.com/videos/29w6fx/-realhumanpraise-for-fox-news',
             ['ComedyCentralShows'])
+        self.assertMatch(
+            'http://thecolbertreport.cc.com/videos/gh6urb/neil-degrasse-tyson-pt--1?xrs=eml_col_031114',
+            ['ComedyCentralShows'])
 
 if __name__ == '__main__':
     unittest.main()
diff --git a/youtube_dl/extractor/comedycentral.py b/youtube_dl/extractor/comedycentral.py
index cbc212065c..ed0c6ea2ca 100644
--- a/youtube_dl/extractor/comedycentral.py
+++ b/youtube_dl/extractor/comedycentral.py
@@ -45,10 +45,11 @@ class ComedyCentralShowsIE(InfoExtractor):
                           (?P<clip>
                               (?:videos/[^/]+/(?P<videotitle>[^/?#]+))
                               |(the-colbert-report-(videos|collections)/(?P<clipID>[0-9]+)/[^/]*/(?P<cntitle>.*?))
-                              |(watch/(?P<date>[^/]*)/(?P<tdstitle>.*)))|
+                              |(watch/(?P<date>[^/]*)/(?P<tdstitle>.*))
+                          )|
                           (?P<interview>
                               extended-interviews/(?P<interID>[0-9a-z]+)/(?:playlist_tds_extended_)?(?P<interview_title>.*?)(/.*?)?)))
-                     $'''
+                     (?:[?#].*|$)'''
     _TEST = {
         'url': 'http://thedailyshow.cc.com/watch/thu-december-13-2012/kristen-stewart',
         'md5': '4e2f5cb088a83cd8cdb7756132f9739d',