def init_db(db_file):
"""Small helper to (re)initalize the database."""
create_tree(db_file)
# Trick to truncate the file.
open(db_file, "w").close()
conn = sqlite3.connect(db_file)
# Create the table and index.
conn.executescript(TABLE_SCHEMA)
return conn
def init_db(db_file):
"""Small helper to (re)initalize the database."""
create_tree(db_file)
# Trick to truncate the file.
open(db_file, "w").close()
conn = sqlite3.connect(db_file);
# Create the table and index.
conn.executescript(TABLE_SCHEMA)
return conn
def _parse_command():
parser = argparse.ArgumentParser(description="""Generate a
tag-track matrix from the MSD tags database.""")
# Options.
parser.add_argument('--max-tracks', '-m', type=int, default=-1)
parser.add_argument('--max-tags', '-M', type=int, default=-1)
parser.add_argument('--min-tag-count', '-c', type=int, default=0)
parser.add_argument('--out', '-o', default=MATRIX_PATH)
parser.add_argument('--tags-out', default=TAGS_PATH)
parser.add_argument('--weights-out', default=WEIGHTS_PATH)
# Required arguments.
parser.add_argument('database')
parser.add_argument('tags')
return parser.parse_args()
if __name__ == '__main__':
args = _parse_command()
print "Generating the [tag_name -> column] mapping.."
tags = generate_tags_dict(args.tags, max_nb=args.max_tags,
min_count=args.min_tag_count)
print "Generating the matrix..."
create_tree(args.out)
weights = generate_matrix(tags, args.database, args.out,
max_nb=args.max_tracks)
print "Writing the [tag_name -> column] mapping..."
dump_tags(tags, args.tags_out)
print "Writing the [column -> global_weight] mapping..."
dump_weights(weights, args.weights_out)
print "Done."
