This commit is contained in:
mike
2025-12-12 23:04:51 +01:00
parent 56b2db82fc
commit 87550e426a
14 changed files with 132 additions and 122 deletions

View File

@@ -60,7 +60,7 @@ class DiscoveryEngine:
size BIGINT NOT NULL,
modified_time DOUBLE PRECISION NOT NULL,
created_time DOUBLE PRECISION NOT NULL,
disk TEXT NOT NULL,
disk_label TEXT NOT NULL,
checksum TEXT,
status TEXT DEFAULT 'indexed',
category TEXT,
@@ -72,18 +72,18 @@ class DiscoveryEngine:
# Create index on path
cursor.execute("""
CREATE INDEX IF NOT EXISTS idx_files_path ON files_bak(path)
""")
CREATE INDEX IF NOT EXISTS idx_files_path ON files(path)
""")
# Create index on disk
cursor.execute("""
CREATE INDEX IF NOT EXISTS idx_files_disk ON files_bak(disk)
""")
CREATE INDEX IF NOT EXISTS idx_files_disk ON files(disk_label)
""")
# Create index on checksum
cursor.execute("""
CREATE INDEX IF NOT EXISTS idx_files_checksum ON files_bak(checksum)
""")
CREATE INDEX IF NOT EXISTS idx_files_checksum ON files(checksum)
""")
conn.commit()
cursor.close()
@@ -136,7 +136,7 @@ class DiscoveryEngine:
size=file_meta.size,
modified_time=file_meta.modified_time,
created_time=file_meta.created_time,
disk=disk
disk_label=disk
)
batch.append(record)
@@ -193,7 +193,7 @@ class DiscoveryEngine:
batch: List of FileRecord objects
"""
query = """
INSERT INTO files_bak (path, size, modified_time, created_time, disk, checksum, status, category, duplicate_of)
INSERT INTO files (path, size, modified_time, created_time, disk_label, checksum, status, category, duplicate_of)
VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s)
ON CONFLICT (path) DO UPDATE SET
size = EXCLUDED.size,
@@ -207,7 +207,7 @@ class DiscoveryEngine:
record.size,
record.modified_time,
record.created_time,
record.disk,
record.disk_label,
record.checksum,
record.status,
record.category,
@@ -276,9 +276,9 @@ class DiscoveryEngine:
cursor = conn.cursor()
if disk:
cursor.execute("SELECT COUNT(*) FROM files_bak WHERE disk = %s", (disk,))
cursor.execute("SELECT COUNT(*) FROM files WHERE disk_label = %s", (disk,))
else:
cursor.execute("SELECT COUNT(*) FROM files_bak")
cursor.execute("SELECT COUNT(*) FROM files")
count = cursor.fetchone()[0]
cursor.close()
@@ -298,9 +298,9 @@ class DiscoveryEngine:
cursor = conn.cursor()
if disk:
cursor.execute("SELECT COALESCE(SUM(size), 0) FROM files_bak WHERE disk = %s", (disk,))
cursor.execute("SELECT COALESCE(SUM(size), 0) FROM files WHERE disk_label = %s", (disk,))
else:
cursor.execute("SELECT COALESCE(SUM(size), 0) FROM files_bak")
cursor.execute("SELECT COALESCE(SUM(size), 0) FROM files")
total = cursor.fetchone()[0]
cursor.close()