Add referral name
This commit is contained in:
parent
7ceb9440de
commit
3ce3a05c7b
|
@ -158,7 +158,9 @@ class User(db.Model, ModelMixin, UserMixin):
|
||||||
db.Boolean, default=True, nullable=False, server_default="1"
|
db.Boolean, default=True, nullable=False, server_default="1"
|
||||||
)
|
)
|
||||||
|
|
||||||
referral_id = db.Column(db.ForeignKey("referral.id"), nullable=True, default=None)
|
referral_id = db.Column(
|
||||||
|
db.ForeignKey("referral.id", ondelete="SET NULL"), nullable=True, default=None
|
||||||
|
)
|
||||||
|
|
||||||
referral = db.relationship("Referral", foreign_keys=[referral_id])
|
referral = db.relationship("Referral", foreign_keys=[referral_id])
|
||||||
|
|
||||||
|
@ -1153,6 +1155,7 @@ class Referral(db.Model, ModelMixin):
|
||||||
"""Referral code so user can invite others"""
|
"""Referral code so user can invite others"""
|
||||||
|
|
||||||
user_id = db.Column(db.ForeignKey(User.id, ondelete="cascade"), nullable=False)
|
user_id = db.Column(db.ForeignKey(User.id, ondelete="cascade"), nullable=False)
|
||||||
|
name = db.Column(db.String(512), nullable=True, default=None)
|
||||||
|
|
||||||
code = db.Column(db.String(128), unique=True, nullable=False)
|
code = db.Column(db.String(128), unique=True, nullable=False)
|
||||||
|
|
||||||
|
|
13
server.py
13
server.py
|
@ -51,6 +51,7 @@ from app.models import (
|
||||||
DeletedAlias,
|
DeletedAlias,
|
||||||
Contact,
|
Contact,
|
||||||
EmailLog,
|
EmailLog,
|
||||||
|
Referral,
|
||||||
)
|
)
|
||||||
from app.monitor.base import monitor_bp
|
from app.monitor.base import monitor_bp
|
||||||
from app.oauth.base import oauth_bp
|
from app.oauth.base import oauth_bp
|
||||||
|
@ -216,6 +217,18 @@ def fake_data():
|
||||||
DeletedAlias.create(user_id=user.id, email="d2@ab.cd")
|
DeletedAlias.create(user_id=user.id, email="d2@ab.cd")
|
||||||
db.session.commit()
|
db.session.commit()
|
||||||
|
|
||||||
|
referral = Referral.create(user_id=user.id, code="REFCODE", name="First referral")
|
||||||
|
db.session.commit()
|
||||||
|
|
||||||
|
User.create(
|
||||||
|
email="winston@continental.com",
|
||||||
|
name="Winston",
|
||||||
|
password="password",
|
||||||
|
activated=True,
|
||||||
|
referral_id=referral.id,
|
||||||
|
)
|
||||||
|
db.session.commit()
|
||||||
|
|
||||||
|
|
||||||
@login_manager.user_loader
|
@login_manager.user_loader
|
||||||
def load_user(user_id):
|
def load_user(user_id):
|
||||||
|
|
Loading…
Reference in a new issue