diff --git a/app.py b/app.py index 23b6dbc..f2a0f9b 100644 --- a/app.py +++ b/app.py @@ -13,14 +13,15 @@ from wtforms.validators import DataRequired, Length import requests from config import Config from geopy.geocoders import Nominatim - geolocator = Nominatim(user_agent="Bachemapa @ baches.qro.mx") + def create_app(config=Config): app = Flask(__name__) app.config.from_object(config) mongo = PyMongo(app) login_manager = LoginManager(app) + login_manager.login_view = 'thelogin' login_manager.session_protection = "strong" class User(UserMixin): @@ -209,20 +210,22 @@ def create_app(config=Config): @app.route('/dashboard') @login_required def dashboard(): - if request.method == 'GET': - if current_user.is_authenticated == False: - return redirect(url_for('thelogin')) - else: - invite_code = str(uuid4()) - qr_update = mongo.db.users.update_one({'_id': ObjectId(current_user.id)}, {'$set': {'referral_code': invite_code}}) - print(qr_update) - if not current_user.is_admin: - pins = list(mongo.db.pins.find({"added_by": current_user.id})) - return render_template('dashboard.html', pins=pins, invite_code=invite_code) - if current_user.is_admin: - users = list(mongo.db.users.find()) - pins = list(mongo.db.pins.find()) - return render_template('dashboard.html', users=users, pins=pins, invite_code=invite_code) + #if request.method == 'GET': + #if current_user.is_authenticated == False: + # return redirect(url_for('thelogin')) + #else: + old_qr = mongo.db.users.find_one({'_id': ObjectId(current_user.id)}) + print(old_qr) + invite_code = str(uuid4()) + qr_update = mongo.db.users.update_one({'_id': ObjectId(current_user.id)}, {'$set': {'referral_code': invite_code}}) + print(invite_code) + if not current_user.is_admin: + pins = list(mongo.db.pins.find({"added_by": current_user.id})) + return render_template('dashboard.html', pins=pins, invite_code=invite_code) + if current_user.is_admin: + users = list(mongo.db.users.find()) + pins = list(mongo.db.pins.find()) + return render_template('dashboard.html', users=users, pins=pins, invite_code=invite_code) @app.cli.command('add_user') def add_user(): diff --git a/config.py b/config.py index a5b5109..092a222 100644 --- a/config.py +++ b/config.py @@ -2,4 +2,5 @@ class Config: MONGO_URI = 'mongodb://localhost:27017/mapDB' UPLOAD_FOLDER = 'uploads' SECRET_KEY = 'supersecretkey' - ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'} \ No newline at end of file + ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'} + LOGIN_URL = '/thelogin' diff --git a/templates/dashboard.html b/templates/dashboard.html index 40c470e..334ffbd 100644 --- a/templates/dashboard.html +++ b/templates/dashboard.html @@ -15,7 +15,7 @@

Tu link es:

-

https://baches.qro.mx/registrame/{{current_user.referral_code}}

+

https://baches.qro.mx/registrame/{{invite_code}}

{% if pins %}