Mercurial > public > sg101
view core/management/commands/ssl_images.py @ 860:6e20d3b1e7c2
Improve command line support for ssl_images.
author | Brian Neal <bgneal@gmail.com> |
---|---|
date | Tue, 25 Nov 2014 21:02:50 -0600 |
parents | fca9281c535c |
children | 0ffdb434d2dd |
line wrap: on
line source
""" 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)