mirror of
https://github.com/clearml/clearml-server
synced 2025-02-12 15:50:46 +00:00
baba8b5b73
Add initial support for project ordering Add support for sortable task duration (used by the UI in the experiment's table) Add support for project name in worker's current task info Add support for results and artifacts in pre-populates examples Add demo server features
73 lines
1.9 KiB
Python
73 lines
1.9 KiB
Python
from datetime import datetime
|
|
from logging import Logger
|
|
|
|
import attr
|
|
|
|
from database.model.auth import Role
|
|
from database.model.auth import User as AuthUser, Credentials
|
|
from database.model.user import User
|
|
from service_repo.auth.fixed_user import FixedUser
|
|
|
|
|
|
def _ensure_auth_user(user_data: dict, company_id: str, log: Logger, revoke: bool = False):
|
|
key, secret = user_data.get("key"), user_data.get("secret")
|
|
if not (key and secret):
|
|
credentials = None
|
|
else:
|
|
creds = Credentials(key=key, secret=secret)
|
|
|
|
user = AuthUser.objects(credentials__match=creds).first()
|
|
if user:
|
|
if revoke:
|
|
user.credentials = []
|
|
user.save()
|
|
return user.id
|
|
|
|
credentials = [] if revoke else [creds]
|
|
|
|
user_id = user_data.get("id", f"__{user_data['name']}__")
|
|
|
|
log.info(f"Creating user: {user_data['name']}")
|
|
|
|
user = AuthUser(
|
|
id=user_id,
|
|
name=user_data["name"],
|
|
company=company_id,
|
|
role=user_data["role"],
|
|
email=user_data["email"],
|
|
created=datetime.utcnow(),
|
|
credentials=credentials,
|
|
)
|
|
|
|
user.save()
|
|
|
|
return user.id
|
|
|
|
|
|
def _ensure_backend_user(user_id: str, company_id: str, user_name: str):
|
|
given_name, _, family_name = user_name.partition(" ")
|
|
|
|
User(
|
|
id=user_id,
|
|
company=company_id,
|
|
name=user_name,
|
|
given_name=given_name,
|
|
family_name=family_name,
|
|
).save()
|
|
|
|
return user_id
|
|
|
|
|
|
def ensure_fixed_user(user: FixedUser, log: Logger):
|
|
if User.objects(company=user.company, id=user.user_id).first():
|
|
return
|
|
|
|
data = attr.asdict(user)
|
|
data["id"] = user.user_id
|
|
data["email"] = f"{user.user_id}@example.com"
|
|
data["role"] = Role.guest if user.is_guest else Role.user
|
|
|
|
_ensure_auth_user(user_data=data, company_id=user.company, log=log)
|
|
|
|
return _ensure_backend_user(user.user_id, user.company, user.name)
|