maxenceg2m/iTunes
1
0
Fork 0
Code Issues 2 Pull Requests Projects Releases Wiki Activity

Working suggester ingester

Browse Source
This commit is contained in:
Maxence G. de Montauzan
2021-08-30 19:13:03 +02:00
parent 042c2558ae
commit 67e1f8bd0c
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
suggester.py
View File

@@ -80,8 +80,7 @@ def process_file(file_name: str, field_name: str, array_file: str = None) -> int
if array_file and array_file in data and data[array_file]: if array_file and array_file in data and data[array_file]:
for key in data[array_file]: for key in data[array_file]:
if key != data[field_name]: # => Absolutely don't work for album, and block a scoring for artists input.extend(get_tokens(key))
input.extend(get_tokens(key))
# TODO Input have the same value several times ==> use to process a score # TODO Input have the same value several times ==> use to process a score
post_document(name=data[field_name], input=input, field_name=field_name.lower()) post_document(name=data[field_name], input=input, field_name=field_name.lower())
@@ -97,7 +96,7 @@ def process_file(file_name: str, field_name: str, array_file: str = None) -> int
if __name__ == '__main__': if __name__ == '__main__':
# Using readlines() # Using readlines()
count = 0 count = 0
count += process_file('/home/budd/workspace/iTunes/es-albums.json', 'Album', 'Artist') count += process_file('/home/budd/workspace/iTunes/es-albums.json', 'Album')
print('Created documents: ' + str(count)) print('Created documents: ' + str(count))
count += process_file('/home/budd/workspace/iTunes/es-artists.json', 'Artist', 'Album Artist') count += process_file('/home/budd/workspace/iTunes/es-artists.json', 'Artist', 'Album Artist')
print('Created documents: ' + str(count)) print('Created documents: ' + str(count))