def create_user(dbsession:Session, registry:Registry, email:str=EMAIL, password:str=PASSWORD, admin:bool=False) -> User:
"""A helper function to create normal and admin users for tests.
:param admin: If True run :py:class:`websauna.system.user.usermixin.SiteCreator` login and set the user to admin group.
"""
user = User(email=email)
if password:
hasher = registry.getUtility(IPasswordHasher)
user.hashed_password = hasher.hash_password(password)
user.user_registration_source = "dummy"
dbsession.add(user)
dbsession.flush()
user.username = user.generate_username()
user.activated_at = now()
assert user.can_login()
# First user, make it admin
if admin:
site_creator = get_site_creator(registry)
site_creator.init_empty_site(dbsession, user)
return user
def create_activation(request, user):
"""Create through-the-web user sign up with his/her email.
We don't want to force the users to pick up an usernames, so we just generate an username.
The user is free to change their username later.
"""
assert user.id
user.username = user.generate_username()
user.registration_source = "email"
db = get_session(request)
Activation = request.registry.getUtility(IActivationClass)
activation = Activation()
db.add(activation)
user.activation = activation
db.flush()
# TODO Create a hook for this, don't assume create_activation() call
# TODO We don't need pystache just for this!
context = {
'link': request.route_url('activate', user_id=uuid_to_slug(user.uuid), code=user.activation.code)
}
send_templated_mail(request, [user.email], "login/email/activate", context)
site_creator = get_site_creator(request.registry)
site_creator.check_empty_site_init(request.dbsession, user)
def create_user(dbsession: Session,
registry: Registry,
email: str = EMAIL,
password: str = PASSWORD,
admin: bool = False) -> User:
"""A helper function to create normal and admin users for tests.
:param admin: If True run :py:class:`websauna.system.user.usermixin.SiteCreator` login and set the user to admin group.
"""
user = User(email=email, password=password)
user.user_registration_source = "dummy"
dbsession.add(user)
dbsession.flush()
user.username = user.generate_username()
user.activated_at = now()
assert user.can_login()
# First user, make it admin
if admin:
site_creator = get_site_creator(registry)
site_creator.init_empty_site(dbsession, user)
return user
def main(argv=sys.argv):
if len(argv) < 3:
usage(argv)
config_uri = argv[1]
request = init_websauna(config_uri)
User = get_user_class(request.registry)
dbsession = request.dbsession
if len(argv) == 4:
password = argv[3]
else:
password = getpass.getpass("Password:"******"Password (again):")
if password != password2:
sys.exit("Password did not match")
with transaction.manager:
u = User(email=argv[2], username=argv[2])
u.password = password
u.registration_source = "command_line"
u.activated_at = now()
dbsession.add(u)
dbsession.flush()
site_creator = get_site_creator(request.registry)
site_creator.init_empty_site(dbsession, u)
print("Created user #{}: {}, admin: {}".format(u.id, u.email, u.is_admin()))
def site_init(e: UserCreated):
"""Initialize the website.
When the first user hits the site, capture its login and add him to the admin group.
"""
request = e.request
user = e.user
site_creator = get_site_creator(request.registry)
site_creator.check_empty_site_init(request.dbsession, user)
def create_user(dbsession: Session,
registry: Registry,
email: str = EMAIL,
password: str = PASSWORD,
admin: bool = False) -> User:
"""A helper function to create normal and admin users for tests.
Example:
.. code-block:: python
import transaction
from websauna.tests.utils import create_user
def test_some_stuff(dbsession, registry):
with transaction.manager:
u = create_user(registry)
# Do stuff with new user
:param email: User's email address. If inot given use unit testing default.
:param password: Password as plain text. If not given use unit testing default.
:param admin: If True run :py:class:`websauna.system.user.usermixin.SiteCreator` login and set the user to admin group.
"""
user = User(email=email)
if password:
hasher = registry.getUtility(IPasswordHasher)
user.hashed_password = hasher.hash_password(password)
user.user_registration_source = "dummy"
dbsession.add(user)
dbsession.flush()
user.username = user.generate_username()
user.activated_at = now()
assert user.can_login()
# First user, make it admin
if admin:
site_creator = get_site_creator(registry)
site_creator.init_empty_site(dbsession, user)
return user
def test_virgin_init_admin(init, dbsession):
"""When the user logs in first time see we create admin group and permissions for it."""
with transaction.manager:
u = User(username="******", email="*****@*****.**")
dbsession.add(u)
dbsession.flush()
assert not u.is_admin()
site_creator = get_site_creator(init.config.registry)
site_creator.init_empty_site(dbsession, u)
with transaction.manager:
u = dbsession.query(User).get(1)
assert dbsession.query(Group).count() == 1
assert u.is_admin()
def create_user(dbsession: Session, registry: Registry, email: str=EMAIL, password: str=PASSWORD, admin: bool=False) -> User:
"""A helper function to create normal and admin users for tests.
Example:
.. code-block:: python
import transaction
from websauna.tests.utils import create_user
def test_some_stuff(dbsession, registry):
with transaction.manager:
u = create_user(registry)
# Do stuff with new user
:param email: User's email address. If inot given use unit testing default.
:param password: Password as plain text. If not given use unit testing default.
:param admin: If True run :py:class:`websauna.system.user.usermixin.SiteCreator` login and set the user to admin group.
"""
user = User(email=email)
if password:
hasher = registry.getUtility(IPasswordHasher)
user.hashed_password = hasher.hash_password(password)
user.user_registration_source = "dummy"
dbsession.add(user)
dbsession.flush()
user.username = user.generate_username()
user.activated_at = now()
assert user.can_login()
# First user, make it admin
if admin:
site_creator = get_site_creator(registry)
site_creator.init_empty_site(dbsession, user)
return user
def test_enter_admin(web_server, browser, dbsession, init):
"""The first user can open the admin page."""
with transaction.manager:
u = create_user(dbsession, init.config.registry)
site_creator = get_site_creator(init.config.registry)
site_creator.init_empty_site(dbsession, u)
assert u.is_admin()
b = browser
b.visit(web_server + "/login")
b.fill("username", EMAIL)
b.fill("password", PASSWORD)
b.find_by_name("login_email").click()
assert b.is_element_visible_by_css("#nav-admin")
b.find_by_css("#nav-admin").click()
assert b.is_element_present_by_css("#admin-main")
def test_enter_admin(web_server, browser, dbsession, init):
"""The first user can open the admin page."""
with transaction.manager:
u = create_user(dbsession, init.config.registry)
site_creator = get_site_creator(init.config.registry)
site_creator.init_empty_site(dbsession, u)
assert u.is_admin()
b = browser
b.visit(web_server + "/login")
b.fill("username", EMAIL)
b.fill("password", PASSWORD)
b.find_by_name("login_email").click()
assert b.is_element_visible_by_css("#nav-admin")
b.find_by_css("#nav-admin").click()
assert b.is_element_present_by_css("#admin-main")
def prepare_new_site(self, registry, dbsession, user):
"""If this is the first user on the site, initialize groups and give this user admin permissions."""
# XXX: Move this to an event
site_creator = get_site_creator(registry)
site_creator.init_empty_site(dbsession, user)
