mrp_bot/vk_api/utils.py

# -*- 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))