4.5

Обработка проверки безопасности (ввод номера).

Author: python273

.gitignore

@@ -1,4 +1,8 @@
 build/
 dist/
+test/
 *.pyc
 upload_to_PyPi.py
+.project
+.pydevproject
+.settings/

example.py

@@ -5,10 +5,10 @@
 __email__ = "mikeking568@gmail.com"
 __contact__ = "https://vk.com/python273"
 
-""" Пример получения последнего сообщения со стены """
-
 
 def main():
+    u""" Пример получения последнего сообщения со стены """
+
     login = u'python@vk.com'
     password = u'mypassword'
 
@@ -21,7 +21,7 @@ def main():
     values = {
         'count': 1  # Получаем только одно сообщение
     }
-    response = vk.method('wall.get', values)  # С использованием метода wall.get
+    response = vk.method('wall.get', values)  # Используем метод wall.get
     print(response[1]['text'])  # Печатаем текст последнего поста со стены
 
 if __name__ == '__main__':

jconfig/jconfig.py

@@ -27,5 +27,8 @@ def parse(self):
             self.update()
             return {}
 
-    def update(self, settings={}):
-        settings = json.dump(settings, open(self.filename, 'w'))
+    def update(self, settings=None):
+        if not settings:
+            settings = {}
+
+        json.dump(settings, open(self.filename, 'w'))

setup.py

@@ -3,7 +3,7 @@
 
 from distutils.core import setup
 setup(name='vk_api',
-      version='4.4',
+      version='4.5',
       description='Module to use API VKontakte vk.com',
       author='Kirill Python',
       author_email='mikeking568@gmail.com',

vk_api/__init__.py

@@ -1,5 +1,5 @@
 __author__ = "Kirill Python"
-__version__ = "4.4"
+__version__ = "4.5"
 __email__ = "mikeking568@gmail.com"
 __contact__ = "https://vk.com/python273"
 

vk_api/upload/__init__.py

@@ -1 +1,5 @@
-from vk_upload import *
+import sys
+if sys.version_info[0] == 2:
+    from vk_upload import *
+else:
+    from .vk_upload import *

vk_api/upload/vk_upload.py

@@ -1,6 +1,7 @@
-class vk_upload():
+class VkUpload():
     def __init__(self, vk):
         self.vk = vk
+        # https://vk.com/dev/upload_files
 
     def photo(self, aid, photos):
         """ Загрузка изображений в альбом пользователя

vk_api/vk_api.py

@@ -1,113 +1,149 @@
 # -*- coding: utf-8 -*-
-import requests
-from jconfig import config
+import jconfig
 import re
-
-cj_from_dict = requests.utils.cookiejar_from_dict
+import requests
 
 
 class VkApi():
     def __init__(self,
-                 login=None, password=None,
+                 login=None, password=None, number=None,
                  sid=None, token=None,
                  app_id=2895443, scope=2097151, proxies=None):
+        u""" Данные для авторизации:
+             login, password
+
+             number - На случай проверки безопасности
+                     Номер: +7 12345678 90
+                     Передавать в number: 12345678
+
+            Для авторизации без логина и пароля:
+             sid, token
+
+            app_id - айди Standalone-приложения
+            scope - права приложения числом
+            proxies - использование прокси
+                    {'http': 'http://127.0.0.1:8888/',
+                    'https' : 'https://127.0.0.1:8888/'}
+
+        """
 
         self.login = login
         self.password = password
+        self.number = number
+
         self.sid = sid
         self.token = {'access_token': token}
-        self.settings = config(login)
+        self.settings = jconfig.config(login)
 
         self.app_id = app_id
         self.scope = scope
 
         self.http = requests.Session()
-        self.http.proxies = proxies  # Ставим прокси если есть
+        self.http.proxies = proxies  # Ставим прокси
         self.http.headers = {  # Притворимся браузером
-            'User-agent': 'Mozilla/5.0 (Windows NT 6.1; WOW64; rv:20.0) Gecko/20100101 Firefox/20.0'
+            'User-agent': 'Mozilla/5.0 (Windows NT 6.1; WOW64; rv:20.0) ' \
+                            'Gecko/20100101 Firefox/20.0'
         }
         self.http.verify = False
 
-        if login and password:  # Oh...
+        if login and password:
             self.sid = self.settings['remixsid']
             self.token = self.settings['access_token']
 
             if not self.check_sid():
-                if not self.vk_login():
-                    raise authorization_error('Authorization error (bad password)')
+                self.vk_login()
+
             if not self.check_token():
-                if not self.api_login():
-                    raise authorization_error('Authorization error (api)')
+                self.api_login()
 
-    def vk_login(self, captcha_sid='', captcha_key=''):
-        """ Авторизцаия ВКонтакте с получением cookies remixsid """
+    def vk_login(self, captcha_sid=None, captcha_key=None):
+        u""" Авторизцаия ВКонтакте с получением cookies remixsid """
 
         url = 'http://login.vk.com/'
-        data = {
+        values = {
             'act': 'login',
             'utf8': '1',
-            'captcha_sid': captcha_sid,
-            'captcha_key': captcha_key,
             'email': self.login,
             'pass': self.password
         }
 
-        if 'remixsid' in self.http.cookies:
-            self.http.cookies.pop('remixsid')
-        response = self.http.post(url, data)
+        if captcha_sid and captcha_key:
+            values.update({
+                'captcha_sid': captcha_sid,
+                'captcha_key': captcha_key
+            })
+
+        self.http.cookies.clear()
+        response = self.http.post(url, values)
 
         if 'remixsid' in self.http.cookies:
             remixsid = self.http.cookies['remixsid']
             self.settings['remixsid'] = remixsid
 
-            # Нужно в авторизации через api_login
+            # Нужно для авторизации в API
             self.settings['forapilogin'] = {
                 'p': self.http.cookies['p'],
                 'l': self.http.cookies['l']
             }
 
             self.sid = remixsid
-            return True
 
         elif 'sid=' in response.url:
             raise authorization_error('Authorization error (capcha)')
+        else:
+            raise authorization_error('Authorization error (bad password)')
+
+        if 'security_check' in response.url:
+            if self.number:
+                number_hash = regexp(r'security_check.*?hash: \'(.*?)\'\};',
+                                     response.text)[0]
+
+                values = {
+                    'act': 'security_check',
+                    'al': '1',
+                    'al_page': '3',
+                    'code': self.number,
+                    'hash': number_hash,
+                    'to': ''
+                }
+
+                response = self.http.post('http://vk.com/login.php', values)
+
+                if response.text.split('<!>')[4] == '4':
+                    return
 
-        elif 'security_check' in response.url:
             raise authorization_error('Authorization error (enter number)')
 
     def check_sid(self):
-        """ Прверка Cookies remixsid на валидность """
+        u""" Прверка Cookies remixsid на валидность """
 
         if self.sid:
-            url = 'https://vk.com/feed2.php'
-            self.http.cookies = cj_from_dict({
+            url = 'http://vk.com/feed2.php'
+            self.http.cookies.update({
                 'remixsid': self.sid,
                 'remixlang': '0',
                 'remixsslsid': '1'
             })
 
-            response = self.http.post(url).json()
+            response = self.http.get(url).json()
 
             if response['user']['id'] != -1:
                 return response
 
     def api_login(self):
-        """ Получение токена через Desktop приложение """
+        u""" Получение токена через Desktop приложение """
 
-        url = 'https://oauth.vk.com/authorize'
-        data = {
+        url = 'http://oauth.vk.com/authorize'
+        values = {
             'client_id': self.app_id,
             'scope': self.scope,
             'response_type': 'token',
         }
 
-        cookies = {}
-        cookies.update(self.settings['forapilogin'])
-        cookies.update({'remixsid': self.sid})
-        cookies = cj_from_dict(cookies)
-        self.http.cookies = cookies
+        self.http.cookies.update(self.settings['forapilogin'])
+        self.http.cookies.update({'remixsid': self.sid})
 
-        response = self.http.post(url, data)
+        response = self.http.post(url, values)
 
         if not 'access_token' in response.url:
             url = regexp(r'location\.href = "(.*?)"\+addr;', response.text)[0]
@@ -123,34 +159,46 @@ def api_login(self):
 
             self.settings['access_token'] = token
             self.token = token
-            return True
+        else:
+            raise authorization_error('Authorization error (api)')
 
     def check_token(self):
-        """ Прверка access_token на валидность """
+        u""" Прверка access_token на валидность """
 
         if self.token.get('access_token'):
             try:
                 self.method('isAppUser')
-            except:
+            except api_error:
                 return False
 
             return True
 
-    def method(self, method, data={}):
-        """ Использование методов API """
+    def method(self, method, values=None):
+        u""" Использование методов API
+
+            method - название метода
+                        'users.get'
+
+            values - параметры
+                        {'uids': 1}
+
+        """
+
+        url = 'http://api.vk.com/method/%s' % method
 
-        url = 'https://api.vk.com/method/%s' % method
-        data.update({'access_token': self.token['access_token']})
+        if not values:
+            values = {}
+        values.update({'access_token': self.token['access_token']})
 
-        response = self.http.post(url, data).json()
+        response = self.http.post(url, values).json()
         if 'error' in response:
             raise api_error(response['error'])
         else:
             return response['response']
 
 
 def regexp(reg, string):
-    """ Поиск по регулярке """
+    u""" Поиск по регулярке """
 
     reg = re.compile(reg, re.IGNORECASE | re.DOTALL)
     reg = reg.findall(string)