Skip to content

Commit a52cd87

Browse files
[autofix.ci] apply automated fixes
1 parent 6fb95ff commit a52cd87

5 files changed

Lines changed: 17 additions & 49 deletions

File tree

‎api/controllers/console/app/workflow_alias.py‎

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,6 @@
1919

2020

2121
class WorkflowAliasApi(Resource):
22-
2322
@setup_required
2423
@login_required
2524
@account_initialization_required
@@ -123,6 +122,3 @@ def delete(self, app_model: App):
123122
return {"message": "Alias deleted successfully"}
124123
except ValueError as e:
125124
raise BadRequest(str(e))
126-
127-
128-

‎api/controllers/service_api/app/completion.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -174,7 +174,7 @@ def _resolve_workflow_alias(self, app_model: App, args: dict) -> None:
174174
session=db.session,
175175
tenant_id=app_model.tenant_id,
176176
app_id=app_model.id,
177-
alias_name=args["workflow_alias"]
177+
alias_name=args["workflow_alias"],
178178
)
179179

180180
if not workflow:

‎api/controllers/service_api/app/workflow.py‎

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -182,10 +182,7 @@ def _resolve_workflow_id(self, app_model: App, identifier: str) -> str:
182182
"""
183183
workflow_alias_service = WorkflowAliasService()
184184
workflow = workflow_alias_service.get_workflow_by_alias(
185-
session=db.session,
186-
tenant_id=app_model.tenant_id,
187-
app_id=app_model.id,
188-
alias_name=identifier
185+
session=db.session, tenant_id=app_model.tenant_id, app_id=app_model.id, alias_name=identifier
189186
)
190187

191188
if workflow:
@@ -198,8 +195,7 @@ def _resolve_workflow_id(self, app_model: App, identifier: str) -> str:
198195
except (ValueError, TypeError):
199196
# Neither alias nor valid UUID format
200197
raise WorkflowIdFormatError(
201-
f"Invalid identifier '{identifier}'. "
202-
f"Must be a valid workflow alias or UUID format."
198+
f"Invalid identifier '{identifier}'. Must be a valid workflow alias or UUID format."
203199
)
204200

205201

‎api/models/workflow_alias.py‎

Lines changed: 8 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,9 @@
1212

1313
class AliasType(Enum):
1414
"""Alias type enumeration"""
15-
SYSTEM = "system" # System aliases like 'production', 'staging'
16-
CUSTOM = "custom" # User-defined custom aliases
15+
16+
SYSTEM = "system" # System aliases like 'production', 'staging'
17+
CUSTOM = "custom" # User-defined custom aliases
1718

1819

1920
class WorkflowAlias(Base):
@@ -38,7 +39,7 @@ class WorkflowAlias(Base):
3839

3940
__tablename__ = "workflow_aliases"
4041
__allow_unmapped__ = True # Allow non-mapped attributes
41-
__slots__ = ('_is_transferred', '_old_workflow_id')
42+
__slots__ = ("_is_transferred", "_old_workflow_id")
4243
__table_args__ = (
4344
sa.PrimaryKeyConstraint("id", name="workflow_alias_pkey"),
4445
# Ensure alias name is unique within an app
@@ -54,23 +55,12 @@ class WorkflowAlias(Base):
5455
app_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
5556
workflow_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
5657
alias_name: Mapped[str] = mapped_column(String(255), nullable=False)
57-
alias_type: Mapped[str] = mapped_column(
58-
String(50),
59-
nullable=False,
60-
server_default=AliasType.CUSTOM.value
61-
)
58+
alias_type: Mapped[str] = mapped_column(String(50), nullable=False, server_default=AliasType.CUSTOM.value)
6259

6360
created_by: Mapped[str] = mapped_column(StringUUID, nullable=False)
64-
created_at: Mapped[datetime] = mapped_column(
65-
sa.DateTime,
66-
nullable=False,
67-
server_default=func.current_timestamp()
68-
)
61+
created_at: Mapped[datetime] = mapped_column(sa.DateTime, nullable=False, server_default=func.current_timestamp())
6962
updated_at: Mapped[datetime] = mapped_column(
70-
sa.DateTime,
71-
nullable=False,
72-
server_default=func.current_timestamp(),
73-
onupdate=func.current_timestamp()
63+
sa.DateTime, nullable=False, server_default=func.current_timestamp(), onupdate=func.current_timestamp()
7464
)
7565

7666
def __init__(self, *args, **kwargs):
@@ -82,6 +72,7 @@ def __init__(self, *args, **kwargs):
8272
def created_by_account(self):
8373
"""Get the account that created this alias"""
8474
from .engine import db
75+
8576
return db.session.get(Account, self.created_by)
8677

8778
@property

‎api/services/workflow_alias_service.py‎

Lines changed: 6 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@
1515

1616

1717
class WorkflowAliasService:
18-
1918
def create_alias(
2019
self,
2120
session: Union[Session, "scoped_session"],
@@ -34,12 +33,7 @@ def create_alias(
3433
raise ValueError("Cannot create or transfer aliases for draft workflows")
3534

3635
existing_alias = session.execute(
37-
select(WorkflowAlias).where(
38-
and_(
39-
WorkflowAlias.app_id == app_id,
40-
WorkflowAlias.alias_name == alias_name
41-
)
42-
)
36+
select(WorkflowAlias).where(and_(WorkflowAlias.app_id == app_id, WorkflowAlias.alias_name == alias_name))
4337
).scalar_one_or_none()
4438

4539
if existing_alias:
@@ -51,8 +45,7 @@ def create_alias(
5145
existing_alias._old_workflow_id = old_workflow_id
5246

5347
logger.info(
54-
"Transferred alias '%s' from workflow %s to workflow %s",
55-
alias_name, old_workflow_id, workflow_id
48+
"Transferred alias '%s' from workflow %s to workflow %s", alias_name, old_workflow_id, workflow_id
5649
)
5750
return existing_alias
5851

@@ -69,10 +62,7 @@ def create_alias(
6962
session.add(alias)
7063
session.flush()
7164

72-
logger.info(
73-
"Created workflow alias: %s for workflow %s",
74-
alias_name, workflow_id
75-
)
65+
logger.info("Created workflow alias: %s for workflow %s", alias_name, workflow_id)
7666
return alias
7767

7868
def get_aliases_by_app(
@@ -82,17 +72,12 @@ def get_aliases_by_app(
8272
app_id: str,
8373
workflow_ids: Optional[list[str]] = None,
8474
) -> list[WorkflowAlias]:
85-
conditions = [
86-
WorkflowAlias.tenant_id == tenant_id,
87-
WorkflowAlias.app_id == app_id
88-
]
75+
conditions = [WorkflowAlias.tenant_id == tenant_id, WorkflowAlias.app_id == app_id]
8976

9077
if workflow_ids:
9178
conditions.append(WorkflowAlias.workflow_id.in_(workflow_ids))
9279

93-
stmt = select(WorkflowAlias).where(
94-
and_(*conditions)
95-
).order_by(WorkflowAlias.created_at.desc())
80+
stmt = select(WorkflowAlias).where(and_(*conditions)).order_by(WorkflowAlias.created_at.desc())
9681

9782
return list(session.execute(stmt).scalars().all())
9883

@@ -108,7 +93,7 @@ def get_workflow_by_alias(
10893
and_(
10994
WorkflowAlias.tenant_id == tenant_id,
11095
WorkflowAlias.app_id == app_id,
111-
WorkflowAlias.alias_name == alias_name
96+
WorkflowAlias.alias_name == alias_name,
11297
)
11398
)
11499
).scalar_one_or_none()

0 commit comments

Comments
 (0)