From 955231199e22803cf847453962dfecf590189b83 Mon Sep 17 00:00:00 2001
From: Son NK <>
Date: Sat, 4 Jul 2020 23:19:21 +0200
Subject: [PATCH] Add User.default_random_alias_public_domain_id

---
 app/models.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/app/models.py b/app/models.py
index dea94e1f..522c4a13 100644
--- a/app/models.py
+++ b/app/models.py
@@ -2,7 +2,7 @@ import enum
 import random
 import uuid
 from email.utils import formataddr
-from typing import List
+from typing import List, Tuple
 
 import arrow
 import bcrypt
@@ -166,8 +166,18 @@ class User(db.Model, ModelMixin, UserMixin):
     # Fields for WebAuthn
     fido_uuid = db.Column(db.String(), nullable=True, unique=True)
 
+    # the default domain that's used when user creates a new random alias
+    # default_random_alias_domain_id XOR default_random_alias_public_domain_id
     default_random_alias_domain_id = db.Column(
-        db.ForeignKey("custom_domain.id"), nullable=True, default=None
+        db.ForeignKey("custom_domain.id", ondelete="SET NULL"),
+        nullable=True,
+        default=None,
+    )
+
+    default_random_alias_public_domain_id = db.Column(
+        db.ForeignKey("public_domain.id", ondelete="SET NULL"),
+        nullable=True,
+        default=None,
     )
 
     # some users could have lifetime premium