Mercurial > public > sg101
changeset 862:22a2362a8c9a
Merge with upstream.
author | Brian Neal <bgneal@gmail.com> |
---|---|
date | Mon, 01 Dec 2014 18:39:40 -0600 |
parents | 6e20d3b1e7c2 (diff) e4f8d87c3d30 (current diff) |
children | 0ffdb434d2dd |
files | |
diffstat | 1 files changed, 120 insertions(+), 0 deletions(-) [+] |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/core/management/commands/ssl_images.py Mon Dec 01 18:39:40 2014 -0600 @@ -0,0 +1,120 @@ +""" +ssl_images is a custom manage.py command to convert forum post and comment +images to https. It does this by rewriting the markup: + - Images with src = http://surfguitar101.com/something are rewritten to be + /something. + - Non SG101 images that use http: are downloaded, resized, and uploaded to + an S3 bucket. The src attribute is replaced with the new S3 URL. +""" +import logging +from optparse import make_option +import os.path + +from django.core.management.base import NoArgsCommand, CommandError +from django.conf import settings + +from comments.models import Comment +from forums.models import Post + + +LOGFILE = os.path.join(settings.PROJECT_PATH, 'logs', 'ssl_images.log') +logger = logging.getLogger(__name__) + + +def _setup_logging(): + logger.setLevel(logging.DEBUG) + logger.propagate = False + handler = logging.FileHandler(filename=LOGFILE, encoding='utf-8') + formatter = logging.Formatter('%(asctime)s %(levelname)s %(message)s') + handler.setFormatter(formatter) + logger.addHandler(handler) + + +class CommentFacade(object): + """Wrapper class to provide uniform access to Comments.""" + def __init__(self, comment): + self.comment = comment + + @property + def markdown(self): + return self.comment.comment + + @markdown.setter + def markdown(self, value): + self.comment.comment = value + + +class PostFacade(object): + """Wrapper class to provide uniform access to Forum posts.""" + def __init__(self, post): + self.post = post + + @property + def markdown(self): + return self.post.body + + @markdown.setter + def markdown(self, value): + self.post.body = value + + +class Command(NoArgsCommand): + help = "Rewrite forum posts and comments to not use http for images" + option_list = NoArgsCommand.option_list + ( + make_option('--forums', + action='store_true', + default=False, + help="process forum posts"), + make_option('--comments', + action='store_true', + default=False, + help="process comments"), + make_option('-i', '--i', + type='int', + help="first slice index; the i in [i:j]"), + make_option('-j', '--j', + type='int', + help="second slice index; the j in [i:j]"), + ) + + def handle_noargs(self, **options): + _setup_logging() + logger.info("Starting; arguments received: %s", options) + + do_comments = options['comments'] + do_forums = options['forums'] + if do_comments and do_forums: + raise CommandError("Please specify --forums or --comments, not both") + elif not do_comments and not do_forums: + raise CommandError("Please specify --forums or --comments") + + if do_comments: + qs = Comment.objects.all() + facade = CommentFacade + else: + qs = Post.objects.all() + facade = PostFacade + + i, j = options['i'], options['j'] + + if i is not None and i < 0: + raise CommandError("-i must be >= 0") + if j is not None and j < 0: + raise CommandError("-j must be >= 0") + if j is not None and i is not None and j <= i: + raise CommandError("-j must be > -i") + + if i is not None and j is not None: + qs = qs[i:j] + elif i is not None and j is None: + qs = qs[i:] + elif i is None and j is not None: + qs = qs[:j] + + s = [] + for model in qs.iterator(): + obj = facade(model) + s.append(obj.markdown) + + import pprint + pprint.pprint(s)