diff core/management/commands/ssl_images.py @ 868:64a5acb83937

WIP on ssl_images command.
author Brian Neal <bgneal@gmail.com>
date Mon, 15 Dec 2014 19:51:12 -0600
parents 98adae6e13a1
children ee56c8c8cf98
line wrap: on
line diff
--- a/core/management/commands/ssl_images.py	Thu Dec 04 20:32:43 2014 -0600
+++ b/core/management/commands/ssl_images.py	Mon Dec 15 19:51:12 2014 -0600
@@ -11,6 +11,7 @@
 import os.path
 import re
 import signal
+import urlparse
 
 from django.core.management.base import NoArgsCommand, CommandError
 from django.conf import settings
@@ -26,6 +27,7 @@
 IMAGE_LINK_RE = re.compile(markdown.inlinepatterns.IMAGE_LINK_RE)
 IMAGE_REF_RE = re.compile(markdown.inlinepatterns.IMAGE_REFERENCE_RE)
 
+SG101_HOSTS = set(['www.surfguitar101.com', 'surfguitar101.com'])
 MODEL_CHOICES = ['comments', 'posts']
 
 quit_flag = False
@@ -46,8 +48,46 @@
     logger.addHandler(handler)
 
 
+def save_image_to_cloud(src):
+    # TODO
+    return src
+
+
 def replace_image_markup(match):
-    return match.group(0)
+    src_parts = match.group(9).split()
+    if src_parts:
+        src = src_parts[0]
+        if src[0] == "<" and src[-1] == ">":
+            src = src[1:-1]
+    else:
+        src = ''
+
+    title = ''
+    if len(src_parts) > 1:
+        title = " ".join(src_parts[1:])
+    alt = match.group(2)
+
+    new_src = ''
+    if src:
+        r = urlparse.urlparse(src)
+        if r.scheme == 'http':
+            if r.hostname in SG101_HOSTS:
+                new_src = r.path    # convert to relative path
+            else:
+                new_src = save_image_to_cloud(src)
+        elif r.scheme == 'https':
+            new_src = src       # already https, accept it as-is
+
+    if new_src:
+        if title:
+            s = u'![{alt}]({src} "{title}")'.format(alt=alt, src=new_src, title=title)
+        else:
+            s = u'![{alt}]({src})'.format(alt=alt, src=new_src)
+    else:
+        # something's messed up, convert to a link using original src
+        s = u'[{alt}]({src})'.format(alt=alt, src=src)
+
+    return s
 
 
 def process_post(text):