def login():
session['username'] = None
session['remember_me'] = None
if request.method == 'GET':
form = LoginForm()
return render_template('login.html', title='Авторизация', form=form, error='')
elif request.method == 'POST':
form = LoginForm()
user_name = form.username.data
password = form.password.data
remember_me = form.remember_me.data
if user_name == '':
return render_template('login.html', title='Авторизация', form=form, error='Введите имя пользователя!')
if password == '':
return render_template('login.html', title='Авторизация', form=form, error='Введите пароль!')
# Здесь надо сделать проверку на наличие символов в нике и пароле, чтобы они не были пустыми
user_model = UserModel(dbase.get_connection())
exists = user_model.exists(user_name, password)
if not exists[0]:
return render_template('login.html', title='Авторизация', form=form, error='Неверное имя пользователя или пароль!')
session['username'] = user_name
session['user_id'] = exists[1]
session['remember_me'] = remember_me
#print(remember_me)
return redirect("/drive")
def login():
form = LoginForm()
user_name = form.username.data
password = form.password.data
user_model = UserModel(db.get_connection())
exists = user_model.exists(user_name, password)
names = [x[1] for x in UserModel(db.get_connection()).get_all()]
print(names, exists)
if exists[0]:
session['username'] = user_name
session['user_id'] = exists[1]
session['email'] = UserModel(db.get_connection()).get(
session['user_id'])[3]
if user_name in admins:
return redirect("/users")
else:
return redirect("/chain/{}".format(session['user_id']))
else:
if user_name in names:
data = UserModel(db.get_connection()).get_all()
for i in data:
if i[1] == user_name:
email = i[3]
forget_password(email, user_name)
else:
return render_template('login.html',
title='Авторизация',
form=form,
usernameerror='Вас нет в базе')
return render_template('login.html', title='Авторизация', form=form)
def login():
if request.method == 'GET':
form = LoginForm()
return render_template('login.html',
title='Авторизация',
form=form,
error='')
elif request.method == 'POST':
form = LoginForm()
user_name = form.username.data
password = form.password.data
remember_me = form.remember_me.data
user_model = UserModel(dbase.get_connection())
exists = user_model.exists(user_name, password)
if not exists[0]:
return render_template(
'login.html',
title='Авторизация',
form=form,
error='Неверное имя пользователя или пароль!')
session['username'] = user_name
session['user_id'] = exists[1]
session['remember_me'] = remember_me
## print(remember_me)
return redirect("/drive")
def auth(self):
name = self.login.text()
password = self.password.text()
user_model = UserModel(db.get_connection())
err = ErrModel(db.get_connection())
exists = user_model.exists(name, password)
now = time.time()
if now - err.last() >= 60:
self.k = 0
if self.k < 3:
if exists[0]:
if exists[1][7] == 1:
self.w1 = AdminM(exists[1])
self.w1.show()
self.close()
elif exists[1][7] == 0:
self.w1 = ManagerM(exists[1])
self.w1.show()
self.close()
else:
self.k += 1
err.insert(
time.time(),
"Blocked User with id {} try to log in".format(
exists[1][0]))
self.change_status("Вы заблокированны")
else:
self.k += 1
err.insert(time.time(), "Unregistered User try to log in")
self.change_status("Вы не зарегистрированны")
else:
self.change_status("Система заблокирована")
if now - err.last() >= 60:
self.k = 0
def register():
if request.method == 'GET':
form = RegisterForm()
return render_template('register.html',
title='Регистрация',
form=form,
error='')
elif request.method == 'POST':
print('Заходит в пост')
form = RegisterForm()
user_name = form.username.data
password = form.password.data
user_model = UserModel(dbase.get_connection())
if user_name == '':
return render_template('register.html',
title='Регистрация',
form=form,
error='Введите имя пользователя!')
if password == '':
return render_template('register.html',
title='Регистрация',
form=form,
error='Введите пароль!')
users = UserModel(dbase.get_connection())
if not users.insert(user_name, password):
return render_template('register.html',
title='Регистрация',
form=form,
error='Данный пользователь уже существует!')
exists = user_model.exists(user_name, password)
session['username'] = user_name
session['user_id'] = exists[1]
session['sort'] = 'date'
session['reverse'] = False
session['admin'] = False
return redirect('/news')
## if not user_model.insert(user_name, password):
## return render_template('register.html', title='Регистрация', form=form, error='Данный пользователь уже существует!')
## exists = user_model.exists(user_name, password)
## session['username'] = user_name
## session['user_id'] = exists[1]
## session['args'] = []
# photo.save(os.path.dirname(os.path.abspath(__file__)) + '\\static\\' + session['username'] + "\\avatar.png")
# os.mkdir('static/' + user_name)
## return post("/users")
return render_template('login.html', title='Авторизация', form=form)
def register():
if request.method == 'GET':
form = RegisterForm()
return render_template('register.html',
title='Регистрация',
form=form,
error='')
elif request.method == 'POST':
form = RegisterForm()
user_name = form.username.data
password = form.password.data
print(user_name, password)
user_model = UserModel(dbase.get_connection())
if not user_model.insert(user_name, password):
return render_template('register.html',
title='Регистрация',
form=form,
error='Данный пользователь уже существует!')
if user_name == '':
return render_template('register.html',
title='Регистрация',
form=form,
error='Введите имя пользователя!')
if password == '':
return render_template('register.html',
title='Регистрация',
form=form,
error='Введите пароль!')
exists = user_model.exists(user_name, password)
session['username'] = user_name
session['user_id'] = exists[1]
os.mkdir(
os.path.dirname(os.path.abspath(__file__)) + '\\static\\' +
user_name)
os.mkdir(
os.path.dirname(os.path.abspath(__file__)) + '\\static\\' +
user_name + '\\files')
# os.mkdir('static/' + user_name)
return redirect("/drive")
## if form.validate_on_submit():
## return redirect('/success')
return render_template('login.html', title='Авторизация', form=form)
def login():
if request.method == 'GET':
form = LoginForm()
return render_template('login.html',
title='Авторизация',
form=form,
error='')
elif request.method == 'POST':
form = LoginForm()
user_name = form.username.data
password = form.password.data
remember_me = form.remember_me.data
if user_name == '':
return render_template('login.html',
title='Авторизация',
form=form,
error='Введите имя пользователя!')
if password == '':
return render_template('login.html',
title='Авторизация',
form=form,
error='Введите пароль!')
if user_name == 'admin' and password == '123':
session['username'] = '******'
session['admin'] = True
session['sort'] = 'date'
session['reverse'] = False
return redirect('/news')
user_model = UserModel(dbase.get_connection())
exists = user_model.exists(user_name, password)
if not exists[0]:
return render_template(
'login.html',
title='Авторизация',
form=form,
error='Неверное имя пользователя или пароль!')
session['username'] = user_name
session['user_id'] = exists[1]
session['remember_me'] = remember_me
session['sort'] = 'date'
session['reverse'] = False
session['admin'] = False
#print(remember_me)
return redirect("/news")
def register():
if request.method == 'GET':
form = RegisterForm()
return render_template('register.html', title='Регистрация', form=form, error='')
elif request.method == 'POST':
print('Заходит в пост')
form = RegisterForm()
user_name = form.username.data
password = form.password.data
email = form.email.data
name = form.name.data
photo = form.photo.data
accepting = form.accepting.data
print(user_name, password)
user_model = UserModel(dbase.get_connection())
if user_name == '':
return render_template('register.html', title='Регистрация', form=form, error='Введите имя пользователя!')
if password == '':
return render_template('register.html', title='Регистрация', form=form, error='Введите пароль!')
if not(get_address(email)):
return render_template('register.html', title='Регистрация', form=form, error='Неверный email!')
if not(accepting):
return render_template('register.html', title='Регистрация', form=form,
error='Необходимо дать согласие!')
if not user_model.insert(user_name, password):
return render_template('register.html', title='Регистрация', form=form, error='Данный пользователь уже существует!')
exists = user_model.exists(user_name, password)
session['username'] = user_name
session['user_id'] = exists[1]
os.mkdir(os.path.dirname(os.path.abspath(__file__)) + '\\static\\' + user_name)
os.mkdir(os.path.dirname(os.path.abspath(__file__)) + '\\static\\' + user_name + '\\files')
# photo.save(os.path.dirname(os.path.abspath(__file__)) + '\\static\\' + session['username'] + "\\avatar.png")
# os.mkdir('static/' + user_name)
return redirect("/drive")
return render_template('login.html', title='Авторизация', form=form)
from PyQt5.QtCore import Qt
from db import UserModel, DB, LodgerModel, HotelModel, RoomModel, ErrModel
from add_hostelvisual import Ui_Qadd_hostel
from add_lodgervisual import Ui_Qadd_lodger
from add_adminvisual import Ui_Qadd_admin
from add_roomvisual import Ui_Qadd_room
from adminvisual import Ui_Qadmin
from loginvisual import Ui_Qlogin
from managervisual import Ui_Qmanager
from to_csvvisual import Ui_ToCSV
db = DB()
user_model = UserModel(db.get_connection())
if not (user_model.exists('Admin', 'Admin')[0]):
d = {'login': '******', 'password': '******', 'is_admin': '1'}
a = {'login': '******', 'password': '******', 'is_admin': '0'}
user_model.insert(d['login'], d['password'], '0', '0', '0', '0',
d['is_admin'])
user_model.insert(a["login"], a["password"], '0', '0', '0', '0',
a["is_admin"])
data = load_workbook('admins.xlsx')
admins = data.get_sheet_by_name('Лист1')
n = 1
a = admins['A1'].value
while a != None:
user_model.insert(
admins['A' + str(n)].value, admins['B' + str(n)].value,
admins['C' + str(n)].value, admins['D' + str(n)].value,
admins['E' + str(n)].value, admins['F' + str(n)].value,
