Skip to content

Commit df1db48

Browse files
committed
Remove even more duplication
1 parent d05622f commit df1db48

File tree

1 file changed

+2
-10
lines changed

1 file changed

+2
-10
lines changed

src/_notion_scripts/upload.py

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -57,18 +57,10 @@ def _block_from_details(
5757
"""
5858
block = Block.wrap_obj_ref(UnoObjAPIBlock.model_validate(obj=details))
5959

60-
if isinstance(block, UnoImage):
60+
if isinstance(block, (UnoImage, UnoVideo, UnoAudio)):
6161
uploaded_file = _upload_local_file(url=block.url, session=session)
6262
if uploaded_file is not None:
63-
return UnoImage(file=uploaded_file, caption=block.caption)
64-
elif isinstance(block, UnoVideo):
65-
uploaded_file = _upload_local_file(url=block.url, session=session)
66-
if uploaded_file is not None:
67-
return UnoVideo(file=uploaded_file, caption=block.caption)
68-
elif isinstance(block, UnoAudio):
69-
uploaded_file = _upload_local_file(url=block.url, session=session)
70-
if uploaded_file is not None:
71-
return UnoAudio(file=uploaded_file, caption=block.caption)
63+
return block.__class__(file=uploaded_file, caption=block.caption)
7264

7365
return block
7466

0 commit comments

Comments
 (0)