1 files changed, 29 insertions, 6 deletions
diff --git a/main/models.py b/main/models.py
index 15a9266..4f2d64e 100644
--- a/main/models.py
+++ b/main/models.py
@@ -1,3 +1,9 @@
+from base64 import b64decode
+from datetime import datetime
+from itertools import groupby
+from pgpdump import BinaryData
+import pytz
+
 from django.db import models
 from django.contrib.auth.models import User
 from django.contrib.sites.models import Site
@@ -5,10 +11,6 @@ from django.contrib.sites.models import Site
 from .fields import PositiveBigIntegerField, PGPKeyField
 from .utils import cache_function, make_choice, set_created_field
 
-from datetime import datetime
-from itertools import groupby
-import pytz
-
 
 class UserProfile(models.Model):
     notify = models.BooleanField(
@@ -183,8 +185,29 @@ class Package(models.Model):
         domain = Site.objects.get_current().domain
         return '%s://%s%s' % (proto, domain, self.get_absolute_url())
 
-    def is_signed(self):
-        return bool(self.pgp_signature)
+    @property
+    @cache_function(15)
+    def signature(self):
+        try:
+            data = b64decode(self.pgp_signature)
+        except TypeError:
+            return None
+        data = BinaryData(data)
+        packets = list(data.packets())
+        return packets[0]
+
+    @property
+    @cache_function(15)
+    def signer(self):
+        sig = self.signature
+        if sig and sig.key_id:
+            try:
+                user = User.objects.get(
+                        userprofile__pgp_key__endswith=sig.key_id)
+            except User.DoesNotExist:
+                user = None
+            return user
+        return None
 
     _maintainers = None