124 lines
4.2 KiB
Python
Executable File
124 lines
4.2 KiB
Python
Executable File
#!/usr/bin/python3
|
|
|
|
# CREATE MUSIC DATABASE ------------------------------------------
|
|
|
|
import sys, os, datetime, fnmatch, glob, random, time, pathlib, re
|
|
from datetime import timedelta
|
|
from os.path import join
|
|
from tinytag import TinyTag
|
|
from random import shuffle
|
|
import sqlite3, json
|
|
import uuid
|
|
#import pypika # sql query builder
|
|
from pypika import Query, Table, Field, Column
|
|
|
|
conn = ''
|
|
|
|
music_library_path = "/home/rob/antena/music/"
|
|
conn = sqlite3.connect("database/show.db")
|
|
|
|
#TODO fix getting bandcamp urls into db
|
|
labelnames = [("sploh", "https://sploh.bandcamp.com"), ("terraformer"), ("pharamafabric") ]
|
|
|
|
# todo find album with no album name
|
|
mus_lib = Query \
|
|
.create_table("MUSIC_LIBRARY") \
|
|
.columns(
|
|
Column("id", "INT", nullable=True),
|
|
Column("label", "VARCHAR(100)", nullable=True),
|
|
Column("album", "VARCHAR(100)", nullable=True),
|
|
Column("track", "VARCHAR(200)", nullable=True),
|
|
Column("artist", "VARCHAR(120)", nullable=True),
|
|
Column("genre", "VARCHAR(120)", nullable=True),
|
|
Column("trackdur", "FLOAT", nullable=True),
|
|
Column("year", "INT", nullable=True),
|
|
Column("url", "VARCHAR(120)", nullable=True),
|
|
Column("path", "VARCHAR(120)", nullable=False),
|
|
Column("lastplay", "DATETINE", nullable=True),
|
|
Column("comment", "VARCHAR(120)", nullable=True))\
|
|
.unique("path") \
|
|
.primary_key("path")
|
|
|
|
def database_create(conn):
|
|
# the MUSIC LIBRARY TABLE
|
|
#TODO UNIQUE constraint prevents duplicats but what if tracks are in another locaton?fixme
|
|
conn.execute(str(mus_lib))
|
|
print('''MUSIC LIBRARY Table created successfully''');
|
|
|
|
|
|
#conn = ''
|
|
|
|
def database_create_episodes_table(conn):
|
|
# the show database
|
|
|
|
# conn = sqlite3.connect("database/show.db")
|
|
|
|
#TODO put the below back as a raw quiry to make if not exists possible
|
|
q = Query \
|
|
.create_table("EPISODES") \
|
|
.columns(
|
|
Column("id", "INT", nullable=True),
|
|
Column("episode", "INT", nullable=True),
|
|
Column("date", 'DATETIME', nullable=True),
|
|
Column("album", "VARCHAR(200)", nullable=True),
|
|
Column("track", "VARCHAR(120)", nullable=True),
|
|
Column("artist", "VARCHAR(120)", nullable=True),
|
|
Column("trackdur", "FLOAT", nullable=True),
|
|
Column("genre", "VARCHAR(120)", nullable=True),
|
|
Column("year", 'DATETIME', nullable=True),
|
|
Column("path", "VARCHAR(120)", nullable=False))\
|
|
.unique("path") \
|
|
.primary_key("path")
|
|
|
|
#TODO get the unique path back into action find bug
|
|
conn.execute(str(q))
|
|
|
|
|
|
|
|
|
|
print("EPISODES Table created successfully");
|
|
|
|
def mk_db_entry(conn):
|
|
print("ADDING TRACKS TO DATABASE > > > one moment please! ");
|
|
label_url = "https://fixme.bandcamp.com/"
|
|
|
|
for subdir, dirs, files in os.walk(music_library_path):
|
|
|
|
for file in files:
|
|
for m in [".flac", ".FLAC", ".mp3"]: # get audio files
|
|
if m in file:
|
|
filepath = pathlib.Path(subdir + "/" +file)
|
|
label = filepath.parts[5] # get LABEL etc name from path
|
|
track = TinyTag.get(os.path.join(subdir, file))# get metadata from file
|
|
cursor = conn.cursor()
|
|
|
|
mus_lib = Table('MUSIC_LIBRARY')
|
|
q = mus_lib.insert(1, label, track.album, track.title, track.artist, \
|
|
track.genre, track.duration, track.year, \
|
|
label_url, str(filepath), "last_played", track.comment)
|
|
cursor.execute(str(q));
|
|
conn.commit()
|
|
|
|
def count_tracks(conn):
|
|
query = f"SELECT COUNT(*) FROM MUSIC_LIBRARY"
|
|
cursor = conn.cursor()
|
|
cursor.execute(query)
|
|
result = cursor.fetchone()
|
|
row_count = result[0]
|
|
print(f'''\n WHOOP! THE MUSIC LIBRARY CONTAINS {row_count} TRACKS!!
|
|
--------------------------------------------''')
|
|
|
|
def main():
|
|
|
|
database_create(conn)
|
|
database_create_episodes_table(conn)
|
|
mk_db_entry(conn)
|
|
count_tracks(conn)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|
|
|
|
|
|
|
|
|