Skip to content

Commit

Permalink
Merge branch 'master' into add-asyncpg-to-director-v2
Browse files Browse the repository at this point in the history
  • Loading branch information
matusdrobuliak66 authored Nov 18, 2024
2 parents 312db8c + a381aa6 commit b62248c
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
import sqlalchemy as sa
from aiopg.sa.connection import SAConnection
from aiopg.sa.result import RowProxy
from models_library.projects import ProjectAtDB
from models_library.projects import ProjectAtDB, ProjectID
from models_library.projects_nodes import Node
from models_library.projects_nodes_io import NodeIDStr
from models_library.users import UserID
Expand Down Expand Up @@ -164,14 +164,18 @@ async def _get_tags_by_project(conn: SAConnection, project_id: str) -> list:

@staticmethod
async def _upsert_tags_in_project(
conn: SAConnection, project_index_id: int, project_tags: list[int]
conn: SAConnection,
project_index_id: int,
project_uuid: ProjectID,
project_tags: list[int],
) -> None:
for tag_id in project_tags:
await conn.execute(
pg_insert(projects_tags)
.values(
project_id=project_index_id,
tag_id=tag_id,
project_uuid_for_rut=project_uuid,
)
.on_conflict_do_nothing()
)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -202,6 +202,7 @@ def _reraise_if_not_unique_uuid_error(err: UniqueViolation):
await self._upsert_tags_in_project(
conn=conn,
project_index_id=project_index,
project_uuid=project_uuid,
project_tags=project_tag_ids,
)
selected_values["tags"] = project_tag_ids
Expand Down

0 comments on commit b62248c

Please sign in to comment.