add update forms

[edumed.git] / contact / models.py

diff --git a/contact/models.py b/contact/models.py
index dbc48b9..f07b33a 100644 (file)
--- a/contact/models.py
+++ b/contact/models.py
@@ -3,6 +3,7 @@ import yaml
 from hashlib import sha1
 from django.db import models
 from django.utils.encoding import smart_unicode, force_str
+from django.db.models import permalink
 from django.utils.translation import ugettext_lazy as _
 from jsonfield import JSONField
 from . import app_settings
@@ -37,6 +38,13 @@ class Contact(models.Model):
         data = force_str(data)
         return sha1(data).hexdigest()
 
+    @permalink
+    def update_url(self):
+        from contact.forms import update_forms, contact_forms
+        form_class = update_forms.get(self.form_tag, contact_forms.get(self.form_tag))
+        confirmation = form_class.confirmation_class.objects.get(contact=self)
+        return 'edit_form', [], {'form_tag': self.form_tag, 'contact_id': self.id, 'key': confirmation.key}
+
 
 class Attachment(models.Model):
     contact = models.ForeignKey(Contact)