from pymilvus import connections, db
from pymilvus import CollectionSchema, FieldSchema
from pymilvus import Collection
from pymilvus import connections
from pymilvus import DataType
from pymilvus import Partition
from pymilvus import utility
from pymilvus import MilvusClient
from pymilvus import Role

connections.connect(host="localhost", user="root", password="Milvus")
client = MilvusClient(uri="http://localhost:19530", user="root", password="Milvus")
client.create_collection("col1", dimension=128)
db.create_database("db1")
role = Role("role")
role.create()
utility.list_roles(True)
utility.create_user(user="user1", password="Milvus")
role.add_user("user1")
utility.list_roles(True)
role.list_grants()

role.grant_v2("ClusterAdmin", "*", db_name="*")

role.list_grants()

connections.disconnect('default')
connections.connect(host="localhost", user="user1", password="Milvus")

role.create_privilege_group("pg")

role.add_privileges_to_group("pg", ["Query","Insert"])

role.list_privilege_groups()
#PrivilegeGroupInfo groups:
#- PrivilegeGroupItem: <privilege_group:pg>, <privileges:('Query', 'Insert')>

role.remove_privileges_from_group("pg", ["Query"])
role.list_privilege_groups()
#PrivilegeGroupInfo groups:
#- PrivilegeGroupItem: <privilege_group:pg>, <privileges:('Insert',)>

role.drop_privilege_group("pg")