Mercurial > public > sg101
annotate gpp/forums/attachments.py @ 290:64c188a9d31f
Adding a legacy app to contain management commands to convert the old data to the new database format. This first commit has the import_old_users command.
author | Brian Neal <bgneal@gmail.com> |
---|---|
date | Fri, 24 Dec 2010 05:28:58 +0000 (2010-12-24) |
parents | 72fd300685d5 |
children |
rev | line source |
---|---|
bgneal@285 | 1 """ |
bgneal@285 | 2 This module contains a class for handling attachments on forum posts. |
bgneal@285 | 3 """ |
bgneal@285 | 4 from oembed.models import Oembed |
bgneal@285 | 5 from forums.models import Attachment |
bgneal@285 | 6 |
bgneal@285 | 7 |
bgneal@285 | 8 class AttachmentProcessor(object): |
bgneal@285 | 9 """ |
bgneal@285 | 10 This class is aggregated by various form classes to handle |
bgneal@285 | 11 attachments on forum posts. New posts can receive attachments and edited |
bgneal@285 | 12 posts can have their attachments replaced, augmented, or deleted. |
bgneal@285 | 13 |
bgneal@285 | 14 """ |
bgneal@285 | 15 def __init__(self, ids): |
bgneal@285 | 16 """ |
bgneal@285 | 17 This class is constructed with a list of Oembed ids. We retrieve the |
bgneal@285 | 18 actual Oembed objects associated with these keys for use in subsequent |
bgneal@285 | 19 operations. |
bgneal@286 | 20 |
bgneal@285 | 21 """ |
bgneal@285 | 22 # ensure all ids are integers |
bgneal@285 | 23 self.pks = [] |
bgneal@285 | 24 for pk in ids: |
bgneal@285 | 25 try: |
bgneal@285 | 26 pk = int(pk) |
bgneal@285 | 27 except ValueError: |
bgneal@285 | 28 continue |
bgneal@285 | 29 self.pks.append(pk) |
bgneal@285 | 30 |
bgneal@285 | 31 self.embeds = [] |
bgneal@285 | 32 if self.pks: |
bgneal@285 | 33 self.embeds = Oembed.objects.in_bulk(self.pks) |
bgneal@285 | 34 |
bgneal@285 | 35 def save_attachments(self, post): |
bgneal@285 | 36 """ |
bgneal@285 | 37 Create and save attachments to the supplied post object. |
bgneal@285 | 38 Any existing attachments on the post are removed first. |
bgneal@285 | 39 |
bgneal@285 | 40 """ |
bgneal@285 | 41 post.attachments.clear() |
bgneal@285 | 42 |
bgneal@285 | 43 for n, pk in enumerate(self.pks): |
bgneal@285 | 44 attachment = Attachment(post=post, embed=self.embeds[pk], order=n) |
bgneal@285 | 45 attachment.save() |
bgneal@286 | 46 |
bgneal@286 | 47 def has_attachments(self): |
bgneal@286 | 48 """ |
bgneal@286 | 49 Return true if we have valid pending attachments. |
bgneal@286 | 50 |
bgneal@286 | 51 """ |
bgneal@286 | 52 return len(self.embeds) > 0 |
bgneal@286 | 53 |
bgneal@286 | 54 def get_ids(self): |
bgneal@286 | 55 """ |
bgneal@286 | 56 Return the list of Oembed ids. |
bgneal@286 | 57 |
bgneal@286 | 58 """ |
bgneal@286 | 59 return self.pks |