simon987/music-graph-scripts
1
0
Fork 0
mirror of https://github.com/simon987/music-graph-scripts.git synced 2025-12-13 22:59:03 +00:00
Code Issues Projects Releases Wiki Activity

Sort name for autocomplete

Browse Source
This commit is contained in:
simon
2019-06-08 11:51:00 -04:00
parent 6460e98ce0
commit a70b9ffa61
4 changed files with 17 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
process_mb_dump.py
View File

@@ -1,5 +1,6 @@
import os
from collections import defaultdict
import re
links = dict()
link_types = dict()
@@ -188,14 +189,24 @@ with open("repo/area.csv", "w") as out:
out_artist = open("repo/artist.csv", "w")
out_artist_area = open("repo/artist_area.csv", "w")
out_artist.write("id:ID(Artist),name,year:int,:LABEL\n")
out_artist.write("id:ID(Artist),name,sortname,year:int,comment,:LABEL\n")
out_artist_area.write(":START_ID(Artist),:END_ID(Area)\n")
ASCII_RE = re.compile(r"[^a-zA-Z0-9.\-!?& ]")
ALPHANUM_RE = re.compile(r"[^\w.\-!?& ]")
for _, artist in artists.items():
sortname = ASCII_RE.sub("_", artist[2]).upper()
if sortname.replace("_", "").strip() == "":
sortname = ALPHANUM_RE.sub("_", artist[3]).upper()
out_artist.write(",".join((
artist[1],
'"' + artist[2].replace("\"", "\"\"") + '"',
sortname,
artist[4] if artist[4] != "\\N" else "0",
('"' + artist[13].replace("\"", "\"\"") + '"') if artist[13] != "\\N" else "",
"Artist" + (";Group\n" if artist[10] == "2" else "\n")
)))