diff options
Diffstat (limited to 'mutual_friends.py')
-rw-r--r-- | mutual_friends.py | 75 |
1 files changed, 0 insertions, 75 deletions
diff --git a/mutual_friends.py b/mutual_friends.py deleted file mode 100644 index 8823619..0000000 --- a/mutual_friends.py +++ /dev/null @@ -1,75 +0,0 @@ -# Copyright 2015 Egor Tensin <Egor.Tensin@gmail.com> -# This file is licensed under the terms of the MIT License. -# See LICENSE.txt for details. - -from collections import OrderedDict -import csv -from enum import Enum -import json -import sys - -from vk.api import API, Language -from vk.user import UserField - -OUTPUT_FIELDS = UserField.UID, UserField.FIRST_NAME, UserField.LAST_NAME, UserField.SCREEN_NAME - -def query_friend_list(api, user): - return api.friends_get(user.get_uid(), fields=OUTPUT_FIELDS) - -def extract_output_fields(user): - new_user = OrderedDict() - for field in OUTPUT_FIELDS: - new_user[str(field)] = user[field] if field in user else None - return new_user - -def print_mutual_friends_csv(mutual_friends): - writer = csv.writer(sys.stdout, lineterminator='\n') - for user in mutual_friends: - user = extract_output_fields(user) - writer.writerow(user.values()) - -def print_mutual_friends_json(mutual_friends): - print(json.dumps([extract_output_fields(user) for user in mutual_friends], indent=3)) - -def print_mutual_friends(mutual_friends, fmt): - if fmt is OutputFormat.CSV: - print_mutual_friends_csv(mutual_friends) - elif fmt is OutputFormat.JSON: - print_mutual_friends_json(mutual_friends) - else: - raise NotImplementedError('unsupported output format: ' + str(fmt)) - -class OutputFormat(Enum): - CSV = 'csv' - JSON = 'json' - - def __str__(self): - return self.value - -if __name__ == '__main__': - import argparse - - def output_format(s): - try: - return OutputFormat(s) - except ValueError: - raise argparse.ArgumentError() - - parser = argparse.ArgumentParser( - description='Learn who your ex and her new boyfriend are both friends with.') - - parser.add_argument(metavar='UID', dest='uids', nargs='+', - help='user IDs or "screen names"') - parser.add_argument('--output-format', type=output_format, - choices=tuple(fmt for fmt in OutputFormat), - default=OutputFormat.CSV, - help='specify output format') - - args = parser.parse_args() - - api = API(Language.EN) - users = api.users_get(args.uids) - - friend_lists = map(lambda user: frozenset(query_friend_list(api, user)), users) - mutual_friends = frozenset.intersection(*friend_lists) - print_mutual_friends(mutual_friends, args.output_format) |