Skip to content

Commit d6e9e9c

Browse files
Francesco MucioFrancesco Mucio
authored andcommitted
style: apply ruff format to create_drive_folder changes
1 parent cc2b23d commit d6e9e9c

2 files changed

Lines changed: 26 additions & 14 deletions

File tree

gdrive/drive_tools.py

Lines changed: 21 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -467,13 +467,15 @@ async def _create_drive_folder_impl(
467467
"mimeType": "application/vnd.google-apps.folder",
468468
}
469469
created_file = await asyncio.to_thread(
470-
lambda: service.files()
471-
.create(
472-
body=file_metadata,
473-
fields="id, name, webViewLink",
474-
supportsAllDrives=True,
470+
lambda: (
471+
service.files()
472+
.create(
473+
body=file_metadata,
474+
fields="id, name, webViewLink",
475+
supportsAllDrives=True,
476+
)
477+
.execute()
475478
)
476-
.execute()
477479
)
478480
link = created_file.get("webViewLink", "")
479481
return (
@@ -542,7 +544,11 @@ async def create_drive_file(
542544
f"[create_drive_file] Invoked. Email: '{user_google_email}', File Name: {file_name}, Folder ID: {folder_id}, fileUrl: {fileUrl}"
543545
)
544546

545-
if not content and not fileUrl and mime_type != "application/vnd.google-apps.folder":
547+
if (
548+
not content
549+
and not fileUrl
550+
and mime_type != "application/vnd.google-apps.folder"
551+
):
546552
raise Exception("You must provide either 'content' or 'fileUrl'.")
547553

548554
file_data = None
@@ -557,13 +563,15 @@ async def create_drive_file(
557563
# Create folder (no content or media_body). Prefer create_drive_folder for new code.
558564
if mime_type == "application/vnd.google-apps.folder":
559565
created_file = await asyncio.to_thread(
560-
lambda: service.files()
561-
.create(
562-
body=file_metadata,
563-
fields="id, name, webViewLink",
564-
supportsAllDrives=True,
566+
lambda: (
567+
service.files()
568+
.create(
569+
body=file_metadata,
570+
fields="id, name, webViewLink",
571+
supportsAllDrives=True,
572+
)
573+
.execute()
565574
)
566-
.execute()
567575
)
568576
link = created_file.get("webViewLink", "")
569577
return (

tests/gdrive/test_drive_tools.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,11 @@ async def test_create_drive_folder():
2727
mock_request.execute.return_value = mock_response
2828
mock_service.files.return_value.create.return_value = mock_request
2929

30-
with patch("gdrive.drive_tools.resolve_folder_id", new_callable=AsyncMock, return_value="root"):
30+
with patch(
31+
"gdrive.drive_tools.resolve_folder_id",
32+
new_callable=AsyncMock,
33+
return_value="root",
34+
):
3135
result = await _create_drive_folder_impl(
3236
service=mock_service,
3337
user_google_email="user@example.com",

0 commit comments

Comments
 (0)