Merge branch 'dev'

Author: python273

jconfig/jconfig.py

@@ -22,13 +22,19 @@ def __init__(self, section, filename='.jconfig'):
     def __getattr__(self, name):
         return self._settings[self._section].get(name)
 
+    def __getitem__(self, key):
+        return self.__getattr__(key)
+
     def __setattr__(self, name, value):
         if name.startswith('_'):
             super(Config, self).__setattr__(name, value)
             return
 
         self._settings[self._section][name] = value
 
+    def __setitem__(self, key, value):
+        return self.__setattr__(key, value)
+
     def clear_section(self):
         self._settings[self._section] = {}
 

setup.py

@@ -14,7 +14,7 @@
 
 setup(
     name='vk_api',
-    version='7.0',
+    version='7.1',
     author='Kirill Python',
     author_email='python273@ya.ru',
     url='https://github.com/python273/vk_api',

vk_api/__init__.py

@@ -9,7 +9,7 @@
 """
 
 __author__ = 'Kirill Python'
-__version__ = '7.0'
+__version__ = '7.1'
 __email__ = 'python273@ya.ru'
 __contact__ = 'https://vk.com/python273'
 

vk_api/vk_api.py

@@ -10,6 +10,7 @@
 
 import re
 import time
+import threading
 
 import requests
 
@@ -39,7 +40,7 @@ def __init__(self, login=None, password=None, number=None, sec_number=None,
                  proxies=None,
                  auth_handler=None, captcha_handler=None,
                  config_filename='vk_config.json',
-                 api_version='5.44', app_id=2895443, scope=33554431,
+                 api_version='5.52', app_id=2895443, scope=33554431,
                  client_secret=None):
         """
         :param login: Логин ВКонтакте
@@ -56,8 +57,9 @@ def __init__(self, login=None, password=None, number=None, sec_number=None,
                         {'http': 'http://127.0.0.1:8888/',
                         'https': 'https://127.0.0.1:8888/'}
         :param auth_handler: Функция для обработки двухфакторной аутентификации,
-                              обязана возвращать строку с кодом для
-                              прохождения аутентификации
+                              обязана возвращать строку с кодом и булевое значение,
+                              означающее, стоит ли вк запомнить это устройство, для
+                              прохождения аутентификации.
         :param captcha_handler: Функция для обработки капчи
         :param config_filename: Расположение config файла
 
@@ -85,10 +87,10 @@ def __init__(self, login=None, password=None, number=None, sec_number=None,
 
         self.http = requests.Session()
         self.http.proxies = proxies  # Ставим прокси
-        self.http.headers = {  # Притворимся браузером
+        self.http.headers.update({  # Притворимся браузером
             'User-agent': 'Mozilla/5.0 (Windows NT 6.1; rv:40.0) '
             'Gecko/20100101 Firefox/40.0'
-        }
+        })
 
         self.last_request = 0.0
 
@@ -99,10 +101,12 @@ def __init__(self, login=None, password=None, number=None, sec_number=None,
             TWOFACTOR_CODE: auth_handler or self.auth_handler
         }
 
+        self.lock = threading.Lock()
+
     def authorization(self, reauth=False):
         """ Полная авторизация с получением токена
 
-        :param reauth: Позволяет переавторизиваться, игнорируя сохраненные 
+        :param reauth: Позволяет переавторизиваться, игнорируя сохраненные
                         куки и токен
         """
 
@@ -148,8 +152,8 @@ def vk_login(self, captcha_sid=None, captcha_key=None):
         remixsid = None
 
         if 'act=authcheck' in response.url:
-            code = self.error_handlers[TWOFACTOR_CODE]()
-            response = self.twofactor(response, code)
+            code, remember_device = self.error_handlers[TWOFACTOR_CODE]()
+            response = self.twofactor(response, code, remember_device)
 
         if 'remixsid' in self.http.cookies:
             remixsid = self.http.cookies['remixsid']
@@ -188,21 +192,27 @@ def vk_login(self, captcha_sid=None, captcha_key=None):
         if 'act=blocked' in response.url:
             raise AccountBlocked('Account is blocked')
 
-    def twofactor(self, response, code):
+    def twofactor(self, response, code, remember_device=False):
         """ Двухфакторная аутентификация
         :param reponse: запрос, содержащий страницу с приглашением к аутентификации
         :param code: код, который необходимо ввести для успешной аутентификации
+        :param remember_device: параметр, означающий,
+               стоит ли запоминать это устройство в целях
+               избежания повторного ввода кода(default: False)
         """
-        assert code != None, "Empty code doesn't acceptable"
-        assert len(code) == 6, "Length of code cannot be other than 6."
+
+        if code == None:
+            raise TwoFactorError("Empty code doesn't acceptable")
+        if len(code) != 6:
+            raise TwoFactorError("Length of code cannot be other than 6.")
 
         auth_hash = search_re(RE_AUTH_HASH, response.text)
         url = 'https://vk.com/al_login.php'
         if auth_hash:
             values = {
                     'act': 'a_authcheck_code',
                     'code': code,
-                    'remember': 0, # TODO: Fix me(device remembering)
+                    'remember': int(remember_device),
                     'hash': auth_hash,
                     }
             response = self.http.post(url, values, cookies=response.cookies)
@@ -383,14 +393,15 @@ def method(self, method, values=None, captcha_sid=None, captcha_key=None):
                 'captcha_key': captcha_key
             })
 
-        # Ограничение 3 запроса в секунду
-        delay = DELAY - (time.time() - self.last_request)
+        with self.lock:
+            # Ограничение 3 запроса в секунду
+            delay = DELAY - (time.time() - self.last_request)
 
-        if delay > 0:
-            time.sleep(delay)
+            if delay > 0:
+                time.sleep(delay)
 
-        response = self.http.post(url, values)
-        self.last_request = time.time()
+            response = self.http.post(url, values)
+            self.last_request = time.time()
 
         if response.ok:
             response = response.json()
@@ -574,6 +585,7 @@ def __init__(self, vk, captcha_sid,
 
         self.key = None
         self.url = url
+        self.image = None
 
     def get_url(self):
         """ Возвращает ссылку на изображение капчи
@@ -585,6 +597,14 @@ def get_url(self):
 
         return self.url
 
+    def get_image(self):
+        """ Возвращает бинарное изображение капчи, получаемое по get_url()
+        """
+
+        if not self.image:
+            self.image = self.vk.http.get(self.get_url()).content
+        return self.image
+
     def try_again(self, key):
         """ Отправляет запрос заново с ответом капчи
 

vk_api/vk_tools.py

@@ -52,7 +52,7 @@ def get_all(self, method, max_count, values=None, key='items', limit=None):
 
         while True:
             run_code = code_get_all_items % (
-                max_count, offset, key, json.dumps(values, ensure_ascii=False),
+                max_count, offset, key, sjson_dumps(values),
                 method, method
             )
 
@@ -187,7 +187,7 @@ def gen_code_one_method(self, pool):
         method = pool[0][0]
 
         list_values = [i[1] for i in pool]
-        json_list_values = json.dumps(list_values, separators=(',', ':'))
+        json_list_values = sjson_dumps(list_values)
         run_code = code_requestspoll_one_method % (
             json_list_values, method
         )
@@ -199,8 +199,8 @@ def gen_code_one_param(self, pool):
            (если в пулле запросы к одному методу, с одним меняющеися параметром)
         """
         run_code = code_requestspoll_one_param % (
-            json.dumps(self.one_param['default'], separators=(',', ':')),
-            json.dumps(pool, separators=(',', ':')),
+            sjson_dumps(self.one_param['default']),
+            sjson_dumps(pool),
             self.one_param['key'],
             self.one_param['method']
         )
@@ -212,7 +212,7 @@ def gen_code_one_param(self, pool):
     def gen_code_many_methods(self, pool):
         """ Генерирует код для нескольких методов """
         reqs = ','.join(
-            'API.{}({})'.format(i[0], json.dumps(i[1]), separators=(',', ':'))
+            'API.{}({})'.format(i[0], sjson_dumps(i[1]))
             for i in pool
         )
         run_code = 'return [{}];'.format(reqs)
@@ -239,9 +239,18 @@ def execute(self):
                 if self.one_param:
                     self.one_param['return'][cur_pool[x]] = response[x]
                 else:
-                    self.pool[i + x][2].update(response[x])
+                    if response[x] is False:
+                        self.pool[i + x][2].update({'_error': True})
+                    else:
+                        self.pool[i + x][2].update(response[x])
 
 
+def sjson_dumps(*args, **kwargs):
+    kwargs['ensure_ascii'] = False
+    kwargs['separators'] = (',', ':')
+
+    return json.dumps(*args, **kwargs)
+
 # Полный код в файле vk_procedures
 code_get_all_items = """
 var m=%s,n=%s,b="%s",v=n;var c={count:m,offset:v}+%s;var r=API.%s(c),k=r.count,

vk_api/vk_upload.py

@@ -112,6 +112,30 @@ def photo_wall(self, photos, user_id=None, group_id=None):
 
         return response
 
+    def audio(self, file_path, artist, title):
+        """ Загрузка аудио
+
+        :param file_path: путь к аудиофайлу
+        :param artist: исполнитель
+        :param title: название
+        """
+
+        url = self.vk.method('audio.getUploadServer')['upload_url']
+
+        filetype = file_path.split('.')[-1]
+        audio_file = [(('file', ( 'file.' + filetype, open(file_path, 'rb'))))]
+
+        response = self.vk.http.post(url, files=audio_file).json()
+
+        response.update({
+            'artist': artist,
+            'title': title
+        })
+
+        response = self.vk.method('audio.save', response)
+
+        return response
+
     def document(self, file_path, title=None, tags=None, group_id=None):
         """ Загрузка документа