def upload_file():
if 'user_image' not in request.files:
flash(f"No user image key in request.files")
return redirect(url_for('home'))
file = request.files['user_image']
if file.filename == "":
return "Please select a file"
if file and allowed_file(file.filename):
file.filename = secure_filename(file.filename)
output = upload_file_to_s3(file, S3_BUCKET)
upload_image = Images(user_image=file.filename,
user_id=current_user.id)
if upload_image.save():
flash(f'Photo successfully uploaded.')
return render_template('upload_pictures.html')
else:
flash(f'Unable to upload picture. Please try again.')
return render_template('upload_pictures.html')
else:
return redirect(url_for('home'))
def create():
user = User.get_or_none(User.id == current_user.id)
#This is to make sure a file is submitted, or to prevent prople from submitting empty form.
if "images" not in request.files:
flash("No file provided", "danger")
return redirect(url_for('images.new'))
#file is an instance of a File Class.
file = request.files.get('images')
file.filename = secure_filename(file.filename)
#upload_file_s3 from util.helpers.py takes in 2 arguement
#1st arguement is the file object itself, the second arguement
#is the foldername. Which is the username in our case.
image_path = upload_file_s3(file, user.username)
image = Images(user=user, image_url=image_path)
if image.save():
flash("Image uploaded", "primary")
return redirect(url_for('users.show', username=current_user.username))
else:
flash("Try again bruh, something went wrong", "danger")
return render_template("images/new.html")
def record_image(self,
f,
uuid,
link,
album_id,
call_type="create",
image_id=None):
im = Imaging(f)
origin = im.thumbnail()
try:
if call_type == "create":
Images().create(uuid=uuid,
link=link,
album=album_id,
thumbnail=origin)
else:
Images.update(link=link, album_id=album_id, thumbnail=origin,
uuid=uuid, updateTime=time()). \
where(Images.id == int(image_id)).execute()
if not os.path.exists(config.UPLOAD_DIR):
os.makedirs(config.UPLOAD_DIR)
file_name = str(uuid) + ".jpeg"
tmp_file = os.path.join(config.UPLOAD_DIR, file_name)
with open(tmp_file, 'wb') as fs:
fs.write(f)
except Exception as e:
print(traceback.format_exc())
log.error(traceback.format_exc())
return uuid
def create():
file = request.files.get('upload_images')
result = upload(file)
image = Images(img=result, user=current_user.id)
image.save()
return redirect(url_for('users.show', username=current_user.name))
def profile_upload():
if request.method == 'POST':
f = request.files['user_file']
f.filename = secure_filename(f.filename)
image = Images(path=f.filename, user=current_user.id)
image.save()
print(upload_file_to_s3(f))
return redirect(url_for('users.show', username=current_user.username))
def create():
file = request.files["user_file"] #if no file in request
if not file:
flash("Please choose a photo to upload.", "danger")
return render_template('images/new.html')
file.filename = secure_filename(file.filename)
output = upload_file_to_s3(file) #import from helpers.py
if not output:
flash("Unable to upload photo, try again.", "danger")
return render_template('images/new.html')
else:
uploadphoto = Images(username=User.id, img_file=file)
images = Images(username_id=current_user.id, img_file=output)
print(output)
flash("Picture uploaded succesfully", "success")
if images.save():
flash("Posted!", "success")
return redirect(url_for('images.new'))
else:
flash("Upload failed", "danger")
return render_template('images/new.html')
def saveImageToDB(imagesInfo, isBig = True):
''' Download images from vk.com
return images which not found in vk servers
'''
size = -1 if isBig else 0
for imagesInfo in imagesInfo:
try:
if Images().checkImgId(imagesInfo['id']):
Images().update(imagesInfo)
else:
response = requests.get(imagesInfo['sizes'][size]['src'])
img = Image.open(BytesIO(response.content))
img = img.convert(mode='RGB')
img_hash = getImageHash(img)
Images().insert(imagesInfo, img_hash)
except urllib.error.HTTPError as err:
print(err)
print('https://vk.com/photo-2481783_' + str(imagesInfo['id']))
except Exception as err:
print(err)
print('https://vk.com/photo-2481783_' + str(imagesInfo['id']))
def upload_img():
from app import app
file = request.files.get("image")
print(f"upload images:{file}")
result = upload(file)
print(get_jwt_identity())
image = Images(user=get_jwt_identity(), img=result)
image.save()
return jsonify({
"images_url": "app.config.get('AWS_S3_DOMAIN')" + file.filename,
"success": True
})
def specific_usr_upload_img(id):
if current_user.id == int(id):
#get uploaded img details
upload_file = request.files['user_upload_img']
# upload_file = request.files.get('user_upload_img')
#upload to s3 bucket
upload_file_to_s3(upload_file, os.environ.get('S3_BUCKET_NAME'))
#save in DB
i = Images(user_id=current_user.id, img_name=upload_file.filename)
# breakpoint()
if i.save():
flash('Image Uploaded Successfully!')
return redirect(url_for('users.user_profile', id=id))
else:
flash('Upload Image Failed!')
return render_template('/users/sign_in.html', errors=i.errors)
else:
flash('Upload Image Failed aaa!')
return redirect('/users/sign_in.html')
