Example #1
0
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