# -*- coding: utf-8 -*-

# @Time    : 2018/8/20 14:35

# @Author  : cxa

# @File    : chttp.py

# @Software: PyCharm

# !/usr/bin/env python3

# -*- coding: utf-8 -*-

from functools import partial, wraps

from random import random

import asyncio, os

from cchardet import detect

try:

    import aiohttp

except ImportError:

    raise RuntimeError('您没有安装aiohttp，请执行安装命令 pip install aiohttp  ')

result, all_tasks, connector, sessiondict = [], [], [], {}

class AhttpRequest():

    def __init__(self, method, url, timeout=None, session=False, headers=None, cookies=None, unsafe=None,

                 mark='1111111111', **kwargs):

        self.method, self.session, self.url, self.mark, self.timeout = method, session, url, mark, timeout

        callback = kwargs.pop('callback', None)

        self.callback = callback

        self.kwargs = kwargs

        if not session:

            self.sessiondict = (cookies, headers, aiohttp.CookieJar(unsafe=True) if unsafe else None)

    def run(self, pool=5, exception_handle=None):

        result = run([self], pool=pool, exception_handle=exception_handle)

        return result[0]

class WithSession():

    def __init__(self, mark, session=True):

        self.get = partial(AhttpRequest, 'GET', session=session, mark=mark)

        self.options = partial(AhttpRequest, 'OPTIONS', session=session, mark=mark)

        self.head = partial(AhttpRequest, 'HEAD', session=session, mark=mark)

        self.post = partial(AhttpRequest, 'POST', session=session, mark=mark)

        self.put = partial(AhttpRequest, 'PUT', session=session, mark=mark)

        self.patch = partial(AhttpRequest, 'PATCH', session=session, mark=mark)

        self.delete = partial(AhttpRequest, 'DELETE', session=session, mark=mark)

get = partial(AhttpRequest, 'GET')

options = partial(AhttpRequest, 'OPTIONS')

head = partial(AhttpRequest, 'HEAD')

post = partial(AhttpRequest, 'POST')

put = partial(AhttpRequest, 'PUT')

patch = partial(AhttpRequest, 'PATCH')

delete = partial(AhttpRequest, 'DELETE')

class ClientSession(aiohttp.ClientSession):

    def close(self):

        """

        对ClientSession类的close方法进行重写

        """

        if not self.closed:

            if self._connector_owner:

                self._connector.close()

            connector.append(self._connector)

        return 'ClientSession.close() is a coroutine'

def Session(cookies=None, headers=None, unsafe=None):

    mark = str(round(random() * 10 ** 10))

    sessiondict[mark] = (cookies, headers, aiohttp.CookieJar(unsafe=True) if unsafe else None)

    return WithSession(mark=mark)

def run(tasks, pool=2, exception_handle=None):

    del result[:]

    del connector[:]

    loop = asyncio.get_event_loop()

    future = asyncio.ensure_future(go(tasks, pool, exception_handle, loop=loop))

    loop.run_until_complete(future)

    # loop.close()

    return result

class AhttpResponse():

    def __init__(self, content, clientResponse):

        self.content = content

        self.clientResponse = clientResponse

    def raw(self):

        return self.clientResponse

    @property

    def url(self):

        return self.clientResponse.url

    @property

    def cookies(self):

        return self.clientResponse.cookies

    @property

    def headers(self):

        return self.clientResponse.headers

    @property

    def status(self):

        return self.clientResponse.status

    @property

    def method(self):

        return self.clientResponse.method

    def text(self, encoding=None):

        encoding = encoding or detect(self.content)['encoding']

        return self.content.decode(encoding=encoding)

    def __repr__(self):

        return "<AhttpResponse [status {}]>".format(self.clientResponse.status)

    __str__ = __repr__

async def go(tasks, pool, exception_handle, loop):

    del all_tasks[:]

    conn = aiohttp.TCPConnector(use_dns_cache=True, loop=loop, verify_ssl=False)

    sem = asyncio.Semaphore(pool)

    classify = {}

    [classify[i.mark].append(i) if classify.get(i.mark, 0) else classify.setdefault(i.mark, [i]) for i in tasks]

    print(classify)

    try:

        for i in classify.pop('1111111111'):

            all_tasks.append(control_sem(sem, i, exception_handle, session=False))

    except:

        pass

    for i in classify:

        async with ClientSession(cookies=sessiondict[i][0], headers=sessiondict[i][1], cookie_jar=sessiondict[i][2],

                                 connector_owner=False, connector=conn) as locals()['session{}'.format(i)]:

            for j in classify[i]:

                all_tasks.append(control_sem(sem, j, exception_handle, session=locals()['session{}'.format(i)]))

    await asyncio.wait(all_tasks)

    # 关闭所有连接

    for i in connector:

        i.close()

    return True

async def fetch(session, i, exception_handle):

    try:

        if session:

            async with session.request(i.method, i.url, timeout=i.timeout, **(i.kwargs)) as resp:

                content = await resp.read()

                myAhttp = AhttpResponse(content, resp)

        else:

            async with aiohttp.ClientSession(cookies=i.sessiondict[0], headers=i.sessiondict[1],

                                             cookie_jar=i.sessiondict[2]) as session2:

                async with session2.request(i.method, i.url, timeout=i.timeout, **(i.kwargs)) as resp:

                    content = await resp.read()

                    myAhttp = AhttpResponse(content, resp)

        if i.callback:

            try:

                i.callback(myAhttp)

            except:

                pass

    except Exception as e:

        myAhttp = None

        exception_handle and exception_handle(i, e)

    finally:

        result.append(myAhttp)

async def control_sem(sem, i, exception_handle, session):

    # 限制信号量

    async with sem:

        await fetch(session, i, exception_handle)

# __all__ = (

#     'Session', 'get', 'options', 'head', 'post', 'put', 'patch', 'delete'

# )

if __name__ == '__main__':

    urls = ["http://www.runoob.com/python/att-time-mktime.html",

            "https://blog.csdn.net/getcomputerstyle/article/details/71515331"]