X-Git-Url: https://code.kerkeslager.com/?p=txt.house;a=blobdiff_plain;f=txt_house%2Fmodels.py;fp=txt_house%2Fmodels.py;h=f394478e72bf67e9514dad71323a8a1ff82034c0;hp=2f57ca04dc020781222df18c1ed46ab1d05823d3;hb=9b232a209ae57cea33a0156a5c4cac7e72edda8f;hpb=3cdbc0aa5864867577563b5b3a699d47cac79770

diff --git a/txt_house/models.py b/txt_house/models.py
index 2f57ca0..f394478 100644
--- 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)