Profile pic sync from KAK FB group is improved

PS: Tried using windows, are line endings borked?
This commit is contained in:
nikolark committed 2015-08-24 21:44:29 +02:00
1 parent 14c57bf9eb
commit 4488acdfcb
7 files changed
+53 -90

No files matched your search

+1
View File
@@ -0,0 +1 @@
__author__ = 'Nikolai'
+47
View File
@@ -0,0 +1,47 @@
# coding: utf-8
from django.contrib.auth.models import User
import difflib
from django.core.management.base import BaseCommand, CommandError
import facebook
from main.models import UserProfile
class Command(BaseCommand):
"""
- Fetches members of KAK Facebook group
- Matches names against the local DB
- Saves the users Facebook picture in his/her user profile.
"""
KAK_GROUP_ID = '108265352570560'
graph = None
def get_kak_members(self):
members = self.graph.get_connections(id=self.KAK_GROUP_ID, connection_name='members')
return [(member['name'], member['id']) for member in members['data']]
def handle(self, *args, **options):
if len(args) != 1:
raise CommandError('Missing access token')
access_token = args[0]
self.graph = facebook.GraphAPI(access_token=access_token)
group_members = self.get_kak_members()
names = [] # to fuzzy search for a name
name_lookup = {} # to update the matching UserProfile object
for u in User.objects.all().order_by('id'):
fullname = '{} {}'.format(u.first_name, u.last_name)
names.append(fullname)
name_lookup[fullname] = u.id
for member in group_members:
match = difflib.get_close_matches(member[0], names, 1) # max 1 match
if match:
u = UserProfile.objects.get(pk=name_lookup[match[0]])
if str(u.image) is "":
u.image = 'https://graph.facebook.com/{0}/picture'.format(member[1])
u.save()