mirrors/bookwyrm
1
0
Fork 1
mirror of https://github.com/bookwyrm-social/bookwyrm.git synced 2024-05-17 16:03:04 +00:00
Code Issues Projects Releases Wiki Activity

Fix mypy error

Browse source
This commit is contained in:
Bart Schuurmans 2024-03-28 13:21:30 +01:00
parent dabf7c6e10
commit bb5d8152f1
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
bookwyrm/utils/tar.py
View file

@ -18,7 +18,7 @@ class BookwyrmTarFile(tarfile.TarFile):
self.addfile(info, fileobj=buffer)
def add_image(
self, image: Any, filename: Optional[str] = None, directory: Any = ""
self, image: Any, filename: Optional[str] = None, directory: str = ""
) -> None:
"""
Add an image to the tar archive
@ -26,12 +26,12 @@ class BookwyrmTarFile(tarfile.TarFile):
:param str directory: the directory in the archive to put the image
"""
if filename is None:
filename = image.name
dst_filename = image.name
else:
filename += os.path.splitext(image.name)[1]
path = os.path.join(directory, filename)
dst_filename = filename + os.path.splitext(image.name)[1]
dst_path = os.path.join(directory, dst_filename)
info = tarfile.TarInfo(name=path)
info = tarfile.TarInfo(name=dst_path)
info.size = image.size
self.addfile(info, fileobj=image)