X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/7094d00cd82535cfd1db00e7bd85e2a01161da83..07689901a9bf30daeccf8a1ceb7193fe771eb3ac:/apps/wiki_img/models.py

diff --git a/apps/wiki_img/models.py b/apps/wiki_img/models.py
new file mode 100644
index 00000000..dd16a87d
--- /dev/null
+++ b/apps/wiki_img/models.py
@@ -0,0 +1,29 @@
+# -*- coding: utf-8 -*-
+#
+# This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#
+from django.db import models
+from django.contrib.auth.models import User
+from django.utils.translation import ugettext_lazy as _
+from dvcs.models import Document
+
+
+class ImageDocument(models.Model):
+    slug = models.SlugField(_('slug'), max_length=120)
+    name = models.CharField(_('name'), max_length=120)
+    image = models.ImageField(_('image'), upload_to='wiki_img')
+    doc = models.OneToOneField(Document, null=True, blank=True)
+    creator = models.ForeignKey(User, null=True, blank=True)
+
+    @staticmethod
+    def listener_initial_commit(sender, instance, created, **kwargs):
+        if created:
+            instance.doc = Document.objects.create(creator=instance.creator)
+            instance.save()
+
+    def __unicode__(self):
+        return self.name
+
+
+models.signals.post_save.connect(ImageDocument.listener_initial_commit, sender=ImageDocument)