Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add duplicated workspace create logic #255

Merged
merged 1 commit into from
Apr 12, 2024
Merged

Add duplicated workspace create logic #255

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions src/spaceone/identity/conf/global_conf.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -79,6 +79,7 @@
"WorkspaceUser.create": ["password"],
"Token.issue": ["credentials"],
"Token.grant": ["token"],
"Job.sync_service_accounts": ["secret_data"],
}
}
}
Expand Down
90 changes: 54 additions & 36 deletions src/spaceone/identity/service/job_service.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import logging
import random
from datetime import datetime, timedelta
from typing import Union, List
from typing import Union, List, Tuple

from spaceone.core.service import *
from spaceone.core.service.utils import *
Expand Down Expand Up @@ -65,7 +65,7 @@ def create_jobs_by_trusted_account(self, params):
current_hour = params.get("current_hour", datetime.utcnow().hour)

for trusted_account_vo in self._get_all_schedule_enabled_trusted_accounts(
current_hour
current_hour
):
try:
self.created_service_account_job(trusted_account_vo, {})
Expand Down Expand Up @@ -327,7 +327,7 @@ def sync_service_accounts(self, params: dict) -> None:
)

def created_service_account_job(
self, trusted_account_vo: TrustedAccount, job_options: dict
self, trusted_account_vo: TrustedAccount, job_options: dict
) -> Union[Job, dict]:
resource_group = trusted_account_vo.resource_group
provider = trusted_account_vo.provider
Expand Down Expand Up @@ -431,10 +431,10 @@ def _get_trusted_secret_data(self, trusted_secret_id: str, domain_id: str) -> di
return secret_data

def _check_duplicate_job(
self,
domain_id: str,
trusted_account_id: str,
this_job_vo: Job,
self,
domain_id: str,
trusted_account_id: str,
this_job_vo: Job,
) -> bool:
query = {
"filter": [
Expand All @@ -458,7 +458,7 @@ def _check_duplicate_job(
return False

def _is_job_failed(
self, job_id: str, domain_id: str, workspace_id: str = None
self, job_id: str, domain_id: str, workspace_id: str = None
) -> bool:
job_vo: Job = self.job_mgr.get_job(domain_id, job_id, workspace_id)

Expand All @@ -468,10 +468,10 @@ def _is_job_failed(
return False

def _close_job(
self,
job_id: str,
domain_id: str,
workspace_id: str = None,
self,
job_id: str,
domain_id: str,
workspace_id: str = None,
):
job_vo: Job = self.job_mgr.get_job(domain_id, job_id, workspace_id)
if job_vo.status == "IN_PROGRESS":
Expand All @@ -480,16 +480,18 @@ def _close_job(
self.job_mgr.update_job_by_vo({"finished_at": datetime.utcnow()}, job_vo)

def _create_workspace(
self, domain_id: str, trusted_account_id: str, location_info: dict
self, domain_id: str, trusted_account_id: str, location_info: dict
) -> Workspace:
name = location_info.get("name")
reference_id = location_info.get("resource_id")

name, tags = self._check_duplicated_workspace_name(name, domain_id)
workspace_vos = self.workspace_mgr.filter_workspaces(
domain_id=domain_id, reference_id=reference_id, is_managed=True
)

_LOGGER.debug(
f"[_create_workspace] {name} 'domain_id': {domain_id}, 'reference_id': {reference_id}, 'is_managed' :True count: {len(workspace_vos)}"
f"[_create_workspace] {name} 'domain_id': {domain_id}, 'reference_id': {reference_id} 'tags': {tags}, 'is_managed' :True count: {len(workspace_vos)}"
)

params = {"trusted_account_id": trusted_account_id}
Expand All @@ -506,7 +508,7 @@ def _create_workspace(
{
"name": name,
"is_managed": True,
"tags": self._set_workspace_theme(),
"tags": self._set_workspace_theme(tags),
"reference_id": reference_id,
"domain_id": domain_id,
"last_synced_at": datetime.utcnow(),
Expand All @@ -516,12 +518,12 @@ def _create_workspace(
return workspace_vo

def _create_project_group(
self,
domain_id: str,
workspace_id: str,
trusted_account_id: str,
location_info: dict,
parent_group_id: str = None,
self,
domain_id: str,
workspace_id: str,
trusted_account_id: str,
location_info: dict,
parent_group_id: str = None,
) -> ProjectGroup:
name = location_info["name"]
reference_id = location_info["resource_id"]
Expand Down Expand Up @@ -569,14 +571,14 @@ def _create_project_group(
return project_group_vo

def _create_project(
self,
result: dict,
domain_id: str,
workspace_id: str,
trusted_account_id: str,
project_group_id: str = None,
sync_options: dict = None,
project_type: str = "PRIVATE",
self,
result: dict,
domain_id: str,
workspace_id: str,
trusted_account_id: str,
project_group_id: str = None,
sync_options: dict = None,
project_type: str = "PRIVATE",
) -> Project:
name = result["name"]
reference_id = result["resource_id"]
Expand Down Expand Up @@ -612,13 +614,13 @@ def _create_project(
return project_vo

def _create_service_account(
self,
result: dict,
project_vo: Project,
trusted_account_id: str,
trusted_secret_id: str,
provider: str,
sync_options: dict = None,
self,
result: dict,
project_vo: Project,
trusted_account_id: str,
trusted_secret_id: str,
provider: str,
sync_options: dict = None,
) -> Union[ServiceAccount, None]:
domain_id = project_vo.domain_id
workspace_id = project_vo.workspace_id
Expand Down Expand Up @@ -703,6 +705,22 @@ def _create_service_account(
)
return service_account_vo

# todo : temporary function need policy about duplicated workspace name
def _check_duplicated_workspace_name(
self, name: str, domain_id: str, tags: dict = None
) -> Tuple[str, dict]:
workspace_vos = self.workspace_mgr.filter_workspaces(
domain_id=domain_id, name=name
)

if tags is None:
tags = {}

if workspace_vos:
tags.update({"origin_name": name})
name = f"{name} ({len(workspace_vos) + 1})"
return name, tags

@staticmethod
def _get_location(result: dict, resource_group: str, sync_options: dict) -> list:
location = []
Expand Down
Loading