mirror olimpiada migration (shared db)

[edumed.git] / contact / models.py

diff --git a/contact/models.py b/contact/models.py
index 21d8405..8d89c8d 100644 (file)
--- a/contact/models.py
+++ b/contact/models.py
@@ -1,6 +1,8 @@
 # -*- coding: utf-8 -*-
 # -*- coding: utf-8 -*-

-from django.core.files.storage import FileSystemStorage
+import yaml
+from hashlib import sha1

 from django.db import models
 from django.db import models

+from django.utils.encoding import smart_unicode

 from django.utils.translation import ugettext_lazy as _
 from jsonfield import JSONField
 from . import app_settings
 from django.utils.translation import ugettext_lazy as _
 from jsonfield import JSONField
 from . import app_settings


@@ -13,6 +15,14 @@ class Contact(models.Model):
     form_tag = models.CharField(_('form'), max_length=32, db_index=True)
     body = JSONField(_('body'))
 
     form_tag = models.CharField(_('form'), max_length=32, db_index=True)
     body = JSONField(_('body'))
 

+    @staticmethod
+    def pretty_print(value, for_html=False):
+        if type(value) in (tuple, list, dict):
+            value = yaml.safe_dump(value, allow_unicode=True, default_flow_style=False)
+            if for_html:
+                value = smart_unicode(value).replace(u" ", unichr(160))
+        return value
+

     class Meta:
         ordering = ('-created_at',)
         verbose_name = _('submitted form')
     class Meta:
         ordering = ('-created_at',)
         verbose_name = _('submitted form')


@@ -21,6 +31,11 @@ class Contact(models.Model):
     def __unicode__(self):
         return unicode(self.created_at)
 
     def __unicode__(self):
         return unicode(self.created_at)
 

+    def digest(self):
+        serialized_body = ';'.join(sorted('%s:%s' % item for item in self.body.iteritems()))
+        data = '%s%s%s%s%s' % (self.id, self.contact, serialized_body, self.ip, self.form_tag)
+        return sha1(data).hexdigest()
+

 
 class Attachment(models.Model):
     contact = models.ForeignKey(Contact)
 
 class Attachment(models.Model):
     contact = models.ForeignKey(Contact)


@@ -29,7 +44,7 @@ class Attachment(models.Model):
 
     @models.permalink
     def get_absolute_url(self):
 
     @models.permalink
     def get_absolute_url(self):

-        return ('contact_attachment', [self.contact_id, self.tag])
+        return 'contact_attachment', [self.contact_id, self.tag]

 
 
 __import__(app_settings.FORMS_MODULE)
 
 
 __import__(app_settings.FORMS_MODULE)