def createRecords(categories, items):
from db.database import session, Item, Category, User, getTable
img = 'default.jpg'
#create users
for i in range(1, 6):
user = User(email='*****@*****.**' % i, name='User %s' % i)
session.add(user)
users = getTable(User)
#create categories
for i in range(1, categories + 1):
category = Category(name="Category %s" % i)
session.add(category)
#create items
for j in range(1, items + 1):
item = Item(name="Item %s-%s" % (i, j),
image=img,
category=category,
description="Description for item %s %s" % (j, LOREM),
user=users[randint(0, 4)])
session.add(item)
session.commit()
def register():
if current_user.is_authenticated:
return redirect(url_for('main.index'))
form = RegistrationForm()
if form.validate_on_submit():
phone = form.phone.data.replace(' ', '')
c = Company.query.get(form.company.data)
# TODO TEMPORARILY DISABLED
# if not c.check_token(form.token.data.strip()):
# flash("Neplatný registračný kód. Prosím kontaktujte vedenie Vašej firmy ohľadom poskytnutia platného kódu.",
# category='danger')
# return redirect(url_for('auth.register'))
# noinspection PyArgumentList
u = User(first_name=form.first_name.data,
surname=form.surname.data,
email=form.email.data,
phone=phone,
company=c,
email_subscription=form.email_subscription.data)
u.set_password(form.password.data)
u.set_urole(UserRole.BASIC)
session.add(u)
session.commit()
flash("Registrácia prebehla úspešne!", category='info')
return redirect(url_for('auth.login'))
return render_template('register.html', title='Registrácia', form=form)
def sendFeedback():
date = datetime.today()
name = request.form['name']
content = request.form['content']
if name and content:
session.add(Feedback(date=date, name=name, content=content))
session.commit()
return jsonify(success='Feedback sent!')
else:
return jsonify(error="Missing argument")
def reset_password(token):
if current_user.is_authenticated:
return redirect(url_for('main.index'))
user = User.verify_reset_password_token(token)
if not user:
return redirect(url_for('main.index'))
form = ResetPasswordForm()
if form.validate_on_submit():
user.set_password(form.password.data)
session.commit()
flash('Vaše heslo bolo úspešne zmenené.', category='info')
return redirect(url_for('auth.login'))
return render_template('reset_password.html', form=form)
def change_password():
form = ChangePasswordForm()
if form.validate_on_submit():
if not current_user.check_password(form.password.data):
flash("Zadali ste nesprávne heslo.", category='danger')
return redirect(url_for('auth.change_password'))
current_user.set_password(form.new_password.data)
session.commit()
flash('Vaše heslo bolo úspešne zmenené.', category='info')
return redirect(url_for('auth.account'))
return render_template('change_password.html',
title='Zmena hesla',
form=form)
def populate(path):
# Initialise database
initialise()
# Build database from JSON
with open(path) as file:
artists = load(file)
for i, artist in enumerate(artists['artists']):
artist.update(
gender='male' if artist['gender'] == 'M' else 'female')
artist.update(longitude=radians(float(artist['longitude'])))
artist.update(latitude=radians(float(artist['latitude'])))
session.add(Artist(**artist))
session.commit()
def delete_account():
form = DeleteAccountForm()
if form.validate_on_submit():
if not current_user.check_password(form.password.data):
flash("Zadali ste nesprávne heslo.", category='danger')
return redirect(url_for('auth.delete_account'))
session.delete(current_user)
session.commit()
flash(
"Váš účet bol úspešne vymazaný. Ďakujeme Vám za využívanie našej aplikácie a budeme radi,"
" ak sa čoskoro vrátite.",
category='info')
return redirect(url_for('auth.login'))
return render_template('delete_account.html',
title='Zruši účet',
form=form)
def account():
form = AccountForm()
email_form = EmailSubscriptionForm()
if request.method == 'GET':
form.first_name.data = current_user.first_name
form.surname.data = current_user.surname
form.email.data = current_user.email
form.phone.data = current_user.phone
form.company.data = current_user.company.title
email_form.email_subscription.data = current_user.email_subscription or False
elif email_form.validate_on_submit():
current_user.email_subscription = email_form.email_subscription.data
session.commit()
flash('Vaše zmeny boli úspešne uložené.', category='info')
return redirect(url_for('auth.account'))
return render_template('account.html',
title='Účet',
form=form,
email_form=email_form)
def edit_profile():
form = EditProfileForm(current_user.email)
if form.validate_on_submit():
current_user.first_name = form.first_name.data
current_user.surname = form.surname.data
current_user.email = form.email.data
current_user.phone = form.phone.data.replace(' ', '')
# current_user.company = form.company.data
session.commit()
flash('Vaše zmeny boli úspešne uložené.', category='info')
return redirect(url_for('auth.account'))
elif request.method == 'GET':
form.first_name.data = current_user.first_name
form.surname.data = current_user.surname
form.email.data = current_user.email
form.phone.data = current_user.phone
return render_template('edit_profile.html',
title='Zmena údajov',
form=form)
def deleteFeedback(id):
session.query(Feedback).filter_by(id=id).delete()
session.commit()
return jsonify(success='feedback deleted!')