Load all SQL fragments into memory
(self)
| 25 | self._load_fragments() |
| 26 | |
| 27 | def _load_fragments(self) -> None: |
| 28 | """Load all SQL fragments into memory""" |
| 29 | if not self.fragment_dir.exists(): |
| 30 | if self.logger: |
| 31 | self.logger.warning(f"Fragment directory not found: {self.fragment_dir}") |
| 32 | return |
| 33 | |
| 34 | for fragment_file in self.fragment_dir.glob("*.sql"): |
| 35 | try: |
| 36 | fragment_name = fragment_file.stem |
| 37 | fragment_content = fragment_file.read_text() |
| 38 | self._fragments_cache[fragment_name] = fragment_content |
| 39 | |
| 40 | if self.logger: |
| 41 | self.logger.debug(f"Loaded fragment: {fragment_name}") |
| 42 | except Exception as e: |
| 43 | if self.logger: |
| 44 | self.logger.error(f"Error loading fragment {fragment_file}: {e}") |
| 45 | |
| 46 | def get_fragment(self, name: str) -> str: |
| 47 | """Get a SQL fragment by name |