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

refactor: add comments

Browse source
This commit is contained in:
Igor Ilic 2025-07-23 16:59:08 +02:00
parent d780b54965
commit 98f2cd3305
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
examples/kuzu_migrate.py
View file

@ -7,6 +7,7 @@ This script migrates Kuzu databases between different versions by:
2. Exporting data from the source database using the old version 2. Exporting data from the source database using the old version
3. Importing data into the target database using the new version 3. Importing data into the target database using the new version
4. If overwrite is enabled target database will replace source database and source database will have the prefix _old 4. If overwrite is enabled target database will replace source database and source database will have the prefix _old
5. If delete-old is enabled target database will be renamed to source database and source database will be deleted
The script automatically handles: The script automatically handles:
- Environment setup (creates virtual environments as needed) - Environment setup (creates virtual environments as needed)
@ -15,8 +16,7 @@ The script automatically handles:
Usage Examples: Usage Examples:
# Basic migration from 0.9.0 to 0.11.0 # Basic migration from 0.9.0 to 0.11.0
python kuzu_migrate.py --old-version 0.9.0 --new-version 0.11.0 \\ python kuzu_migrate.py --old-version 0.9.0 --new-version 0.11.0 --old-db /path/to/old/database --new-db /path/to/new/database
--old-db /path/to/old/database --new-db /path/to/new/database
Requirements: Requirements:
- Python 3.7+ - Python 3.7+
@ -118,6 +118,8 @@ def rename_databases(old_db: str, new_db: str, delete_old: bool):
""" """
When overwrite is enabled, back up the original old_db (file with .lock and .wal or directory) When overwrite is enabled, back up the original old_db (file with .lock and .wal or directory)
by renaming it to *_old, and replace it with the newly imported new_db files. by renaming it to *_old, and replace it with the newly imported new_db files.
When delete_old is enabled replace the old database with the new one and delete old database
""" """
base_dir = os.path.dirname(old_db) base_dir = os.path.dirname(old_db)
name = os.path.basename(old_db.rstrip(os.sep)) name = os.path.basename(old_db.rstrip(os.sep))
@ -202,7 +204,7 @@ to isolate different Kuzu versions.
migrate(args.old_version, args.new_version, args.old_db, args.new_db) migrate(args.old_version, args.new_version, args.old_db, args.new_db)
if args.overwrite: if args.overwrite or args.delete_old:
rename_databases(args.old_db, args.new_db, args.delete_old) rename_databases(args.old_db, args.new_db, args.delete_old)
# migrate("0.9.0", "0.11.0", "/Users/igorilic/Desktop/cognee/cognee/.cognee_system/databases/cognee_graph_kuzu", "/Users/igorilic/Desktop/cognee/cognee/.cognee_system/databases/cognee_graph") # migrate("0.9.0", "0.11.0", "/Users/igorilic/Desktop/cognee/cognee/.cognee_system/databases/cognee_graph_kuzu", "/Users/igorilic/Desktop/cognee/cognee/.cognee_system/databases/cognee_graph")