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()