mirror of
https://github.com/clearml/clearml-server
synced 2025-02-11 07:14:23 +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
38 lines
990 B
Python
38 lines
990 B
Python
from logging import Logger
|
|
from uuid import uuid4
|
|
|
|
from bll.queue import QueueBLL
|
|
from config import config
|
|
from database.model.company import Company
|
|
from database.model.queue import Queue
|
|
from database.model.settings import Settings, SettingKeys
|
|
|
|
log = config.logger(__file__)
|
|
|
|
|
|
def _ensure_company(company_id, company_name, log: Logger):
|
|
company = Company.objects(id=company_id).only("id").first()
|
|
if company:
|
|
return company_id
|
|
|
|
log.info(f"Creating company: {company_name}")
|
|
company = Company(id=company_id, name=company_name)
|
|
company.save()
|
|
return company_id
|
|
|
|
|
|
def _ensure_default_queue(company):
|
|
"""
|
|
If no queue is present for the company then
|
|
create a new one and mark it as a default
|
|
"""
|
|
queue = Queue.objects(company=company).only("id").first()
|
|
if queue:
|
|
return
|
|
|
|
QueueBLL.create(company, name="default", system_tags=["default"])
|
|
|
|
|
|
def _ensure_uuid():
|
|
Settings.add_value(SettingKeys.server__uuid, str(uuid4()))
|