base
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user