Hack in sha256 support - needs to be refactored

ivatar/ivataraccount/migrations/0006_auto_20180626_1445.py

@@ -0,0 +1,23 @@
+# Generated by Django 2.0.6 on 2018-06-26 14:45
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('ivataraccount', '0005_auto_20180522_1155'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='confirmedemail',
+            name='digest_sha256',
+            field=models.CharField(max_length=64, null=True),
+        ),
+        migrations.AlterField(
+            model_name='confirmedemail',
+            name='digest',
+            field=models.CharField(max_length=32),
+        ),
+    ]

ivatar/ivataraccount/models.py

@@ -259,7 +259,8 @@ class ConfirmedEmail(BaseAccountModel):
         null=True,
         on_delete=models.deletion.CASCADE,
     )
-    digest = models.CharField(max_length=64)
+    digest = models.CharField(max_length=32)
+    digest_sha256 = models.CharField(max_length=64, null=True)
     objects = ConfirmedEmailManager()
 
     class Meta:  # pylint: disable=too-few-public-methods
@@ -284,6 +285,7 @@ class ConfirmedEmail(BaseAccountModel):
         self.digest = hashlib.md5(
             self.email.strip().lower().encode('utf-8')
         ).hexdigest()
+        self.digest_sha256 = hashlib.sha256(self.email.strip().lower().encode('utf-8')).hexdigest()
         return super().save(force_insert, force_update, using, update_fields)
 
 

ivatar/views.py

@@ -22,6 +22,7 @@ class AvatarImageView(TemplateView):
             # Fetch by digest from mail
             pass
         elif len(kwargs['digest']) == 64:
+            if model.objects.filter(digest=kwargs['digest']).count():
                 # Fetch by digest from OpenID
                 model = ConfirmedOpenId
         else:  # pragma: no cover
@@ -30,6 +31,9 @@ class AvatarImageView(TemplateView):
 
         try:
             obj = model.objects.get(digest=kwargs['digest'])
+        except ObjectDoesNotExist:
+            try:
+                obj = model.objects.get(digest_sha256=kwargs['digest'])
             except ObjectDoesNotExist:
                 # TODO: Use default!?
                 raise Exception('Mail/openid ("%s") does not exist"' %