mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-05-29 05:00:04 +02:00
24184024bb
* bump fastapi and starlette * bumping llama index and nltk and associated deps * bump to fix python-multipart * bump aiohttp * update package lock for examples/widget * bump black * sentencesplitter has changed namespaces * fix reorder import check, fix missing passlib * update package-lock.json * black formatter updated * reformatted again * change to black compatible reorder * change to black compatible reorder-python-imports fork * fix pytest dependency * black format again * we don't need cdk.txt. update packages to be consistent across all packages --------- Co-authored-by: Richard Kuo (Onyx) <rkuo@onyx.app> Co-authored-by: Richard Kuo <rkuo@rkuo.com>
60 lines
2.0 KiB
Python
60 lines
2.0 KiB
Python
from datetime import datetime
|
|
from datetime import timedelta
|
|
|
|
import jwt
|
|
import requests
|
|
|
|
from onyx.server.manage.models import AllUsersResponse
|
|
from onyx.server.models import FullUserSnapshot
|
|
from onyx.server.models import InvitedUserSnapshot
|
|
from tests.integration.common_utils.constants import API_SERVER_URL
|
|
from tests.integration.common_utils.constants import GENERAL_HEADERS
|
|
from tests.integration.common_utils.test_models import DATestUser
|
|
|
|
|
|
def generate_auth_token() -> str:
|
|
payload = {
|
|
"iss": "control_plane",
|
|
"exp": datetime.utcnow() + timedelta(minutes=5),
|
|
"iat": datetime.utcnow(),
|
|
"scope": "tenant:create",
|
|
}
|
|
token = jwt.encode(payload, "", algorithm="HS256")
|
|
return token
|
|
|
|
|
|
class TenantManager:
|
|
@staticmethod
|
|
def get_all_users(
|
|
user_performing_action: DATestUser | None = None,
|
|
) -> AllUsersResponse:
|
|
response = requests.get(
|
|
url=f"{API_SERVER_URL}/manage/users",
|
|
headers=(
|
|
user_performing_action.headers
|
|
if user_performing_action
|
|
else GENERAL_HEADERS
|
|
),
|
|
)
|
|
response.raise_for_status()
|
|
|
|
data = response.json()
|
|
return AllUsersResponse(
|
|
accepted=[FullUserSnapshot(**user) for user in data["accepted"]],
|
|
invited=[InvitedUserSnapshot(**user) for user in data["invited"]],
|
|
slack_users=[FullUserSnapshot(**user) for user in data["slack_users"]],
|
|
accepted_pages=data["accepted_pages"],
|
|
invited_pages=data["invited_pages"],
|
|
slack_users_pages=data["slack_users_pages"],
|
|
)
|
|
|
|
@staticmethod
|
|
def verify_user_in_tenant(
|
|
user: DATestUser, user_performing_action: DATestUser | None = None
|
|
) -> None:
|
|
all_users = TenantManager.get_all_users(user_performing_action)
|
|
for accepted_user in all_users.accepted:
|
|
if accepted_user.email == user.email and accepted_user.id == user.id:
|
|
return
|
|
raise ValueError(f"User {user.email} not found in tenant")
|