refactor: Improve SQLite cache error handling and import organization

This commit is contained in:
Paul Gauthier 2024-10-29 15:04:00 -07:00 committed by Paul Gauthier (aider)
parent 0feed0047c
commit ef0fcb8f38

View file

@ -2,6 +2,7 @@ import colorsys
import math import math
import os import os
import random import random
import shutil
import sqlite3 import sqlite3
import sys import sys
import time import time
@ -168,7 +169,7 @@ class RepoMap:
def tags_cache_error(self): def tags_cache_error(self):
"""Handle SQLite errors by trying to recreate cache, falling back to dict if needed""" """Handle SQLite errors by trying to recreate cache, falling back to dict if needed"""
if not isinstance(self.TAGS_CACHE, Cache): if isinstance(getattr(self, "TAGS_CACHE", None), dict):
return return
path = Path(self.root) / self.TAGS_CACHE_DIR path = Path(self.root) / self.TAGS_CACHE_DIR
@ -177,8 +178,6 @@ class RepoMap:
try: try:
# Delete existing cache dir # Delete existing cache dir
if path.exists(): if path.exists():
import shutil
shutil.rmtree(path) shutil.rmtree(path)
# Try to create new cache # Try to create new cache