70 lines
2.5 KiB
Python
70 lines
2.5 KiB
Python
import DBA
|
|
from configparser import ConfigParser
|
|
from utils import get_id3_tags
|
|
from utils import safe_get
|
|
|
|
config = ConfigParser()
|
|
config.read("config.ini")
|
|
|
|
|
|
def add_files_to_library(files):
|
|
"""Adds audio file(s) to the sqllite db
|
|
`files` | list() | List of fully qualified paths to audio file(s)
|
|
Returns a count of records added
|
|
"""
|
|
print(f"utils | adding files to library: {files}")
|
|
extensions = config.get("settings", "extensions").split(",")
|
|
insert_data = [] # To store data for batch insert
|
|
for filepath in files:
|
|
if any(filepath.lower().endswith(ext) for ext in extensions):
|
|
filename = filepath.split("/")[-1]
|
|
audio = get_id3_tags(filepath)
|
|
if "title" not in audio:
|
|
return
|
|
# Append data tuple to insert_data list
|
|
insert_data.append(
|
|
(
|
|
filepath,
|
|
safe_get(audio, "title", [])[0],
|
|
safe_get(audio, "album", [])[0] if "album" in audio else None,
|
|
safe_get(audio, "artist", [])[0] if "artist" in audio else None,
|
|
",".join(safe_get(audio, "genre", []))
|
|
if "genre" in audio
|
|
else None,
|
|
filename.split(".")[-1],
|
|
safe_get(audio, "date", [])[0] if "date" in audio else None,
|
|
safe_get(audio, "bitrate", [])[0] if "birate" in audio else None,
|
|
)
|
|
)
|
|
|
|
# Check if batch size is reached
|
|
if len(insert_data) >= 1000:
|
|
with DBA.DBAccess() as db:
|
|
db.executemany(
|
|
"INSERT OR IGNORE INTO library (filepath, title, album, artist, genre, codec, album_date, bitrate) VALUES (?, ?, ?, ?, ?, ?, ?, ?)",
|
|
insert_data,
|
|
)
|
|
insert_data = [] # Reset the insert_data list
|
|
|
|
# Insert any remaining data
|
|
if insert_data:
|
|
with DBA.DBAccess() as db:
|
|
db.executemany(
|
|
"INSERT OR IGNORE INTO library (filepath, title, album, artist, genre, codec, album_date, bitrate) VALUES (?, ?, ?, ?, ?, ?, ?, ?)",
|
|
insert_data,
|
|
)
|
|
return True
|
|
|
|
|
|
# id int unsigned auto_increment,
|
|
# title varchar(255),
|
|
# album varchar(255),
|
|
# artist varchar(255),
|
|
# genre varchar(255),
|
|
# codec varchar(15),
|
|
# album_date date,
|
|
# bitrate int unsigned,
|
|
# date_added TIMESTAMP default CURRENT_TIMESTAMP,
|
|
|
|
# scan_for_music(config.get('directories', 'library1'))
|