Profile pic sync from KAK FB group is improved
PS: Tried using windows, are line endings borked?
This commit is contained in:
1 parent
14c57bf9eb
commit
4488acdfcb
7 files changed
+53
-90
No files matched your search
@@ -0,0 +1 @@
|
||||
__author__ = 'Nikolai'
|
||||
@@ -0,0 +1 @@
|
||||
__author__ = 'Nikolai'
|
||||
@@ -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()
|
||||
|
||||
Reference in new issue
Block a user