def main(): args = parse_args() # Analyze data. nicknames, relations = analyze(args.filenames, args.directed, args.no_unrelated_nicknames) # Show details. if args.verbose: connection_template = '%3dx %s <-> %s' if args.directed: connection_template = connection_template.replace('<', '') print() for rel in sorted(relations, key=lambda x: str.lower(x[0])): print(connection_template % (rel[2], rel[0], rel[1])) print() print('Found %d nicknames in %d relations.' % (len(nicknames), len(relations))) # Store result. data = { 'nicknames': list(nicknames), 'relations': relations, 'directed': args.directed, } save_data(data, args.output_filename)
def __enter__(self): self.filename = mktemp() save_data(self.data, self.filename) return self.filename