def registr():
form = LoginForm()
user_name = form.username.data
password = form.password.data
email = form.email.data
user_model = UserModel(db.get_connection())
names = [x[1] for x in UserModel(db.get_connection()).get_all()]
if request.method == 'POST':
if user_name in names:
return render_template('registr.html',
title='Регистрация',
form=form,
usernameerror=['никнейм занят'])
else:
a = check_password(password)
if a[0]:
user_model.insert(user_name, password, email)
follower_notification(user_name, email)
return redirect('/login')
else:
return render_template('registr.html',
title='Регистрация',
form=form,
passworderror=[a[1]])
return render_template('registr.html', title='Регистрация', form=form)
def add_admin(self):
name = self.name.text()
familyname = self.familyname.text()
fathername = self.fathername.text()
login = self.login.text()
number = self.number.text()
password = self.password.text()
lodger = UserModel(db.get_connection())
lodger.insert(login, password, familyname, name, fathername, number, 1)
self.close()
def post(self):
args = parser.parse_args()
if check_arg(args, 'username') and check_arg(args, 'password'):
users = UserModel(dbase.get_connection())
if not users.insert(args['username'], args['password']):
return redirect('/register')
exists = user_model.exists(user_name, password)
session['username'] = user_name
session['user_id'] = exists[1]
session['sort'] = 'date'
session['reverse'] = False
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
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 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 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,
admins['G' + str(n)].value)
n += 1
a = admins['A' + str(n)].value
