mirror of
https://github.com/onyx-dot-app/onyx.git
synced 2026-10-23 03:42:59 -05:00
pablodanswer
20df20ae51
* add vespa multi tenancy * k * formatting * Billing (#2667) * k * data -> control * nit * nit: error handling * auth + app * nit: color standardization * nit * nit: typing * k * k * feat: functional upgrading * feat: add block for downgrading to seats < active users * add auth * remove accomplished todo + prints * nit * tiny nit * nit: centralize security * add tenant expulsion/gating + invite user -> increment billing seat no. * add cloud configs * k * k * nit: update * k * k * k * k * nit
85 lines
2.5 KiB
Python
85 lines
2.5 KiB
Python
import json
|
|
import smtplib
|
|
from datetime import datetime
|
|
from email.mime.multipart import MIMEMultipart
|
|
from email.mime.text import MIMEText
|
|
from textwrap import dedent
|
|
from typing import Any
|
|
|
|
from danswer.configs.app_configs import SMTP_PASS
|
|
from danswer.configs.app_configs import SMTP_PORT
|
|
from danswer.configs.app_configs import SMTP_SERVER
|
|
from danswer.configs.app_configs import SMTP_USER
|
|
from danswer.configs.app_configs import WEB_DOMAIN
|
|
from danswer.db.models import User
|
|
|
|
|
|
class DateTimeEncoder(json.JSONEncoder):
|
|
"""Custom JSON encoder that converts datetime objects to ISO format strings."""
|
|
|
|
def default(self, obj: Any) -> Any:
|
|
if isinstance(obj, datetime):
|
|
return obj.isoformat()
|
|
return super().default(obj)
|
|
|
|
|
|
def get_json_line(
|
|
json_dict: dict[str, Any], encoder: type[json.JSONEncoder] = DateTimeEncoder
|
|
) -> str:
|
|
"""
|
|
Convert a dictionary to a JSON string with datetime handling, and add a newline.
|
|
|
|
Args:
|
|
json_dict: The dictionary to be converted to JSON.
|
|
encoder: JSON encoder class to use, defaults to DateTimeEncoder.
|
|
|
|
Returns:
|
|
A JSON string representation of the input dictionary with a newline character.
|
|
"""
|
|
return json.dumps(json_dict, cls=encoder) + "\n"
|
|
|
|
|
|
def mask_string(sensitive_str: str) -> str:
|
|
return "****...**" + sensitive_str[-4:]
|
|
|
|
|
|
def mask_credential_dict(credential_dict: dict[str, Any]) -> dict[str, str]:
|
|
masked_creds = {}
|
|
for key, val in credential_dict.items():
|
|
if not isinstance(val, str):
|
|
raise ValueError(
|
|
f"Unable to mask credentials of type other than string, cannot process request."
|
|
f"Recieved type: {type(val)}"
|
|
)
|
|
|
|
masked_creds[key] = mask_string(val)
|
|
return masked_creds
|
|
|
|
|
|
def send_user_email_invite(user_email: str, current_user: User) -> None:
|
|
msg = MIMEMultipart()
|
|
msg["Subject"] = "Invitation to Join Danswer Workspace"
|
|
msg["From"] = current_user.email
|
|
msg["To"] = user_email
|
|
|
|
email_body = dedent(
|
|
f"""\
|
|
Hello,
|
|
|
|
You have been invited to join a workspace on Danswer.
|
|
|
|
To join the workspace, please visit the following link:
|
|
|
|
{WEB_DOMAIN}/auth/login
|
|
|
|
Best regards,
|
|
The Danswer Team
|
|
"""
|
|
)
|
|
|
|
msg.attach(MIMEText(email_body, "plain"))
|
|
with smtplib.SMTP(SMTP_SERVER, SMTP_PORT) as smtp_server:
|
|
smtp_server.starttls()
|
|
smtp_server.login(SMTP_USER, SMTP_PASS)
|
|
smtp_server.send_message(msg)
|