def register():
form = RegisterForm()
if form.validate_on_submit():
if form.password.data != form.password_again.data:
return render_template('register.html',
form=form,
message="Пароли не совпадают")
session = db_session.create_session()
if session.query(User).filter(User.email == form.email.data).first():
return render_template('register.html',
form=form,
message="Такой пользователь уже есть")
user = User()
user.surname = form.surname.data
user.name = form.name.data
user.age = form.age.data
user.position = form.position.data
user.speciality = form.specially.data
user.address = form.address.data
user.email = form.email.data
user.city_from = form.city_from.data
user.set_password(form.password.data)
session.add(user)
session.commit()
return redirect('/login')
return render_template('register.html', form=form, title='Register')
def add_user():
if not request.json:
return jsonify({'error': 'Empty request'})
elif not all(key in request.json for key in
['id',
'surname',
'name',
'age',
'position',
'speciality',
'address',
'password',
'email',
'city_from']):
return jsonify({'error': 'Bad request'})
session = db_session.create_session()
if session.query(User).filter(User.id == request.json['id']).first():
return jsonify({'error': 'Id already exists'})
user = User()
user.id = request.json['id']
user.surname = request.json['surname']
user.name = request.json['name']
user.age = request.json['age']
user.position = request.json['position']
user.speciality = request.json['speciality']
user.address = request.json['address']
user.email = request.json['email']
user.city_from = request.json['city_from']
user.set_password(request.json['password'])
session.add(user)
session.commit()
return jsonify({'success': 'OK'})
def get_users():
session = db_session.create_session()
if request.method == "GET":
return jsonify([get_user(i).json for i in range(1, session.query(User).count() + 1)])
elif request.method == "POST":
if not request.json:
return jsonify({'error': 'Empty request'})
elif not all(key in (rj := request.json) for key in
{'address',
'position',
'speciality',
'surname',
'name',
'login',
'age',
'password',
'city_from'}):
return jsonify({'error': 'Bad request'})
elif session.query(Jobs).get(request.json['job_id']):
return jsonify(error="Id already exists")
user = User()
user.address = request.json['address']
user.position = request.json['position']
user.speciality = request.json['speciality']
user.surname = request.json['surname']
user.name = request.json['name']
user.email = request.json['login']
user.city_from = request.json['city_from']
user.age = int(request.json['age'])
user.modified_date = datetime.datetime.now()
user.set_password(request.json['password'])
session.add(user)
session.commit()
return jsonify(success="OK")
def create_user():
session = db_session.create_session()
if not request.json:
return jsonify({'error': 'Empty request'})
elif not all(key in request.json for key in
['id', 'surname', 'name', 'age', 'position', 'speciality', 'address', 'email',
'city_from', 'password']):
return jsonify({'error': 'Missing some keys to create, you need keys to create:'
'id, surname, name, age, position, speciality, address, email, city_from, password'})
if session.query(User).get(request.json.get('id')):
print('Id already exists')
session = db_session.create_session()
user = User()
user.id = request.json.get('id')
user.surname = request.json.get('surname')
user.name = request.json.get('name')
user.age = request.json.get('age')
user.position = request.json.get('position')
user.speciality = request.json.get('speciality')
user.address = request.json.get('address')
user.email = request.json.get('email')
user.city_from = request.json.get('city_from')
user.set_passwor(request.json.get('password'))
session.add(user)
session.commit()
return jsonify({'success': 'OK'})
def register():
session = db_session.create_session()
form = RegisterForm(request.form)
if request.method == "POST" and form.validate_on_submit():
if session.query(User).filter(User.email == form.login.data).first():
flash("Error with input Login: Invalid login")
else:
user = User()
user.address = form.address.data
user.position = form.position.data
user.speciality = form.speciality.data
user.surname = form.surname.data
user.name = form.name.data
user.email = form.login.data
user.city_from = form.city_from.data
user.age = int(form.age.data)
user.modified_date = datetime.datetime.now()
user.set_password(form.password.data)
session.add(user)
session.commit()
return redirect("/login")
if form.errors:
for k, v in form.errors.items():
flash(f"Error with input {k}: {v[0]}")
return render_template("standart.html",
base_title="Регистрация",
st_title="Register",
form=form,
current_user=current_user,
stpath=onepath)
def post(self):
args = parser.parse_args()
session = db_session.create_session()
if session.query(User).get(args['id']):
abort(400, message='Id already exists')
user = User()
user.id = args['id']
user.name = args['name']
user.surname = args['surname']
user.age = args['age']
user.position = args['position']
user.speciality = args['speciality']
user.address = args['address']
user.email = args['email']
user.city_from = args['city_from']
user.set_password(args['password'])
session.add(user)
session.commit()
return jsonify({'success': 'OK'})
def add_user_func(name,
city_from,
surname=None,
age=None,
position=None,
speciality=None,
address=None,
email=None,
hashed_password=None):
new_user = User()
new_user.surname = surname
new_user.name = name
new_user.age = age
new_user.city_from = city_from
new_user.position = position
new_user.speciality = speciality
new_user.address = address
new_user.email = email
new_user.hashed_password = hashed_password
session = db_session.create_session()
session.add(new_user)
session.commit()
def main():
db_session.global_init("db/astro.db")
capitan = User()
capitan.surname = "Scott"
capitan.name = "Ridley"
capitan.age = 21
capitan.position = "captain"
capitan.speciality = "research engineer"
capitan.address = "module_1"
capitan.email = "*****@*****.**"
capitan.set_password('password')
capitan.city_from = 'Berlin'
colon1 = User()
colon1.surname = "Weir"
colon1.name = "Andy"
colon1.age = 18
colon1.position = "common"
colon1.speciality = "worker"
colon1.address = "module_1"
colon1.email = "*****@*****.**"
colon1.city_from = 'New York'
colon2 = User()
colon2.surname = "Sanders"
colon2.name = "Teddy"
colon2.age = 19
colon2.position = "common"
colon2.speciality = "worker"
colon2.address = "module_2"
colon2.email = "*****@*****.**"
colon2.city_from = 'Paris'
category1 = Category()
category1.name = '1'
category2 = Category()
category2.name = '2'
category3 = Category()
category3.name = '3'
job1 = Jobs()
job1.team_leader = 1
job1.job = "deployment of residential modules 1 and 2"
job1.work_size = 15
job1.collaborators = "2, 3"
job1.start_date = datetime.datetime.now()
job1.is_finished = False
job1.categories.append(category1)
job2 = Jobs()
job2.team_leader = 2
job2.job = "Exploration of mineral resourses"
job2.work_size = 15
job2.collaborators = "4, 3"
job2.start_date = datetime.datetime.now()
job2.is_finished = False
job2.categories.append(category2)
job3 = Jobs()
job3.team_leader = 3
job3.job = "Development of management system"
job3.work_size = 25
job3.collaborators = "5"
job3.start_date = datetime.datetime.now()
job3.is_finished = True
job3.categories.append(category3)
department1 = Department()
department1.title = 'Department of geological exploration'
department1.chief = 2
department1.members = '1, 2, 3'
department1.email = '*****@*****.**'
department2 = Department()
department2.title = 'Department of biological research'
department2.chief = 3
department2.members = '2, 3'
department2.email = '*****@*****.**'
department3 = Department()
department3.title = 'Department of construction'
department3.chief = 1
department3.members = '1, 3'
department3.email = '*****@*****.**'
db_sess = db_session.create_session()
db_sess.add(capitan)
db_sess.add(colon1)
db_sess.add(colon2)
db_sess.add(job1)
db_sess.add(job2)
db_sess.add(job3)
db_sess.add(department1)
db_sess.add(department2)
db_sess.add(department3)
db_sess.add(category1)
db_sess.add(category2)
db_sess.add(category3)
db_sess.commit()
