forked from orson/bachemap
61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
from flask import Flask, render_template, request, redirect, url_for, flash
|
|
from flask_pymongo import PyMongo
|
|
from werkzeug.utils import secure_filename
|
|
import os
|
|
|
|
app = Flask(__name__)
|
|
app.config["MONGO_URI"] = "mongodb://localhost:27017/mapPinsDB"
|
|
app.config['UPLOAD_FOLDER'] = 'uploads'
|
|
app.config['SECRET_KEY'] = 'supersecretkey'
|
|
app.config['ALLOWED_EXTENSIONS'] = {'png', 'jpg', 'jpeg', 'gif'}
|
|
|
|
mongo = PyMongo(app)
|
|
|
|
from flask_wtf import FlaskForm
|
|
from wtforms import StringField, FileField, SubmitField
|
|
from wtforms.validators import DataRequired
|
|
|
|
class PinForm(FlaskForm):
|
|
description = StringField('Description', validators=[DataRequired()])
|
|
photo = FileField('Photo', validators=[DataRequired()])
|
|
submit = SubmitField('Add Pin')
|
|
|
|
def allowed_file(filename):
|
|
return '.' in filename and filename.rsplit('.', 1)[1].lower() in app.config['ALLOWED_EXTENSIONS']
|
|
|
|
@app.route('/')
|
|
def index():
|
|
pins = mongo.db.pins.find()
|
|
return render_template('index.html', pins=pins)
|
|
|
|
@app.route('/add_pin', methods=['GET', 'POST'])
|
|
def add_pin():
|
|
form = PinForm()
|
|
if form.validate_on_submit():
|
|
description = form.description.data
|
|
photo = form.photo.data
|
|
if photo and allowed_file(photo.filename):
|
|
try:
|
|
filename = secure_filename(photo.filename)
|
|
filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
|
|
photo.save(filepath)
|
|
|
|
pin = {
|
|
'description': description,
|
|
'photo': filepath,
|
|
'lat': request.form['lat'],
|
|
'lng': request.form['lng']
|
|
}
|
|
mongo.db.pins.insert_one(pin)
|
|
return redirect(url_for('index'))
|
|
except Exception as e:
|
|
flash(f'An error occurred: {e}')
|
|
return redirect(url_for('add_pin'))
|
|
else:
|
|
flash('Invalid file type. Only images are allowed.')
|
|
return render_template('add_pin.html', form=form)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app.run(debug=True)
|