328 lines
11 KiB
Markdown
328 lines
11 KiB
Markdown
# CLAUDE.md
|
|
|
|
This file provides guidance to Claude Code (claude.ai/code) when working with code in this repository.
|
|
|
|
## Available Personas
|
|
|
|
This project uses specialized personas for different types of work. Use these commands to switch modes:
|
|
|
|
- **`/developer`** - Full development mode with validation workflow (options-first, wait for approval before coding)
|
|
- **`/unit-tester`** - Specialized mode for writing comprehensive unit tests for existing code
|
|
- **`/technical-writer`** - User documentation writing mode (README, guides, tutorials)
|
|
- **`/reset`** - Return to default Claude Code mode
|
|
|
|
Each persona has specific rules and workflows defined in `.claude/` directory. See the respective files for detailed guidelines.
|
|
|
|
## Project Overview
|
|
|
|
MyDbEngine is a lightweight, git-inspired versioned database engine for Python. It maintains complete history of all data modifications using immutable snapshots with SHA-256 content addressing. The project supports multi-tenant storage with thread-safe operations.
|
|
|
|
### Quick Start Example
|
|
|
|
```python
|
|
from dbengine.dbengine import DbEngine
|
|
|
|
# Initialize engine
|
|
engine = DbEngine(root=".mytools_db")
|
|
engine.init("tenant_1")
|
|
|
|
# Pattern 1: Snapshot-based (complete state saves)
|
|
engine.save("tenant_1", "user_1", "config", {"theme": "dark", "lang": "en"})
|
|
data = engine.load("tenant_1", "config")
|
|
|
|
# Pattern 2: Record-based (incremental updates)
|
|
engine.put("tenant_1", "user_1", "users", "john", {"name": "John", "age": 30})
|
|
engine.put("tenant_1", "user_1", "users", "jane", {"name": "Jane", "age": 25})
|
|
all_users = engine.get("tenant_1", "users") # Returns list of all users
|
|
```
|
|
|
|
## Development Commands
|
|
|
|
### Testing
|
|
```bash
|
|
# Run all tests
|
|
pytest
|
|
|
|
# Run specific test file
|
|
pytest tests/test_dbengine.py
|
|
pytest tests/test_serializer.py
|
|
|
|
# Run single test function
|
|
pytest tests/test_dbengine.py::test_i_can_save_and_load
|
|
```
|
|
|
|
### Building and Packaging
|
|
```bash
|
|
# Build package
|
|
python -m build
|
|
|
|
# Clean build artifacts
|
|
make clean
|
|
|
|
# Clean package artifacts only
|
|
make clean-package
|
|
```
|
|
|
|
### Installation
|
|
```bash
|
|
# Install in development mode with test dependencies
|
|
pip install -e .[dev]
|
|
```
|
|
|
|
## Architecture
|
|
|
|
### Core Components
|
|
|
|
**DbEngine** (`src/dbengine/dbengine.py`)
|
|
- Main database engine class using RLock for thread safety
|
|
- Manages tenant-specific storage in `.mytools_db/{tenant_id}/` structure
|
|
- Tracks latest versions via `head` file (JSON mapping entry names to digests)
|
|
- Stores objects in content-addressable format: `objects/{digest_prefix}/{full_digest}`
|
|
- Shared `refs/` directory for cross-tenant pickle-based references
|
|
|
|
**Serializer** (`src/dbengine/serializer.py`)
|
|
- Converts Python objects to/from JSON-compatible dictionaries
|
|
- Handles circular references using object ID tracking
|
|
- Supports custom serialization via handlers (see handlers.py)
|
|
- Special tags: `__object__`, `__id__`, `__tuple__`, `__set__`, `__ref__`, `__enum__`
|
|
- Objects can define `use_refs()` method to specify fields that should be pickled instead of JSON-serialized
|
|
|
|
**Handlers** (`src/dbengine/handlers.py`)
|
|
- Extensible handler system for custom type serialization
|
|
- BaseHandler interface: `is_eligible_for()`, `tag()`, `serialize()`, `deserialize()`
|
|
- Currently implements DateHandler for datetime.date objects
|
|
- Use `handlers.register_handler()` to add custom handlers
|
|
|
|
**Utils** (`src/dbengine/utils.py`)
|
|
- Type checking utilities: `is_primitive()`, `is_dictionary()`, `is_list()`, etc.
|
|
- Class introspection: `get_full_qualified_name()`, `importable_name()`, `get_class()`
|
|
- Stream digest computation with SHA-256
|
|
|
|
### Storage Architecture
|
|
|
|
```
|
|
.mytools_db/
|
|
├── {tenant_id}/
|
|
│ ├── head # JSON: {"entry_name": "latest_digest"}
|
|
│ └── objects/
|
|
│ └── {digest_prefix}/ # First 24 chars of digest
|
|
│ └── {full_digest} # JSON snapshot with metadata
|
|
└── refs/ # Shared pickled references
|
|
└── {digest_prefix}/
|
|
└── {full_digest}
|
|
```
|
|
|
|
### Metadata System
|
|
|
|
Each snapshot includes automatic metadata fields:
|
|
- `__parent__`: List containing digest of previous version (or `[None]` for first)
|
|
- `__user_id__`: User ID who created the snapshot (was `__user__` in TAG constant)
|
|
- `__date__`: ISO timestamp `YYYYMMDD HH:MM:SS %z`
|
|
|
|
### Two Usage Patterns
|
|
|
|
**Pattern 1: Snapshot-based (`save()`/`load()`)**
|
|
- Save complete object states
|
|
- Best for configuration objects or complete state snapshots
|
|
- Direct control over what gets saved
|
|
|
|
**Pattern 2: Record-based (`put()`/`put_many()`/`get()`)**
|
|
- Incremental updates to dictionary-like collections
|
|
- Automatically creates snapshots only when data changes
|
|
- Returns `True/False` indicating if snapshot was created
|
|
- Best for managing collections of items
|
|
|
|
**Important**: Do not mix patterns for the same entry - they expect different data structures.
|
|
|
|
### Common Pitfalls
|
|
|
|
⚠️ **Mixing save() and put() on the same entry**
|
|
- `save()` expects to store complete snapshots (any object)
|
|
- `put()` expects dictionary-like structures with key-value pairs
|
|
- Using both on the same entry will cause data structure conflicts
|
|
|
|
⚠️ **Refs are shared across tenants**
|
|
- Objects stored via `use_refs()` go to shared `refs/` directory
|
|
- Not isolated per tenant - identical objects reused across all tenants
|
|
- Good for deduplication, but be aware of cross-tenant sharing
|
|
|
|
⚠️ **Parent digest is always a list**
|
|
- `__parent__` field is stored as `[digest]` or `[None]`
|
|
- Always access as `data[TAG_PARENT][0]`, not `data[TAG_PARENT]`
|
|
- This allows for future support of multiple parents (merge scenarios)
|
|
|
|
### Reference System
|
|
|
|
Objects can opt into pickle-based storage for specific fields:
|
|
1. Define `use_refs()` method returning set of field names
|
|
2. Serializer stores those fields in shared `refs/` directory
|
|
3. Reduces JSON snapshot size and enables cross-tenant deduplication
|
|
4. Example: `DummyObjWithRef` in test_dbengine.py
|
|
|
|
## Extension Points
|
|
|
|
### Custom Type Handlers
|
|
|
|
To serialize custom types that aren't handled by default serialization:
|
|
|
|
**1. Create a handler class:**
|
|
```python
|
|
from dbengine.handlers import BaseHandler, TAG_SPECIAL
|
|
|
|
class MyCustomHandler(BaseHandler):
|
|
def is_eligible_for(self, obj):
|
|
return isinstance(obj, MyCustomType)
|
|
|
|
def tag(self):
|
|
return "MyCustomType"
|
|
|
|
def serialize(self, obj) -> dict:
|
|
return {
|
|
TAG_SPECIAL: self.tag(),
|
|
"data": obj.to_dict()
|
|
}
|
|
|
|
def deserialize(self, data: dict) -> object:
|
|
return MyCustomType.from_dict(data["data"])
|
|
```
|
|
|
|
**2. Register the handler:**
|
|
```python
|
|
from dbengine.handlers import handlers
|
|
|
|
handlers.register_handler(MyCustomHandler())
|
|
```
|
|
|
|
**When to use handlers:**
|
|
- Complex types that need custom serialization logic
|
|
- Types that can't be pickled reliably
|
|
- Types requiring validation during deserialization
|
|
- External library types (datetime.date example in handlers.py)
|
|
|
|
### Using References (use_refs)
|
|
|
|
For objects with large nested data structures that should be pickled instead of JSON-serialized:
|
|
|
|
```python
|
|
class MyDataObject:
|
|
def __init__(self, metadata, large_dataframe):
|
|
self.metadata = metadata
|
|
self.large_dataframe = large_dataframe # pandas DataFrame, for example
|
|
|
|
@staticmethod
|
|
def use_refs():
|
|
"""Return set of field names to pickle instead of JSON-serialize"""
|
|
return {"large_dataframe"}
|
|
```
|
|
|
|
**When to use refs:**
|
|
- Large data structures (DataFrames, numpy arrays)
|
|
- Objects that lose information in JSON conversion
|
|
- Data shared across multiple snapshots/tenants (deduplication benefit)
|
|
|
|
**Trade-offs:**
|
|
- ✅ Smaller JSON snapshots
|
|
- ✅ Cross-tenant deduplication
|
|
- ❌ Less human-readable (binary pickle format)
|
|
- ❌ Python version compatibility concerns with pickle
|
|
|
|
## Testing Notes
|
|
|
|
- Test fixtures use `DB_ENGINE_ROOT = "TestDBEngineRoot"` for isolation
|
|
- Tests clean up temp directories using `shutil.rmtree()` in fixtures
|
|
- Test classes like `DummyObj`, `DummyObjWithRef`, `DummyObjWithKey` demonstrate usage patterns
|
|
- Thread safety is built-in via RLock but not explicitly tested
|
|
|
|
## Key Design Decisions
|
|
|
|
- **Immutability**: Snapshots never modified after creation (git-style)
|
|
- **Content Addressing**: Identical objects stored only once (deduplication via SHA-256)
|
|
- **Change Detection**: `put()` and `put_many()` skip saving if data unchanged
|
|
- **Thread Safety**: All DbEngine operations protected by RLock
|
|
- **No Dependencies**: Core engine has zero runtime dependencies (pytest only for dev)
|
|
|
|
## Development Workflow and Guidelines
|
|
|
|
### Development Process
|
|
|
|
**Code must always be testable**. Before writing any code:
|
|
|
|
1. **Explain available options first** - Present different approaches to solve the problem
|
|
2. **Wait for validation** - Ensure mutual understanding of requirements before implementation
|
|
3. **No code without approval** - Only proceed after explicit validation
|
|
|
|
### Collaboration Style
|
|
|
|
**Ask questions to clarify understanding or suggest alternative approaches:**
|
|
- Ask questions **one at a time**
|
|
- Wait for complete answer before asking the next question
|
|
- Indicate progress: "Question 1/5" if multiple questions are needed
|
|
- Never assume - always clarify ambiguities
|
|
|
|
### Communication
|
|
|
|
**Conversations**: French or English
|
|
**Code, documentation, comments**: English only
|
|
|
|
### Code Standards
|
|
|
|
**Follow PEP 8** conventions strictly:
|
|
- Variable and function names: `snake_case`
|
|
- Explicit, descriptive naming
|
|
- **No emojis in code**
|
|
|
|
**Documentation**:
|
|
- Use Google or NumPy docstring format
|
|
- Document all public functions and classes
|
|
- Include type hints where applicable
|
|
|
|
### Dependency Management
|
|
|
|
**When introducing new dependencies:**
|
|
- List all external dependencies explicitly
|
|
- Propose alternatives using Python standard library when possible
|
|
- Explain why each dependency is needed
|
|
|
|
### Unit Testing with pytest
|
|
|
|
**Test naming patterns:**
|
|
- Passing tests: `test_i_can_xxx` - Tests that should succeed
|
|
- Failing tests: `test_i_cannot_xxx` - Edge cases that should raise errors/exceptions
|
|
|
|
**Test structure:**
|
|
- Use **functions**, not classes (unless inheritance is required)
|
|
- Before writing tests, **list all planned tests with explanations**
|
|
- Wait for validation before implementing tests
|
|
|
|
**Example:**
|
|
```python
|
|
def test_i_can_save_and_load_object():
|
|
"""Test that an object can be saved and loaded successfully."""
|
|
engine = DbEngine(root="test_db")
|
|
engine.init("tenant_1")
|
|
digest = engine.save("tenant_1", "user_1", "entry_1", {"key": "value"})
|
|
assert digest is not None
|
|
|
|
def test_i_cannot_save_with_empty_tenant_id():
|
|
"""Test that saving with empty tenant_id raises DbException."""
|
|
engine = DbEngine(root="test_db")
|
|
with pytest.raises(DbException):
|
|
engine.save("", "user_1", "entry_1", {"key": "value"})
|
|
```
|
|
|
|
### File Management
|
|
|
|
**Always specify the full file path** when adding or modifying files:
|
|
```
|
|
✅ Modifying: src/dbengine/dbengine.py
|
|
✅ Creating: tests/test_new_feature.py
|
|
```
|
|
|
|
### Error Handling
|
|
|
|
**When errors occur:**
|
|
1. **Explain the problem clearly first**
|
|
2. **Do not propose a fix immediately**
|
|
3. **Wait for validation** that the diagnosis is correct
|
|
4. Only then propose solutions
|