mirror of
https://github.com/simple-login/app.git
synced 2024-11-16 17:08:30 +01:00
9646f84b79
* Only send new alias events on user creation (cherry picked from commit ab8f998dd40dc83c8f8a528a156ba50eae376aaf) * Trigger a sync when a new partner user is created * Improve tests * Move it to the partner_utils --------- Co-authored-by: Carlos Quintana <74399022+cquintana92@users.noreply.github.com>
53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
from typing import Optional
|
|
|
|
import arrow
|
|
from arrow import Arrow
|
|
|
|
from app import config
|
|
from app.models import PartnerUser, PartnerSubscription, User, Job
|
|
from app.user_audit_log_utils import emit_user_audit_log, UserAuditLogAction
|
|
|
|
|
|
def create_partner_user(
|
|
user: User, partner_id: int, partner_email: str, external_user_id: str
|
|
) -> PartnerUser:
|
|
instance = PartnerUser.create(
|
|
user_id=user.id,
|
|
partner_id=partner_id,
|
|
partner_email=partner_email,
|
|
external_user_id=external_user_id,
|
|
)
|
|
Job.create(
|
|
name=config.JOB_SEND_ALIAS_CREATION_EVENTS,
|
|
payload={"user_id": user.id},
|
|
run_at=arrow.now(),
|
|
)
|
|
emit_user_audit_log(
|
|
user=user,
|
|
action=UserAuditLogAction.LinkAccount,
|
|
message=f"Linked account to partner_id={partner_id} | partner_email={partner_email} | external_user_id={external_user_id}",
|
|
)
|
|
|
|
return instance
|
|
|
|
|
|
def create_partner_subscription(
|
|
partner_user: PartnerUser,
|
|
expiration: Optional[Arrow],
|
|
msg: Optional[str] = None,
|
|
) -> PartnerSubscription:
|
|
instance = PartnerSubscription.create(
|
|
partner_user_id=partner_user.id,
|
|
end_at=expiration,
|
|
)
|
|
|
|
message = "User upgraded through partner subscription"
|
|
if msg:
|
|
message += f" | {msg}"
|
|
emit_user_audit_log(
|
|
user=partner_user.user,
|
|
action=UserAuditLogAction.Upgrade,
|
|
message=message,
|
|
)
|
|
|
|
return instance
|