Philip Guzman
54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
from pathlib import Path
|
|
import argparse
|
|
|
|
from serato_doctor.config import ScanConfig
|
|
from serato_doctor.crate_parser import parse_crates
|
|
from serato_doctor.models.library import Library
|
|
from serato_doctor.scanner import scan_audio
|
|
from serato_doctor.report import write_csv, write_missing_report
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(prog="serato-doctor")
|
|
parser.add_argument("--serato", default=str(Path.home() / "Music/_Serato_"))
|
|
parser.add_argument("--music", default=str(Path.home() / "Library/CloudStorage/OneDrive-Personal/Jukebox"))
|
|
parser.add_argument("--out", default=str(Path.home() / "Desktop/serato_doctor_scan.csv"))
|
|
parser.add_argument("--report", default=str(Path.home() / "Desktop/serato_doctor_missing_report.txt"))
|
|
parser.add_argument(
|
|
"--reference-root",
|
|
action="append",
|
|
default=[],
|
|
help="Old library root stored in crates; may be supplied more than once",
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
config = ScanConfig.build(
|
|
serato=Path(args.serato),
|
|
music=Path(args.music),
|
|
out=Path(args.out),
|
|
report=Path(args.report),
|
|
reference_roots=(Path(root) for root in args.reference_root),
|
|
)
|
|
|
|
library = Library.build(
|
|
references=parse_crates(
|
|
config.serato / "Subcrates", config.reference_roots
|
|
),
|
|
tracks=scan_audio(config.music),
|
|
)
|
|
results = library.reconcile_by_filename()
|
|
missing_count = sum(1 for result in results if not result.exists_by_filename)
|
|
|
|
write_csv(results, config.out)
|
|
write_missing_report(results, config.report)
|
|
|
|
print(f"Crate references: {len(library.references)}")
|
|
print(f"Disk tracks: {len(library.tracks)}")
|
|
print(f"Missing by filename: {missing_count}")
|
|
print(f"CSV: {config.out}")
|
|
print(f"Report: {config.report}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|