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