Refactor apps to be in repo folder

core/models.py

@@ -0,0 +1,51 @@
+import requests
+
+from io import BytesIO
+
+from django.conf import settings
+from django.core.files.uploadedfile import InMemoryUploadedFile
+from django.db import models, transaction
+
+from django_images.models import Image as BaseImage, Thumbnail
+from taggit.managers import TaggableManager
+
+from users.models import User
+
+
+class ImageManager(models.Manager):
+    # FIXME: Move this into an asynchronous task
+    def create_for_url(self, url):
+        file_name = url.split("/")[-1].split('#')[0].split('?')[0]
+        buf = BytesIO()
+        response = requests.get(url)
+        buf.write(response.content)
+        obj = InMemoryUploadedFile(buf, 'image', file_name,
+                                   None, buf.tell(), None)
+        # create the image and its thumbnails in one transaction, removing
+        # a chance of getting Database into a inconsistent state when we
+        # try to create thumbnails one by one later
+        image = self.create(image=obj)
+        for size in settings.IMAGE_SIZES.keys():
+            Thumbnail.objects.get_or_create_at_size(image.pk, size)
+        return image
+
+
+class Image(BaseImage):
+    objects = ImageManager()
+
+    class Meta:
+        proxy = True
+
+
+class Pin(models.Model):
+    submitter = models.ForeignKey(User)
+    url = models.URLField(null=True)
+    origin = models.URLField(null=True)
+    description = models.TextField(blank=True, null=True)
+    image = models.ForeignKey(Image, related_name='pin')
+    published = models.DateTimeField(auto_now_add=True)
+    tags = TaggableManager()
+
+    def __unicode__(self):
+        return '%s - %s' % (self.submitter, self.published)
+