first commit

vk_api/utils.py

@@ -0,0 +1,166 @@
+# -*- coding: utf-8 -*-
+"""
+:authors: python273
+:license: Apache License, Version 2.0, see LICENSE file
+
+:copyright: (c) 2019 python273
+"""
+
+from __future__ import print_function
+
+import random
+
+try:
+    import simplejson as json
+except ImportError:
+    import json
+
+from http.cookiejar import Cookie
+
+
+def search_re(reg, string):
+    """ Поиск по регулярке """
+    s = reg.search(string)
+
+    if s:
+        groups = s.groups()
+        return groups[0]
+
+
+def clear_string(s):
+    if s:
+        return s.strip().replace(' ', '')
+
+
+def get_random_id():
+    """ Get random int32 number (signed) """
+    return random.getrandbits(31) * random.choice([-1, 1])
+
+
+def code_from_number(prefix, postfix, number):
+    prefix_len = len(prefix)
+    postfix_len = len(postfix)
+
+    if number[0] == '+':
+        number = number[1:]
+
+    if (prefix_len + postfix_len) >= len(number):
+        return
+
+    # Сравниваем начало номера
+    if number[:prefix_len] != prefix:
+        return
+
+    # Сравниваем конец номера
+    if number[-postfix_len:] != postfix:
+        return
+
+    return number[prefix_len:-postfix_len]
+
+
+def sjson_dumps(*args, **kwargs):
+    kwargs['ensure_ascii'] = False
+    kwargs['separators'] = (',', ':')
+
+    return json.dumps(*args, **kwargs)
+
+
+HTTP_COOKIE_ARGS = [
+    'version', 'name', 'value',
+    'port', 'port_specified',
+    'domain', 'domain_specified',
+    'domain_initial_dot',
+    'path', 'path_specified',
+    'secure', 'expires', 'discard', 'comment', 'comment_url', 'rest', 'rfc2109'
+]
+
+
+def cookie_to_dict(cookie):
+    cookie_dict = {
+        k: v for k, v in cookie.__dict__.items() if k in HTTP_COOKIE_ARGS
+    }
+
+    cookie_dict['rest'] = cookie._rest
+    cookie_dict['expires'] = None
+
+    return cookie_dict
+
+
+def cookie_from_dict(d):
+    return Cookie(**d)
+
+
+def cookies_to_list(cookies):
+    return [cookie_to_dict(cookie) for cookie in cookies]
+
+
+def set_cookies_from_list(cookie_jar, l):
+    for cookie in l:
+        cookie_jar.set_cookie(cookie_from_dict(cookie))
+
+
+def enable_debug_mode(vk_session, print_content=False):
+    """ Включает режим отладки:
+        - Вывод сообщений лога
+        - Вывод http запросов
+
+    :param vk_session: объект VkApi
+    :param print_content: печатать ответ http запросов
+    """
+
+    import logging
+    import sys
+    import time
+
+    import requests
+
+    from . import __version__
+
+    pypi_version = requests.get(
+        'https://pypi.org/pypi/vk_api/json'
+    ).json()['info']['version']
+
+    if __version__ != pypi_version:
+        print()
+        print('######### MODULE IS NOT UPDATED!!1 ##########')
+        print()
+        print('Installed vk_api version is:', __version__)
+        print('PyPI vk_api version is:', pypi_version)
+        print()
+        print('######### MODULE IS NOT UPDATED!!1 ##########')
+        print()
+
+    class DebugHTTPAdapter(requests.adapters.HTTPAdapter):
+        def send(self, request, **kwargs):
+            start = time.time()
+            response = super(DebugHTTPAdapter, self).send(request, **kwargs)
+            end = time.time()
+
+            total = end - start
+
+            body = request.body
+            if body and len(body) > 1024:
+                body = body[:1024] + '[STRIPPED]'
+
+            print(
+                '{:0.2f} {} {} {} {} {} {}'.format(
+                    total,
+                    request.method,
+                    request.url,
+                    request.headers,
+                    repr(body),
+                    response.status_code,
+                    response.history
+                )
+            )
+
+            if print_content:
+                print(response.text)
+
+            return response
+
+    vk_session.http.mount('http://', DebugHTTPAdapter())
+    vk_session.http.mount('https://', DebugHTTPAdapter())
+
+    vk_session.logger.setLevel(logging.INFO)
+    vk_session.logger.addHandler(logging.StreamHandler(sys.stdout))