eryn/rogueojiiofwales
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

DRYed some stuff

Browse source
This commit is contained in:
Jonas Obrist 2012-08-18 18:15:14 +02:00
parent 903be7f383
commit d0ccd61afd
1 changed files with 13 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

34
githubnetwork/models.py
View file

@ -73,28 +73,20 @@ class GHUser(BaseAPIModel):
self.complete = True
self.save()
# set followers/following
followers = []
cache = {}
for shortuser in api.get_iter('users/%s/followers' % self.gh_login):
follower = cache.get(shortuser['login'], None)
if not follower:
try:
follower = GHUser.objects.get(gh_login=shortuser['login'])
except self.DoesNotExist:
follower = GHUser.objects.create(**self._translate(shortuser))
followers.append(follower)
cache[follower.gh_login] = follower
self.followers = followers
following = []
for shortuser in api.get_iter('users/%s/following' % self.gh_login):
follower = cache.get(shortuser['login'], None)
if not follower:
try:
follower = GHUser.objects.get(gh_login=shortuser['login'])
except self.DoesNotExist:
follower = GHUser.objects.create(**self._translate(shortuser))
following.append(follower)
self.following = following
def inner(iter):
users = []
for shortuser in iter:
user = cache.get(shortuser['login'], None)
if not user:
try:
user = GHUser.objects.get(gh_login=shortuser['login'])
except self.DoesNotExist:
user = GHUser.objects.create(**self._translate(shortuser))
users.append(user)
cache[user.gh_login] = user
self.followers = inner(api.get_iter('users/%s/followers' % self.gh_login))
self.following = inner(api.get_iter('users/%s/followers' % self.gh_login))
return self