def vk_get_user(credentials):
user = g.users_dao.get(unicode(credentials.token_response['user_id']),
AUTH_SRC_VK)
if user is not None:
return user, False
conn = httplib2.Http()
credentials.authorize(conn)
params = {
'v': '5.37',
'lang': 'ru',
'user_ids': credentials.token_response['user_id'],
'fields': 'photo_50, photo_200_orig, city'}
resp, content = conn.request('https://api.vk.com/method/users.get',
'POST',
urllib.urlencode(params))
if resp.status != 200:
logging.error("Error getting user profile from vk api: %d",
resp.status)
logging.error(content)
return None
data = json.loads(content)['response'][0]
if 'error' in data:
logging.error("Error getting profile data")
logging.error(data['error'])
return None
user = UserMixin()
user.oauth_id = unicode(credentials.token_response['user_id'])
user.name = u"{} {}".format(
data.get('first_name', ''),
data.get('last_name', ''))
user.src = AUTH_SRC_VK
if 'city' in data:
user.location = data['city']['title']
else:
user.location = None
fd = urllib.urlopen(data['photo_200_orig'])
if fd.getcode() == 200:
user.image_id = g.images_dao.create(fd.read(), fd.info().gettype())
else:
user.image_id = None
fd = urllib.urlopen(data['photo_50'])
if fd.getcode() == 200:
user.preview_id = g.images_dao.create(fd.read(), fd.info().gettype())
else:
user.preview_id = None
user.id = g.users_dao.create(user)
return user, True
def su_get_user(credentials):
conn = httplib2.Http()
credentials.authorize(conn)
resp, content = conn.request('http://www.southural.ru/oauth2/UserInfo',
'POST')
if resp.status != 200:
logging.error("Error getting user profile from su api: %d",
resp.status)
logging.error(content)
return None
data = json.loads(content)
if 'error' in data:
logging.error("Error getting user data from su")
logging.error(data['error'])
return None
user = g.users_dao.get(data['sub'], AUTH_SRC_SU)
if user is not None:
return user, False
user = UserMixin()
user.oauth_id = data['sub']
user.name = data['name']
user.src = AUTH_SRC_SU
user.location = data.get('city', None)
user.image_id = None
user.preview_id = None
user.id = g.users_dao.create(user)
return user, True