def add_user(): if not current_user.is_authenticated: return redirect(url_for('login_blueprint.login')) if not current_user.user_role.can_add_edit_user: return render_template('errors/error-500.html'), 500 form = CreateUserForm() setup_user_form(form) form.process() if 'create' in request.form: new_dict = request.form.to_dict(flat=True) user = User.query.filter_by(username=new_dict['username']).first() if user: return route_template( 'users/add_user_and_profile', error_msg=_('Имя пользователя уже зарегистрировано'), form=form, change=None) # if "is_admin" in new_dict: # new_dict["is_admin"] = int(new_dict["is_admin"]) == 1 if 'region_id' in new_dict: if new_dict['region_id'] == '-1': new_dict['region_id'] = None user = User(**new_dict) db.session.add(user) db.session.commit() return redirect("{}?added_user".format(url_for('main_blueprint.users'))) else: return route_template( 'users/add_user_and_profile', form=form, change=None, error_msg=None, is_profile=False)
def test_unique_nickname(self): new_user1 = User(username='******', password='******', email='*****@*****.**', firstname='John', lastname='L', admin=False) db.session.add(new_user1) db.session.commit() try: new_user2 = User(username='******', password='******', email='*****@*****.**', firstname='John', lastname='L', admin=False) db.session.add(new_user2) db.session.commit() saved = True except: saved = False assert saved == False
def create_new_demo_user(username, machine): user = User(username=username) user.set_password("secret_bot_password!!!") db.session.add(user) db.session.commit() user_session = UserSession(user_id=user.id, machine_id=machine.id, device_ip="", timestamp_login=datetime.now().timestamp(), active=True) db.session.add(user_session) db.session.commit() return user
def create_user(): roll_no = request.values.get("roll_no") email = request.values.get("email") password = request.values.get("password") status = request.values.get("status") user = User(roll_no, email, password, status) try: db.session.add(user) db.session.commit() resp = {"success": "User added successfully!"} return jsonify(resp) except: db.session.rollback() resp = {"error": "User couldn't be added"} return jsonify(resp)
def new_user(): """ The screen to register a new user.""" form = RegisterForm() if form.validate_on_submit(): # noinspection PyArgumentList u = User(username=form.username.data) u.set_password(form.password.data) db.session.add(u) db.session.commit() current_app.logger.info(f"Created new user: {u}") return redirect(url_for('admin.admin_home')) nav_bar_title = "New User" return render_template("admin/newuser.html", title="Register", nav_bar_title=nav_bar_title, form=form)
def login(): if current_user.is_authenticated: return redirect(url_for('index.index')) form = LoginForm() if request.method == 'POST' and form.validate_on_submit(): next_page = request.args.get('next') user = User(form.username.data) if user.username and user.check_password(form.password.data): login_user(user) if not next_page or url_parse(next_page).netloc != '': next_page = url_for('index.home') else: flash('Incorrect username/password.') next_page = request.referrer return redirect(next_page) return render_template('login.html', form=form)
def setUp(self): db.create_all() db.session.add(User("admin", "*****@*****.**", "admin")) db.session.add(Post("Test post", "admin")) db.session.commit()
print(get_config_mode) try: config_mode = config_dict[get_config_mode.capitalize()] except KeyError: exit('Error: Invalid CONFIG_MODE environment variable entry.') app = create_app(config_mode) Migrate(app, db) engine = create_engine(config_mode.SQLALCHEMY_DATABASE_URI) from sqlalchemy.orm import sessionmaker Session = sessionmaker(bind=engine) session = Session() user = User() user.username = args["username"] user.password = hash_pass(args["password"]) user.user_role_id = args["admin_role_id"] # user.full_name = "test1" # user.organization = "test2" # user.telephone = "s" # user.email = "s" # user.region_id = 1 # user.username = args["username"] # user.organization = Column(String) # user.telephone = Column(String) # user.email = Column(String) session.add(user)
def create_user(): user = User(**request.form) db.session.add(user) db.session.commit() return jsonify('success')