def insert_user_permissions(doc, state):
user = doc.user
#first we remove all existing permissions for warehouse
clear_user_permissions(user, "Warehouse")
#next we add permissions
warehouses = doc.warehouse_allocated
#data ={}
for warehouse in warehouses:
docname = warehouse.warehouse_name
assigned = warehouse.assigned
if assigned:
frappe.permissions.add_user_permission("Warehouse", docname, user)
def test_clear_user_permissions(self):
current_user = frappe.session.user
frappe.set_user('Administrator')
clear_user_permissions_for_doctype('Blog Category', '*****@*****.**')
clear_user_permissions_for_doctype('Blog Post', '*****@*****.**')
add_user_permission('Blog Post', '-test-blog-post-1', '*****@*****.**')
add_user_permission('Blog Post', '-test-blog-post-2', '*****@*****.**')
add_user_permission("Blog Category", '_Test Blog Category 1', '*****@*****.**')
deleted_user_permission_count = clear_user_permissions('*****@*****.**', 'Blog Post')
self.assertEqual(deleted_user_permission_count, 2)
blog_post_user_permission_count = frappe.db.count('User Permission', filters={
'user': '******',
'allow': 'Blog Post'
})
self.assertEqual(blog_post_user_permission_count, 0)
blog_category_user_permission_count = frappe.db.count('User Permission', filters={
'user': '******',
'allow': 'Blog Category'
})
self.assertEqual(blog_category_user_permission_count, 1)
# reset the user
frappe.set_user(current_user)
def test_clear_user_permissions(self):
current_user = frappe.session.user
frappe.set_user("Administrator")
clear_user_permissions_for_doctype("Blog Category", "*****@*****.**")
clear_user_permissions_for_doctype("Blog Post", "*****@*****.**")
add_user_permission("Blog Post", "-test-blog-post-1", "*****@*****.**")
add_user_permission("Blog Post", "-test-blog-post-2", "*****@*****.**")
add_user_permission("Blog Category", "-test-blog-category-1", "*****@*****.**")
deleted_user_permission_count = clear_user_permissions("*****@*****.**", "Blog Post")
self.assertEqual(deleted_user_permission_count, 2)
blog_post_user_permission_count = frappe.db.count(
"User Permission", filters={"user": "******", "allow": "Blog Post"}
)
self.assertEqual(blog_post_user_permission_count, 0)
blog_category_user_permission_count = frappe.db.count(
"User Permission", filters={"user": "******", "allow": "Blog Category"}
)
self.assertEqual(blog_category_user_permission_count, 1)
# reset the user
frappe.set_user(current_user)
def test_clear_user_permissions(self):
current_user = frappe.session.user
frappe.set_user('Administrator')
clear_user_permissions_for_doctype('Blog Category', '*****@*****.**')
clear_user_permissions_for_doctype('Blog Post', '*****@*****.**')
add_user_permission('Blog Post', '-test-blog-post-1', '*****@*****.**')
add_user_permission('Blog Post', '-test-blog-post-2', '*****@*****.**')
add_user_permission("Blog Category", '-test-blog-category-1', '*****@*****.**')
deleted_user_permission_count = clear_user_permissions('*****@*****.**', 'Blog Post')
self.assertEqual(deleted_user_permission_count, 2)
blog_post_user_permission_count = frappe.db.count('User Permission', filters={
'user': '******',
'allow': 'Blog Post'
})
self.assertEqual(blog_post_user_permission_count, 0)
blog_category_user_permission_count = frappe.db.count('User Permission', filters={
'user': '******',
'allow': 'Blog Category'
})
self.assertEqual(blog_category_user_permission_count, 1)
# reset the user
frappe.set_user(current_user)
