julius/speckles
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add filename option

Browse Source
This commit is contained in:
julius 2024-10-21 13:32:10 +02:00
parent dc3503a407
commit dfaef09ccb
Signed by: julius
GPG Key ID: C80A63E6A5FD7092
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
main.py
View File

@ -92,7 +92,7 @@ def make_wallpaper(
size: float = 3, size: float = 3,
fileformat: str = "svg", fileformat: str = "svg",
orientation: str = "landscape", orientation: str = "landscape",
local: bool = False, filename: str = "",
markers: str = ".", markers: str = ".",
perlin: bool = True, perlin: bool = True,
): ):
@ -163,14 +163,13 @@ def make_wallpaper(
pad_inches=0, pad_inches=0,
) )
buf.seek(0) buf.seek(0)
if local: if filename:
filename = f"speckles.{fileformat}"
with open(filename, "wb") as f: with open(filename, "wb") as f:
f.write(buf.getbuffer()) f.write(buf.getbuffer())
buf.close()
return filename return filename
else: else:
return StreamingResponse(content=buf, media_type=MEDIA_TYPES[fileformat]) return StreamingResponse(content=buf, media_type=MEDIA_TYPES[fileformat])
buf.close()
if __name__ == "__main__": if __name__ == "__main__":