refactor: Update connector ID check for YAML file uploads
Taylor committed
Sep 8, 2024 at 17:28 UTC
a572720e6984580d73ce9e87b0107a88ffdcfabe
2 files changed
+3
-2
backend/app/connectors/routes.py
+1
-1
@@ -221,7 +221,7 @@ async def upload_yaml_file(
221
if not file.filename.endswith(".yaml"):
222
raise HTTPException(status_code=400, detail="Only .yaml files are allowed.")
223
try:
224
- save_file_result = await ConnectorServices.save_file(file, session=session)
224
+ save_file_result = await ConnectorServices.save_file(file, connector_id, session=session)
225
if save_file_result:
226
await ConnectorServices.verify_connector_by_id(
227
connector_id,
backend/app/connectors/services.py
+2
-1
@@ -379,6 +379,7 @@ class ConnectorServices:
379
async def save_file(
380
cls,
381
file: UploadFile,
382
+ connector_id: int,
383
session: AsyncSession,
384
) -> Union[ConnectorResponse, bool]:
385
"""
@@ -403,7 +404,7 @@ class ConnectorServices:
404
) # Assuming file doesn't need to be read in chunks
405
406
# Update connector using async session and ORM
406
- query = select(Connectors).where(Connectors.id == 6)
407
+ query = select(Connectors).where(Connectors.id == connector_id)
408
connector_record = (await session.execute(query)).scalars().first()
409
410
if connector_record: