gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: Resolve issue with ACLs

Browse source
This commit is contained in:
Igor Ilic 2025-06-04 16:36:57 +02:00
parent 289e395d0e
commit bdfa1704d9
1 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
cognee/modules/data/deletion/prune_system.py
View file

@ -3,6 +3,7 @@ from cognee.infrastructure.databases.graph.get_graph_engine import get_graph_eng
from cognee.infrastructure.databases.relational import get_relational_engine
from cognee.modules.data.methods import delete_dataset, get_authorized_existing_datasets
from cognee.modules.users.exceptions import PermissionDeniedError
from sqlalchemy import delete
async def prune_system(user=None, graph=True, vector=True, metadata=False):
@ -22,7 +23,15 @@ async def prune_system(user=None, graph=True, vector=True, metadata=False):
user_datasets = []
for dataset in user_datasets:
await delete_dataset(dataset)
# Delete ACLs related to dataset
from cognee.modules.users.models import ACL
from cognee.modules.data.models import Dataset
engine = get_relational_engine()
async with engine.get_async_session() as session:
await session.execute(delete(ACL).where(ACL.dataset_id == dataset.id))
await session.execute(delete(Dataset).where(Dataset.id == dataset.id))
await session.commit()
else:
if graph:
graph_engine = await get_graph_engine()