mirror of
https://github.com/StefBuwalda/WebTech.git
synced 2025-10-29 19:00:00 +00:00
24 lines
832 B
Python
24 lines
832 B
Python
import os
|
|
from werkzeug.utils import secure_filename
|
|
from application import app
|
|
from flask_login import current_user # type: ignore
|
|
|
|
|
|
# save image to static folder
|
|
def saveImage(image: ...):
|
|
filename = secure_filename(image.filename)
|
|
# Path should be /application/static/[user_id]/[filename]
|
|
save_path = os.path.join(
|
|
app.config["UPLOAD_FOLDER"], # type: ignore
|
|
str(current_user.id),
|
|
filename,
|
|
)
|
|
# Create path is it doesn't exist
|
|
os.makedirs(os.path.dirname(save_path), exist_ok=True)
|
|
# Save the image
|
|
image.save(save_path) # type: ignore
|
|
# Return the filename that is stored in database.
|
|
# Only done to keep a single default image, this should be done differently
|
|
filename2 = str(current_user.id) + "/" + filename # [user_id]/[filename]
|
|
return filename2
|