def audio(): form = WatermarkForm() global AUDIO_PATH if form.validate_on_submit(): f = request.files['file'] save_path = os.path.join(current_app.instance_path, 'upload', f.filename) f.save(save_path) AUDIO_PATH = save_path if len(AUDIO_PATH) == 0: return redirect(url_for('audio')) else: watermark_string = form.watermark.data temp_file_path = os.path.join(current_app.instance_path, 'temp', 'temp.wav') lsb_watermark(AUDIO_PATH, watermark_string, temp_file_path) return redirect(url_for('audio_get', filename='temp.wav')) return render_template('audio.html', form=form)
def video(): form = WatermarkForm() global VIDEO_PATH if form.validate_on_submit(): f = request.files['file'] save_path = os.path.join(current_app.instance_path, 'upload', f.filename) f.save(save_path) VIDEO_PATH = save_path if len(VIDEO_PATH) == 0: return redirect(url_for('video')) else: watermark_string = form.watermark.data temp_file_path = os.path.join(current_app.instance_path, 'temp', 'temp.mp4') embed_video(VIDEO_PATH, watermark_string, temp_file_path) return redirect(url_for('video_get', filename='temp.mp4')) return render_template('video.html', form=form)
def image(): global IMAGE_PATH form = WatermarkForm() if form.validate_on_submit(): f = request.files['file'] save_path = os.path.join(current_app.instance_path, 'upload', f.filename) f.save(save_path) IMAGE_PATH = save_path if len(IMAGE_PATH) == 0: return redirect(url_for('image')) else: watermark_string = form.watermark.data temp_file_path = os.path.join(current_app.instance_path, 'temp', 'temp.jpg') print(temp_file_path) embed_watermark(IMAGE_PATH, watermark_string, temp_file_path) print('ok') return redirect(url_for('image_get', filename='temp.jpg')) #return redirect(url_for('image')) return render_template('image.html', form=form)