#!/usr/bin/python3 from disambiguate import disambiguate import sys import json import plac from ftfy import fix_encoding @plac.annotations( nameDepth = ("Maximum number of people to check" ,"option", "p", int ), worksDepth = ("Maximum number of works we take for a person" ,"option", "q", int ), ) def main(nameDepth = 20, worksDepth = 20): for line in sys.stdin: line = fix_encoding(line) data = json.loads(line) infos = data['value'] info = infos[0].copy() db = disambiguate(info,nameDepth=nameDepth, worksDepth=worksDepth ) result = db.disambiguation() if len(result)>0: data['value'] = result[0][0] else: data['value'] = "None" sys.stdout.write(json.dumps(data)) sys.stdout.write('\n') if __name__ == "__main__": plac.call(main)