Merge pull request #3 from CptLemming/master

Capture attributes on HEAD tag. Fixes #2 issue.
mishbahr · Oct 16, 2015 · 1b8b8e4 · 1b8b8e4
2 parents 2d33bd5 + f9ae1cf
commit 1b8b8e4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/AUTHORS.rst b/AUTHORS.rst
@@ -16,5 +16,4 @@ Development Lead
 Contributors
 ------------
 
-None yet. Why not be the first?
-
+* Ashley Wilson <scifilem@gmail.com>
diff --git a/responsive/middleware.py b/responsive/middleware.py
@@ -56,8 +56,8 @@ def process_response(self, request, response):
                 'cookie_age': 60 * 60 * 24 * settings.RESPONSIVE_COOKIE_AGE,  # convert to secs
                 'cookie_expires': expires.strftime('%a, %d %b %Y %H:%M:%S GMT')
             })
-            pattern = re.compile(b'<head>', re.IGNORECASE)
-            response.content = pattern.sub(b'<head>' + smart_bytes(snippet), response.content)
+            pattern = re.compile(b'<head(.*?)>', re.IGNORECASE)
+            response.content = pattern.sub(b'<head\g<1>>' + smart_bytes(snippet), response.content)
 
             if response.get('Content-Length', None):
                 response['Content-Length'] = len(response.content)