aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/mutual_friends.py
diff options
context:
space:
mode:
Diffstat (limited to 'mutual_friends.py')
-rw-r--r--mutual_friends.py75
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)