loggers for clean, drop and init

Author: Zhe Yu

src/vectorcode/subcommands/clean.py

@@ -1,16 +1,21 @@
+import logging
 import os
 
 from chromadb.api import AsyncClientAPI
 
 from vectorcode.cli_utils import Config
 from vectorcode.common import get_client, get_collections
 
+logger = logging.getLogger(name=__name__)
+
 
 async def run_clean_on_client(client: AsyncClientAPI, pipe_mode: bool):
     async for collection in get_collections(client):
         meta = collection.metadata
+        logger.debug(f"{meta.get('path')}: {await collection.count()} chunk(s)")
         if await collection.count() == 0 or not os.path.isdir(meta["path"]):
             await client.delete_collection(collection.name)
+            logger.info(f"Deleted collection for {meta['path']}")
             if not pipe_mode:
                 print(f"Deleted {meta['path']}.")
 

src/vectorcode/subcommands/drop.py

@@ -1,8 +1,12 @@
+import logging
+
 from chromadb.errors import InvalidCollectionException
 
 from vectorcode.cli_utils import Config
 from vectorcode.common import get_client, get_collection
 
+logger = logging.getLogger(name=__name__)
+
 
 async def drop(config: Config) -> int:
     client = await get_client(config)
@@ -11,7 +15,8 @@ async def drop(config: Config) -> int:
         collection_path = collection.metadata["path"]
         await client.delete_collection(collection.name)
         print(f"Collection for {collection_path} has been deleted.")
+        logger.info(f"Deteted collection at {collection_path}.")
         return 0
     except (ValueError, InvalidCollectionException):
-        print(f"There's no existing collection for {config.project_root}")
+        logger.error(f"There's no existing collection for {config.project_root}")
         return 1

src/vectorcode/subcommands/init.py

@@ -1,16 +1,17 @@
+import logging
 import os
 import shutil
-import sys
 
 from vectorcode.cli_utils import Config
 
+logger = logging.getLogger(name=__name__)
+
 
 async def init(configs: Config) -> int:
     project_config_dir = os.path.join(str(configs.project_root), ".vectorcode")
     if os.path.isdir(project_config_dir) and not configs.force:
-        print(
+        logger.warning(
             f"{configs.project_root} is already initialised for VectorCode.",
-            file=sys.stderr,
         )
         return 1
 
@@ -21,6 +22,7 @@ async def init(configs: Config) -> int:
             os.path.expanduser("~"), ".config", "vectorcode", item
         )
         if os.path.isfile(global_file_path):
+            logger.debug(f"Copying global {item} to {project_config_dir}")
             shutil.copyfile(global_file_path, local_file_path)
 
     print(f"VectorCode project root has been initialised at {configs.project_root}")

tests/subcommands/test_init.py

@@ -32,8 +32,6 @@ async def test_init_already_initialized(capsys):
         # Try to initialize again without force
         return_code = await init(configs)
         assert return_code == 1
-        captured = capsys.readouterr()
-        assert f"{temp_dir} is already initialised for VectorCode." in captured.err
 
 
 @pytest.mark.asyncio