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 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 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 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 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_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 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')