def create_roles(self):
self.role_vca = CategoryUserRole(user=self.user)
self.role_vca.group = self.group_vca
self.role_vca.save(sync_role=True)
self.role_vca.category.add(self.nested_first_level_two)
self.role_vca.save(sync_role=True)
self.role_all = CategoryUserRole(user=self.user)
self.role_all.group = self.group_all
self.role_all.save(sync_role=True)
self.role_all.category.add(self.nested_second_level_two)
self.role_all.save(sync_role=True)
def create_roles(self):
self.role_vca = CategoryUserRole(user=self.user)
self.role_vca.group = self.group_vca
self.role_vca.save(sync_role=True)
self.role_vca.category.add(self.nested_first_level_two)
self.role_vca.save(sync_role=True)
self.role_all = CategoryUserRole(user=self.user)
self.role_all.group = self.group_all
self.role_all.save(sync_role=True)
self.role_all.category.add(self.nested_second_level_two)
self.role_all.save(sync_role=True)
def create_users(run_transaction, verbosity, **kwargs):
map_permissions() # map permissions (and create them in actual database)
map_groups() # creates and maps all groups
# Basic categories for roles
basic_cat = []
basic_cat.append(Category.objects.get(slug='zena', tree_parent__isnull=True))
basic_cat.append(Category.objects.get(slug='bydleni', tree_parent__isnull=True))
basic_cat.append(Category.objects.get(slug=u'zdravi', tree_parent__isnull=True))
basic_cat.append(Category.objects.get(slug=u'recepty-hp', tree_parent__isnull=True))
gr_redaktor = Group.objects.get(name='redaktor')
gr_sefredaktor = Group.objects.get(name='sefredaktor')
# create users and assign default roles
cur = conn.cursor()
cur.execute(SQL_ALL_USERS)
num = int(cur.rowcount)
for idx in xrange(num):
row = cur.fetchone()
u, created = User.objects.get_or_create(
username=row[1],
first_name=row[2],
last_name=row[3],
email=row[4]
)
if not created:
continue
printv('User created %s' % row[1])
u.password = row[5]
u.is_staff = row[6]
u.is_active = row[7]
u.is_superuser = row[8]
u.save()
if not (u.is_staff and u.is_active) or u.is_superuser:
printv('User is superuser [%d] or is not staff [%d], role won\'t be created.' % (u.is_superuser, u.is_staff))
continue
# Create roles
if u.username.startswith('admin_'):
group = gr_sefredaktor
else:
group = gr_redaktor
roles = CategoryUserRole.objects.filter(user=u, group=group)
if roles:
printv('Role already exists for user [%s].' % u)
continue
printv('Creating role...')
role = CategoryUserRole(user=u, group=group)
role.save(sync_role=False)
map(lambda cat: role.category.add(cat), basic_cat)
role.save(sync_role=False)
printv('Role created for user [%s], in group [%s].' % (u, group))
class UserWithPermissionTestCase(NewmanTestCase):
def setUp(self):
super(UserWithPermissionTestCase, self).setUp()
self.create_permissions()
self.create_groups()
self.create_roles()
def create_permissions(self):
self.article_ct = ContentType.objects.get_for_model(Article)
for i in ['view']:
setattr(
self, "permission_%s_article" % i,
Permission.objects.create(content_type=self.article_ct,
codename='%s_article' % i,
name="Can view aritcle"))
def create_groups(self):
self.group_vca = Group.objects.create(
name=u'Permission Group: View, Change, Add')
for perm in ["view", "change", "add"]:
self.group_vca.permissions.add(
Permission.objects.get(content_type=self.article_ct,
codename="%s_article" % perm))
self.group_vca.save()
self.group_all = Group.objects.create(
name=u'Permission Group: Do Whatever Ya Want')
for perm in ["view", "change", "add", "delete"]:
self.group_all.permissions.add(
Permission.objects.get(content_type=self.article_ct,
codename="%s_article" % perm))
self.group_all.save()
def create_roles(self):
self.role_vca = CategoryUserRole(user=self.user)
self.role_vca.group = self.group_vca
self.role_vca.save(sync_role=True)
self.role_vca.category.add(self.nested_first_level_two)
self.role_vca.save(sync_role=True)
self.role_all = CategoryUserRole(user=self.user)
self.role_all.group = self.group_all
self.role_all.save(sync_role=True)
self.role_all.category.add(self.nested_second_level_two)
self.role_all.save(sync_role=True)
def _create_author_and_article(self):
article = Article.objects.create(title=u'Testable rabbit',
description=u'Perex',
category=self.nested_first_level_two)
ArticleContents.objects.create(title=u'Testable rabbit, 你好',
content=u'Long vehicle',
article=article)
article.authors.add(self.author)
article.save()
self.assert_equals(1, Article.objects.count())
return article
class UserWithPermissionTestCase(NewmanTestCase):
def setUp(self):
super(UserWithPermissionTestCase, self).setUp()
self.create_permissions()
self.create_groups()
self.create_roles()
def create_permissions(self):
self.article_ct = ContentType.objects.get_for_model(Article)
for i in ['view']:
setattr(self, "permission_%s_article" % i, Permission.objects.create(content_type=self.article_ct, codename='%s_article' % i, name="Can view aritcle"))
def create_groups(self):
self.group_vca = Group.objects.create(name=u'Permission Group: View, Change, Add')
for perm in ["view", "change", "add"]:
self.group_vca.permissions.add(Permission.objects.get(content_type=self.article_ct, codename="%s_article" % perm))
self.group_vca.save()
self.group_all = Group.objects.create(name=u'Permission Group: Do Whatever Ya Want')
for perm in ["view", "change", "add", "delete"]:
self.group_all.permissions.add(Permission.objects.get(content_type=self.article_ct, codename="%s_article" % perm))
self.group_all.save()
def create_roles(self):
self.role_vca = CategoryUserRole(user=self.user)
self.role_vca.group = self.group_vca
self.role_vca.save(sync_role=True)
self.role_vca.category.add(self.nested_first_level_two)
self.role_vca.save(sync_role=True)
self.role_all = CategoryUserRole(user=self.user)
self.role_all.group = self.group_all
self.role_all.save(sync_role=True)
self.role_all.category.add(self.nested_second_level_two)
self.role_all.save(sync_role=True)
def _create_author_and_article(self):
article = Article.objects.create(title=u'Testable rabbit', description=u'Perex', category=self.nested_first_level_two)
ArticleContents.objects.create(title=u'Testable rabbit, 你好', content=u'Long vehicle', article=article)
article.authors.add(self.author)
article.save()
self.assert_equals(1, Article.objects.count())
return article
def create_users(run_transaction, verbosity, **kwargs):
map_permissions() # map permissions (and create them in actual database)
map_groups() # creates and maps all groups
# Basic categories for roles
basic_cat = []
basic_cat.append(
Category.objects.get(slug='zena', tree_parent__isnull=True))
basic_cat.append(
Category.objects.get(slug='bydleni', tree_parent__isnull=True))
basic_cat.append(
Category.objects.get(slug=u'zdravi', tree_parent__isnull=True))
basic_cat.append(
Category.objects.get(slug=u'recepty-hp', tree_parent__isnull=True))
gr_redaktor = Group.objects.get(name='redaktor')
gr_sefredaktor = Group.objects.get(name='sefredaktor')
# create users and assign default roles
cur = conn.cursor()
cur.execute(SQL_ALL_USERS)
num = int(cur.rowcount)
for idx in xrange(num):
row = cur.fetchone()
u, created = User.objects.get_or_create(username=row[1],
first_name=row[2],
last_name=row[3],
email=row[4])
if not created:
continue
printv('User created %s' % row[1])
u.password = row[5]
u.is_staff = row[6]
u.is_active = row[7]
u.is_superuser = row[8]
u.save()
if not (u.is_staff and u.is_active) or u.is_superuser:
printv(
'User is superuser [%d] or is not staff [%d], role won\'t be created.'
% (u.is_superuser, u.is_staff))
continue
# Create roles
if u.username.startswith('admin_'):
group = gr_sefredaktor
else:
group = gr_redaktor
roles = CategoryUserRole.objects.filter(user=u, group=group)
if roles:
printv('Role already exists for user [%s].' % u)
continue
printv('Creating role...')
role = CategoryUserRole(user=u, group=group)
role.save(sync_role=False)
map(lambda cat: role.category.add(cat), basic_cat)
role.save(sync_role=False)
printv('Role created for user [%s], in group [%s].' % (u, group))
