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."