def process_reg():
form = RegistrationForm()
if form.validate_on_submit():
new_user = User(username=form.username.data,
email=form.email.data,
role='user')
if form.first_name.data:
new_user.first_name = form.first_name.data
if form.last_name.data:
new_user.last_name = form.last_name.data
if form.city.data:
new_user.city = form.city.data
new_user.set_password(form.password.data)
db.session.add(new_user)
db.session.commit()
flash('Вы успешно зарегистрировались')
return redirect(url_for('user.login'))
else:
for field, errors in form.errors.items():
for error in errors:
flash('Ошибка в поле {}: {}'.format(
getattr(form, field).label.text, error))
return redirect(url_for('user.register'))
def create_user():
with app.app_context(): # строка после которой можно работать с базой данных
user_email = input('Your email address: ')
if User.query.filter(User.user_email == user_email).count():
print('A user with the same name already exists.')
sys.exit(0)
password = getpass('Password: '******'Repeat password: '******'t match.")
sys.exit(0)
date_now = datetime.now()
date_reg = date_now.strftime('%d.%m.%Y')
picture_user = input('Write the name of the avatar:')
username = input('Write a nickname:')
new_user = User(user_email=user_email, role='admin', date_reg=date_reg, picture_user=picture_user, username=username)
new_user.set_password(password)
db.session.add(new_user)
db.session.commit()
print('User with id={} added'.format(new_user.id))
def register():
if current_user.is_authenticated:
return redirect(get_redirect_target())
form = RegistrationForm()
title = 'Регистрация'
if form.validate_on_submit():
new_user = User(username=form.username.data,
email=form.email.data,
role='user')
new_user.set_password(form.password.data)
db.session.add(new_user)
db.session.commit()
flash('Вы успешно зарегистрировались!')
return redirect(url_for('user.login'))
elif form.errors:
for field, errors in form.errors.items():
for error in errors:
flash('Ошибка в поле "{}": - {}'.format(
getattr(form, field).label.text, error))
return redirect(get_redirect_target())
return render_template('user/registration.html',
page_title=title,
form=form)
def process_reg():
form = RegistrationForm()
if form.validate_on_submit():
news_user = User(username=form.username.data, email=form.email.data, role='user')
news_user.set_password(form.password.data)
db.session.add(news_user)
db.session.commit()
flash('Вы успешно зарегистрировались!')
return redirect(url_for('user.login'))
flash('Пожалуйста, исправьте ошибки в форме')
return redirect(url_for('user.register'))
def register():
if current_user.is_authenticated:
return redirect(url_for('index'))
form = RegistrationForm()
if form.validate_on_submit():
user = User(username=form.username.data, email=form.email.data)
user.set_password(form.password.data)
db.session.add(user)
db.session.commit()
flash('Congratulations, you are now a registered user!')
return redirect(url_for('user.login'))
return render_template('user/register.jinja', title='Register', form=form)
def register():
if current_user.is_authenticated:
return redirect(url_for('home.index'))
title = 'Регистрация'
form = RegistrationForm()
if form.validate_on_submit():
user = User(username=form.username.data, email=form.email.data, role='user')
user.set_password(form.password.data)
db.session.add(user)
db.session.commit()
flash('Вы успешно зарегистрировались!')
return redirect(url_for('user.login'))
return render_template('user/register.html', page_title=title, form=form)
def process_registr():
form = RegistrationForm()
if form.validate_on_submit():
new_user = User(username=form.username.data,
email=form.email.data, status='user')
new_user.set_password(form.password)
db.session.add(new_user)
db.session.commit()
flash('Вы успешно зарегистрировались!')
return redirect(url_for('user.login'))
else:
for field, errors in form.errors.items():
for error in errors:
flash(f'Ошибка в поле {field} - {error}')
return redirect(url_for('user.registr'))
def edit(id):
form1 = EditUser()
form2 = ChangePassword(prefix="pass")
if id > 0:
user = User.query.filter_by(id=id).first()
if request.method == 'GET':
form1.id.data = user.id
form1.username.data = user.username
form1.email.data = user.email
form1.role.data = user.role
form2.id.data = user.id
if request.method == 'POST':
if form1.validate_on_submit():
user.username = form1.username.data
user.email = form1.email.data
user.role = form1.role.data
text = "Пользователь {} изменен".format(user.username)
elif form2.validate_on_submit():
user.set_password(form2.password.data)
text = "Пароль пользователя {} изменен".format(user.username)
else:
return jsonify(status='error', text='{}'.format(form1.errors))
try:
db.session.commit()
except IntegrityError:
text = "Ошибка сохранения пользователя {}".format(
user.username)
return jsonify(status='error', text=text)
return jsonify(status='ok', text=text)
else:
if form1.validate_on_submit():
new_user = User(username=form1.username.data,
email=form1.email.data,
role=form1.role.data)
new_pass = random_password()
new_user.set_password(new_pass)
db.session.add(new_user)
try:
db.session.commit()
except IntegrityError:
text = "Ошибка сохранения пользователя {}".format(
form1.username.data)
return jsonify(status='error', text=text)
text = "Пользователь {} добавлен. Пароль {}".format(
form1.username.data, new_pass)
return jsonify(status='ok', text=text)
html = render_template('user/edit.html', form1=form1, form2=form2)
return jsonify(html=html)
def process_reg():
form = RegistrationForm()
if form.validate_on_submit():
new_user = User(username=form.username.data,
email=form.email.data,
role="user")
new_user.set_password(form.password.data)
session.add(new_user)
session.commit()
flash('You registered successfuly!')
return redirect(url_for('user.login'))
else:
for field, errors in form.errors.items():
for error in errors:
flash(f'Error in "{getattr(form, field).label.text}": {error}')
return redirect(url_for('user.register'))
def process_signup():
form = RegistrationForm()
if form.validate_on_submit():
new_user = User(username=form.username.data,
email=form.email.data,
role="user")
new_user.set_password(form.password.data)
db.session.add(new_user)
db.session.commit()
flash("Account created!")
return redirect(url_for("user.login"))
else:
for field, errors in form.errors.items():
for error in errors:
flash('Error in "{}": {}'.format(
getattr(form, field).label.text, error))
return redirect(url_for("user.register"))
def process_reg():
form = RegistrationForm()
if form.validate_on_submit():
news_user = User(username=form.username.data,
email=form.email.data,
role='user')
news_user.set_password(form.password.data)
db.session.add(news_user)
db.session.commit()
flash('Registration successful')
return redirect(url_for('user.login'))
else:
for field, errors in form.errors.items():
for error in errors:
flash('Error in field {}: {}'.format(
getattr(form, field).label.text, error))
return redirect(url_for('user.register'))
def register_admin(username):
username = User.query.filter_by(username=username).first_or_404()
title = 'Регистрация нового админа'
form = RegistrationForm()
if form.validate_on_submit():
user = User(username=form.username.data,
email=form.email.data,
role='admin')
user.set_password(form.password.data)
db.session.add(user)
db.session.commit()
flash('Новый админ успешно зарегистрирован!')
return redirect(
url_for('admin.users_management', username=username.username))
return render_template('admin/register_admin.html',
page_title=title,
form=form)
def process_reg():
form = RegistrationForm()
if form.validate_on_submit():
new_user = User(username=form.username.data, role='user', status='[]')
new_user.set_password(form.password.data)
db.session.add(new_user)
db.session.commit()
flash('Вы успешно зарегистрировались!')
return redirect(url_for('user.login'))
else:
for field, errors in form.errors.items():
for error in errors:
flash(
f'Ошибка в поле "{getattr(form, field).label.text}": - {error}'
)
return redirect(url_for('user.register'))
def process_reg():
form = RegistrationForm()
if form.validate_on_submit():
new_user = User(username=form.username.data,
email=form.email.data,
role="user")
new_user.set_password(form.password.data)
db.session.add(new_user)
db.session.commit()
flash("Вы успешно Зарегестрировались!")
return redirect(url_for("user.login"))
else:
for field, errors in form.errors.items():
for error in errors:
flash("Ошибка в поле '{}': - {}".format(
getattr(form, field).label.text, error))
return redirect(url_for("user.register"))
def process_reg():
form = RegistrationForm()
if form.validate_on_submit():
new_user = User(username=form.username.data,
email=form.email.data,
role='user')
new_user.set_password(form.password.data)
db.session.add(new_user)
db.session.commit()
flash('Вы успешно заегистрировались!')
return redirect(url_for('user.login'))
else: # вывод ошибок
for field, errors in form.errors.items(
): # field - название поля, errors - список ошибок
for error in errors:
flash('Ошибка в поле "{}": - {}'.format(
getattr(form, field).label.text, error))
return redirect(url_for('user.register'))
def process_reg():
form = RegistrationForm()
if form.validate_on_submit():
new_user = User(username=form.username.data, role='user')
new_user.set_password(form.password.data)
db.session.add(new_user)
db.session.commit()
flash('Вы успешно зарегистрировались!')
return redirect(url_for('user.login'))
else:
for field, errors in form.errors.items():
for error in errors:
flash('Ошибка в поле "{}": - {}'.format(
getattr(form, field).label.text, error))
return redirect(url_for('user.register'))
flash('Пожалуйста, исправьте ошибки в форме')
return redirect(url_for('user.register'))
def process_reg():
form = RegistrationForm()
if form.validate_on_submit():
new_user = User(username=form.username.data,
email=form.email.data,
role=form.role.data)
new_user.set_password(form.password.data)
db.session.add(new_user)
db.session.commit()
flash('Пользователь успешно зарегистрирован!')
return redirect(url_for('admin.register'))
else:
for field, errors in form.errors.items():
for error in errors:
flash(
f'Ошибка в поле "{getattr(form, field).label.text}" - {error}'
)
return redirect(url_for('admin.register'))
def register():
"""Регистрация нового пользователя."""
title = 'Регистрация'
if current_user.is_authenticated:
return redirect(url_for('user.login'))
form = RegistrationForm()
if form.validate_on_submit():
user = User(username=form.username.data, email=form.email.data)
user.set_password(form.password.data)
db.session.add(user)
db.session.commit()
flash('You are now a registered user!', 'success')
return redirect(url_for('user.login'))
template_path = os.path.join('user', 'register.html')
return render_template(
template_path,
title=title,
form=form,
)
def process_registration():
form = RegistrationForm()
if form.validate_on_submit():
new_user = User(username=form.username.data,
email=form.email.data,
role='user')
new_user.set_password(form.password.data)
db.session.add(new_user)
db.session.commit()
flash('Вы успешно зарегистрировались, войдите под своим именем.')
return redirect(url_for('user.login'))
else:
for field, errors in form.errors.items():
for error in errors:
flash(
f"Ошибка в поле '{getattr(form, field).label.text}': - {error}"
)
return redirect(url_for('user.registration'))
flash('Пожалуйста, исправьте ошибки в форме')
return redirect(url_for('user.registration'))
def process_reg():
form = RegistrationForm()
if form.validate_on_submit():
new_user = User(username=form.username.data,
email=form.email.data,
role='user')
new_user.set_password(form.password.data)
db.session.add(new_user)
db.session.commit()
flash('Вы успешно зарегистрировались!')
login_user(new_user)
flash('Вы успешно вошли на сайт')
print(current_user.is_authenticated)
return redirect(url_for('advert.index'))
else:
for field, errors in form.errors.items():
for error in errors:
flash(
f'Ошибка в поле "{getattr(form, field).label.text}": - {error}'
)
return redirect(url_for('user.register'))
def process_reg():
form = Reg_form()
if form.validate_on_submit():
new_user = User(username=form.username.data,
email=form.email.data,
role='user',
is_active=True)
new_user.set_password(form.password.data)
db.session.add(new_user)
db.session.commit()
flash('Вы успешно зарегались')
return redirect(url_for('users.login'))
else:
for field, errors in form.errors.items():
for error in errors:
flash('ошибка в поле"{}":{}'.format(
getattr(form, field).label.text, error))
redirect(url_for('users.reg'))
flash('Форма невалидна, исправте')
return redirect(url_for('users.reg'))
def process_reg():
form = RegistrationForm()
if form.validate_on_submit():
news_user = User(username=form.username.data,
email=form.email.data,
role='user')
news_user.set_password(form.password.data)
db.session.add(news_user)
db.session.commit()
flash('Вы успешно зарегистрировались!')
# перенаправляем пользователя на форму входа на сайт
return redirect(url_for('user.login'))
else:
for field, errors in form.errors.items():
for error in errors:
flash(
f"Ошибка в поле '{getattr(form, field).label.text}': - {error}"
)
return redirect(url_for('user.register'))
def process_registration():
if current_user.is_authenticated:
return redirect(url_for('market.index'))
regform = RegistrationForm()
if regform.validate_on_submit():
new_user = User(number=f'8{regform.number.data}',
mail=regform.mail.data,
name=regform.name.data,
role='user')
new_user.set_password(regform.password.data)
db.session.add(new_user)
db.session.commit()
flash('Вы успешно зарегистрировались!')
logging.info(f'Новый пользователь с id={new_user.id}')
return redirect(url_for('market.index'))
else:
for field, errors in regform.errors.items():
for error in errors:
flash(f"В поле {getattr(regform, field).label.text}: -{error}")
return redirect(url_for('market.registration'))
def process_reg():
form = RegisterForm()
if form.validate_on_submit():
date_now = datetime.now()
date_reg = date_now.strftime('%d.%m.%Y')
new_user = User(username=form.username.data,
user_email=form.user_email.data,
user_password=form.user_password.data,
role='user',
date_reg=date_reg)
new_user.set_password(form.user_password.data)
db.session.add(new_user)
db.session.commit()
flash('You have successfully registered.', 'success')
return redirect(url_for('user.login'))
else:
for field, errors in form.errors.items():
for error in errors:
flash(
f"Error in the field {getattr(form, field).label.text}: {error}",
'warning')
return redirect(url_for('user.register'))
def register():
if current_user.is_authenticated:
return redirect(url_for('purchase.index'))
form = RegistrationForm()
if request.method == 'POST':
if form.validate_on_submit():
new_user = User(username=form.username.data,
email=form.email.data,
role='user')
new_user.set_password(form.password.data)
db.session.add(new_user)
db.session.commit()
flash(
'Вы успешно зарегистрировались! Для получения полного доступа к сайту, вам на почту отправлено письмо с инструкцией. Проверьте вашу почту!'
)
send_confirm_registration_email.delay(new_user)
return redirect(url_for('user.login'))
else:
for field, errors in form.errors.items():
for error in errors:
flash('Ошибка в поле "{}": - {}'.format(
getattr(form, field).label.text, error))
return redirect(url_for('user.register'))
return render_template('user/registration.html', form=form)
from getpass import getpass
import sys
from webapp import create_app
from webapp.db import db
from webapp.user.models import User
app = create_app()
with app.app_context():
username = input('Введите имя:')
if User.query.filter(User.username == username).count():
print('Пользователь с таким именем уже есть')
sys.exit(0)
password1 = getpass('Введите пароль')
password2 = getpass('Повторите пароль')
if not password1 == password2:
print('Пароли не одинаковые')
sys.exit(0)
new_user = User(username=username, role='admin')
new_user.set_password(password1)
db.session.add(new_user)
db.session.commit()
print('Создан пользователь с id={}'.format(new_user.id))
import sys # Модуль взаимодействия с системными функциями
from webapp import create_app
from webapp.db import db
from webapp.user.models import User
app = create_app() # Создаем апликейшн
with app.app_context(): # Запрашиваем имя
username = input('Введите имя пользователя: ')
if User.query.filter(User.username ==
username).count(): # Проверяем нет ли такого имени
print('Пользователь с тамим именем уже существует')
sys.exit(0)
password1 = getpass('Введите пароль: ')
password2 = getpass('Повторите пароль: ')
if not password1 == password2: # Проверяем одинаковый ли пароль
print('Пароли не одинаковые')
sys.exit(0)
new_user = User(
username=username,
role='admin') # Роль задаем по умолчанию админ, можно изменить
new_user.set_password(password1) # Делаем из пароля хэш
db.session.add(new_user)
db.session.commit()
print('Создан пользователь с id={}'.format(new_user.id))
from webapp.db import db
from webapp.catalog.models import Catalog
from webapp.user.models import User
app = create_app()
with app.app_context():
if not User.query.count():
users = ['admin', 'user']
for username in users:
if not User.query.filter_by(username=username).count():
if username == 'admin':
new_user = User(username=username, role='admin')
else:
new_user = User(username=username, role='user')
new_user.set_password(username)
db.session.add(new_user)
print(f'Добавлен пользователь {username}')
db.session.commit()
if not Catalog.query.count():
catalog = {
'Фрукты, овощи, ягоды': {
'Фрукты': ['Бананы', 'Яблоки', 'Авокадо'],
'Овощи': [
'Огурцы', 'Помидоры', 'Дайкон', 'Картофель', 'Лук',
'Морковь'
],
'Ягоды': ['Малина', 'Черника', 'Виктория']
},
'Молоко, сыр, яйца': {
from getpass import getpass # безопасный input
import sys # работа с системными функциями, будет использована для корректного завершения работы
from webapp import create_app
from webapp.db import db
from webapp.user.models import User
app = create_app()
with app.app_context(): # контекст необходим для локальной работы БД
username = input('Введите имя: ')
if User.query.filter(User.username == username).count(): # проверка существования пользователя (возвращает количество совпадений в БД)
print('Пользователь с таким именем уже существует')
sys.exit(0) # выход из программы
password1 = getpass('Веедите пароль: ')
password2 = getpass('Повторите пароль: ')
if not password1 == password2:
print('Пароли не одинаковые')
sys.exit(0)
new_user = User(username = username, role = 'admin')
new_user.set_password(password1) # использование метода set_password класса User
db.session.add(new_user)
db.session.commit()
print('Создан пользователь с id = {}'.format(new_user.id))
def test_password_hasing(self):
u = User(username='******')
u.set_password('cat')
self.assertFalse(u.check_password('dog'))
self.assertTrue(u.check_password('cat'))
