92 lines
2.6 KiB
Python
Executable File
92 lines
2.6 KiB
Python
Executable File
# #!/usr/bin/env python
|
|
#
|
|
import os
|
|
|
|
from config import Config
|
|
from helpers import (
|
|
get_tags,
|
|
)
|
|
from log import log
|
|
from models import Tracks
|
|
|
|
|
|
def check_db(session):
|
|
"""
|
|
Database consistency check.
|
|
|
|
A report is generated if issues are found, but there are no automatic
|
|
corrections made.
|
|
|
|
Search for tracks that are in the music directory but not the datebase
|
|
Check all paths in database exist
|
|
"""
|
|
|
|
db_paths = set([a.path for a in Tracks.get_all(session)])
|
|
|
|
os_paths_list = []
|
|
for root, _dirs, files in os.walk(Config.ROOT):
|
|
for f in files:
|
|
path = os.path.join(root, f)
|
|
ext = os.path.splitext(f)[1]
|
|
if ext in [".flac", ".mp3"]:
|
|
os_paths_list.append(path)
|
|
os_paths = set(os_paths_list)
|
|
|
|
# Find any files in music directory that are not in database
|
|
files_not_in_db = list(os_paths - db_paths)
|
|
|
|
# Find paths in database missing in music directory
|
|
paths_not_found = []
|
|
missing_file_count = 0
|
|
more_files_to_report = False
|
|
for path in list(db_paths - os_paths):
|
|
if missing_file_count >= Config.MAX_MISSING_FILES_TO_REPORT:
|
|
more_files_to_report = True
|
|
break
|
|
|
|
missing_file_count += 1
|
|
|
|
track = Tracks.get_by_path(session, path)
|
|
if not track:
|
|
# This shouldn't happen as we're looking for paths in
|
|
# database that aren't in filesystem, but just in case...
|
|
log.error(f"update_db: {path} not found in db")
|
|
continue
|
|
|
|
paths_not_found.append(track)
|
|
|
|
# Output messages (so if running via cron, these will get sent to
|
|
# user)
|
|
if files_not_in_db:
|
|
print("Files in music directory but not in database")
|
|
print("--------------------------------------------")
|
|
print("\n".join(files_not_in_db))
|
|
print("\n")
|
|
if paths_not_found:
|
|
print("Invalid paths in database")
|
|
print("-------------------------")
|
|
for t in paths_not_found:
|
|
print(
|
|
f"""
|
|
Track ID: {t.id}
|
|
Path: {t.path}
|
|
Title: {t.title}
|
|
Artist: {t.artist}
|
|
"""
|
|
)
|
|
if more_files_to_report:
|
|
print("There were more paths than listed that were not found")
|
|
|
|
|
|
def update_bitrates(session):
|
|
"""
|
|
Update bitrates on all tracks in database
|
|
"""
|
|
|
|
for track in Tracks.get_all(session):
|
|
try:
|
|
t = get_tags(track.path)
|
|
track.bitrate = t["bitrate"]
|
|
except FileNotFoundError:
|
|
continue
|