Get editing working

[txt.house] / txt_house / models.py

diff --git a/txt_house/models.py b/txt_house/models.py
index 2f57ca0..f394478 100644 (file)
--- a/txt_house/models.py
+++ b/txt_house/models.py
@@ -9,11 +9,19 @@ class TextFile(models.Model):
     text = models.TextField()
     edit_key = models.CharField(max_length=43)
 
-    def get_absolute_url(self):
-        pk = base64.urlsafe_b64encode(
+    def get_pk(self):
+        return base64.urlsafe_b64encode(
             self.pk.to_bytes(math.ceil(self.pk.bit_length() / 8), 'big'),
         ).decode('utf-8')
 
+    def get_view_url(self):
+        pk = self.get_pk()
+
         return reverse('text-file', kwargs={ 'pk': pk })
 
+    def get_edit_url(self):
+        pk = self.get_pk()
+
+        return reverse('edit', kwargs={ 'pk': pk })
+
 admin.site.register(TextFile)