def save(self):
user = User(username=self.username.data,
email=self.email.data,
password=self.password.data,
date_joined=datetime.utcnow(),
primary_group_id=4)
return user.save()
def user(default_groups):
"""Creates a user with normal permissions."""
user = User(username="******", email="*****@*****.**",
password="******", primary_group=default_groups[3],
activated=True)
user.save()
return user
def __add_group_to_user(self, user_discord_id):
roles = DiscordRole.query\
.join(DiscordUserRole)\
.join(DiscordUser)\
.filter(DiscordUser.id == user_discord_id)\
.all()
user = User.query.filter(User.discord == user_discord_id).first()
for role in roles:
if role.id in hub_current_server.discord_role_to_group:
group_id = hub_current_server.discord_role_to_group[role.id]
group = Group.query.filter(Group.id == group_id).first()
if not user:
discord_user = DiscordUser.query.filter(DiscordUser.id == user_discord_id).first()
assert discord_user
user = User(discord=user_discord_id, display_name=discord_user.pure_name, activated=True)
user.primary_group = group
else:
last_group = user.primary_group
user.primary_group = group
user.add_to_group(last_group)
user.save()
LogAction(current_user, "added user {name} (discord: {discord_id}) to group {group}".format(
name=user.display_name,
discord_id=user_discord_id,
group=group.name
))
flash("User {} was added to group {}".format(user.display_name, group.name))
return True
flash("Failed to find appropriate group for user {}".format(user.display_name), "danger")
return False
def admin_user(default_groups):
"""Creates a admin user."""
user = User(username="******", email="*****@*****.**",
password="******", primary_group=default_groups[0],
activated=True)
user.save()
return user
def super_moderator_user(default_groups):
"""Creates a super moderator user."""
user = User(username="******", email="*****@*****.**",
password="******", primary_group=default_groups[1],
activated=True)
user.save()
return user
def create_admin_user(username, password, email):
"""
Creates the administrator user
"""
admin_group = Group.query.filter_by(admin=True).first()
user = User(username=username, password=password, email=email,
date_joined=datetime.utcnow(), primary_group_id=admin_group.id)
user.save()
def post(self):
args = self.reqparse.parse_args()
user = User(username=args['username'],
password=args['password'],
email=args['email'],
date_joined=datetime.utcnow(),
primary_group_id=4)
user.save()
return {'user': marshal(user, user_fields)}, 201
def save(self):
data = self.data
if not data["username"]:
data["username"] = None
if not data["email"]:
data["email"] = None
data.pop('submit', None)
data.pop('csrf_token', None)
user = User(**data)
return user.save()
def unactivated_user(default_groups):
"""
Creates an unactivated user in the default user group
"""
user = User(username='******',
email='*****@*****.**',
password='******',
primary_group=default_groups[3],
activated=False)
user.save()
return user
def moderator_user(user, forum, default_groups):
"""Creates a test user with moderator permissions."""
user = User(username="******", email="*****@*****.**",
password="******", primary_group=default_groups[2],
activated=True)
user.save()
forum.moderators.append(user)
forum.save()
return user
def Fred(default_groups):
"""Fred is an interloper and bad intentioned user, he attempts to
access areas he shouldn't, he posts trollish and spammy content,
he does everything he can to destroy the board.
Our job is stop Fred.
"""
fred = User(username='******', email='*****@*****.**',
password='******', primary_group=default_groups[3],
activated=True)
fred.save()
return fred
def create_admin_user(username, password, email):
"""
Creates the administrator user
"""
admin_group = Group.query.filter_by(admin=True).first()
user = User()
user.username = username
user.password = password
user.email = email
user.primary_group_id = admin_group.id
user.save()
def create_test_data():
"""Creates 5 users, 2 categories and 2 forums in each category.
It also creates a new topic topic in each forum with a post.
"""
create_default_groups()
create_default_settings()
# create 5 users
for u in range(1, 6):
username = "******" % u
email = "*****@*****.**" % u
user = User(username=username, password="******", email=email)
user.primary_group_id = u
user.save()
user1 = User.query.filter_by(id=1).first()
user2 = User.query.filter_by(id=2).first()
# create 2 categories
for i in range(1, 3):
category_title = "Test Category %s" % i
category = Category(title=category_title,
description="Test Description")
category.save()
# create 2 forums in each category
for j in range(1, 3):
if i == 2:
j += 2
forum_title = "Test Forum %s %s" % (j, i)
forum = Forum(title=forum_title,
description="Test Description",
category_id=i)
forum.save()
# create a topic
topic = Topic()
post = Post()
topic.title = "Test Title %s" % j
post.content = "Test Content"
topic.save(post=post, user=user1, forum=forum)
# create a second post in the forum
post = Post()
post.content = "Test Post"
post.save(user=user2, topic=topic)
def save_user(text):
text = text.split("\n")
for username in text:
# print('saving', username)
try:
username = username.decode('ascii')
user = User.query.filter(User.username == username).all()
if len(user) == 0 and len(username) > 1:
user = User(username=username,
email="{}@gmail.com".format(username),
_password="******",
primary_group_id=4,
activated=1)
user.save()
except:
print('not asciiable')
def create_admin_user(username, password, email):
"""Creates the administrator user.
Returns the created admin user.
:param username: The username of the user.
:param password: The password of the user.
:param email: The email address of the user.
"""
admin_group = Group.query.filter_by(admin=True).first()
user = User()
user.username = username
user.password = password
user.email = email
user.primary_group_id = admin_group.id
user.activated = True
user.save()
return user
def test_has_custom_rank_returns_false_if_user_rank_isnt_custom(self):
assert not Rank.has_custom_rank(User())
def test_has_custom_rank_returns_true_if_user_rank_is_custom(self):
assert Rank.has_custom_rank(
User(user_rank=UserRank(rank=Rank(requirement=None))))
def save(self):
data = self.data
data.pop('submit', None)
data.pop('csrf_token', None)
user = User(**data)
return user.save()
def test_has_rank_returns_true_if_user_has_rank(self):
assert Rank.has_rank(
User(user_rank=UserRank(rank=Rank(requirement=1))))
def create_test_data(users=5, categories=2, forums=2, topics=1, posts=1):
"""Creates 5 users, 2 categories and 2 forums in each category.
It also creates a new topic topic in each forum with a post.
Returns the amount of created users, categories, forums, topics and posts
as a dict.
:param users: The number of users.
:param categories: The number of categories.
:param forums: The number of forums which are created in each category.
:param topics: The number of topics which are created in each forum.
:param posts: The number of posts which are created in each topic.
"""
create_default_groups()
create_default_settings()
data_created = {'users': 0, 'categories': 0, 'forums': 0,
'topics': 0, 'posts': 0}
# create 5 users
for u in range(1, users + 1):
username = "******" % u
email = "*****@*****.**" % u
user = User(username=username, password="******", email=email)
user.primary_group_id = u
user.activated = True
user.save()
data_created['users'] += 1
user1 = User.query.filter_by(id=1).first()
user2 = User.query.filter_by(id=2).first()
# lets send them a few private messages
for i in range(1, 3):
# TODO
pass
# create 2 categories
for i in range(1, categories + 1):
category_title = "Test Category %s" % i
category = Category(title=category_title,
description="Test Description")
category.save()
data_created['categories'] += 1
# create 2 forums in each category
for j in range(1, forums + 1):
if i == 2:
j += 2
forum_title = "Test Forum %s %s" % (j, i)
forum = Forum(title=forum_title, description="Test Description",
category_id=i)
forum.save()
data_created['forums'] += 1
for t in range(1, topics + 1):
# create a topic
topic = Topic(title="Test Title %s" % j)
post = Post(content="Test Content")
topic.save(post=post, user=user1, forum=forum)
data_created['topics'] += 1
for p in range(1, posts + 1):
# create a second post in the forum
post = Post(content="Test Post")
post.save(user=user2, topic=topic)
data_created['posts'] += 1
return data_created
def save(self):
user = User(**self.data)
return user.save()
def test_has_rank_returns_false_if_user_doesnt_have_rank(self):
assert not Rank.has_rank(User())
