如何使我的程序与asyncio异步?

时间:2018-01-07 02:35:07

标签: python python-3.x asynchronous beautifulsoup python-asyncio

这是我的代码:

import urllib
import webbrowser
from bs4 import BeautifulSoup
import requests
import re

address = 'https://google.com/search?q='
# Default Google search address start
file = open( "OCR.txt", "rt" )
# Open text document that contains the question
word = file.read()
file.close()

myList = [item for item in word.split('\n')]
newString = ' '.join(myList)
# The question is on multiple lines so this joins them together with proper spacing

qstr = urllib.parse.quote_plus(newString)
# Encode the string

newWord = address + qstr
# Combine the base and the encoded query

response = requests.get(newWord)

#with open('output.html', 'wb') as f:
#    f.write(response.content)
#webbrowser.open('output.html')

answers = open("ocr2.txt", "rt")

ansTable = answers.read()
answers.close()

ans = ansTable.splitlines()

ans1 = str(ans[0])
ans2 = str(ans[2])
ans3 = str(ans[4])

ans1Score = 0
ans2Score = 0
ans3Score = 0

links = []

soup = BeautifulSoup(response.text, 'lxml')

for r in soup.find_all(class_='r'):

    linkRaw = str(r)

    link = re.search("(?P<url>https?://[^\s]+)", linkRaw).group("url")

    if '&' in link:

        finalLink = link.split('&')
        link = str(finalLink[0])

    links.append(link)

#print(links)
#print(' ')

for g in soup.find_all(class_='g'):

    webBlock = str(g)

    ans1Tally = webBlock.count(ans1)
    ans2Tally = webBlock.count(ans2)
    ans3Tally = webBlock.count(ans3)

    if  ans1 in webBlock:

        ans1Score += ans1Tally

    else:

        ans1Found = False

    if ans2 in webBlock:

        ans2Score += ans2Tally

    else:

        ans2Found = False

    if ans3 in webBlock:

        ans3Score += ans3Tally

    else:

        ans3Found = False

    if ans1Found and ans2Found and ans3Found is False:

        searchLink = str(links[0])

        if searchLink.endswith('pdf'):
            pass

        else:

            response2 = requests.get(searchLink)
            soup2 = BeautifulSoup(response2.text, 'lxml')

            for p in soup2.find_all('p'):

                extraBlock = str(p)

                extraAns1Tally = extraBlock.count(ans1)
                extraAns2tally = extraBlock.count(ans2)
                extraAns3Tally = extraBlock.count(ans3)

                if ans1 in extraBlock:

                    ans1Score += extraAns1Tally

                if ans2 in extraBlock:

                    ans2Score += extraAns2Tally

                if ans3 in extraBlock:

                    ans3Score += extraAns3Tally

                with open("Results.txt", "w") as results:
                    results.write(newString + '\n\n')    
                    results.write(ans1+": "+str(ans1Score)+'\n')
                    results.write(ans2+": "+str(ans2Score)+'\n')
                    results.write(ans3+": "+str(ans3Score))

    links.pop(0)

    print(' ')
    print('-----')
    print(ans1+": "+str(ans1Score))
    print(ans2+": "+str(ans2Score))
    print(ans3+": "+str(ans3Score))
    print('-----')

现在基本上它是一次抓取每个“g”一个,当这个程序可以从同时抓取每个链接大量受益。例如,我想让它们同时进行所有刮擦,而不是等到它完成之前。很抱歉,如果这是一个简单的问题,但我对asyncio的经验很少,所以如果有人可以提供帮助,那将非常感激。谢谢!

1 个答案:

答案 0 :(得分:3)

要编写您需要的异步程序:

  • 使用async def
  • 定义函数
  • 使用await
  • 进行调用
  • 创建事件循环并在其中运行一些功能
  • 使用asyncio.gather
  • 同时运行请求

所有其他几乎与往常一样。您应该使用一些非同步模块,而不是使用阻止request模块。例如,aiohttp

python -m pip install aiohttp

并像这样使用它:

async def get(url):
    async with aiohttp.ClientSession() as session:
        async with session.get('https://api.github.com/events') as resp:
            return await resp.text()

这是代码,其中包含我所声明的一些更改。我没有检查它是否真的有用,因为我没有你使用的文件。您还应该在g in soup.find_all(class_='g'):内部移动逻辑以分离函数,并使用asyncio.gather运行多个这些函数以使asyncio受益。

import asyncio
import aiohttp
import urllib
import webbrowser
from bs4 import BeautifulSoup
import re


async def get(url):
    async with aiohttp.ClientSession() as session:
        async with session.get('https://api.github.com/events') as resp:
            return await resp.text()


async def main():
    address = 'https://google.com/search?q='
    # Default Google search address start
    file = open( "OCR.txt", "rt" )
    # Open text document that contains the question
    word = file.read()
    file.close()

    myList = [item for item in word.split('\n')]
    newString = ' '.join(myList)
    # The question is on multiple lines so this joins them together with proper spacing

    qstr = urllib.parse.quote_plus(newString)
    # Encode the string

    newWord = address + qstr
    # Combine the base and the encoded query

    text = await get(newWord)

    #with open('output.html', 'wb') as f:
    #    f.write(response.content)
    #webbrowser.open('output.html')

    answers = open("ocr2.txt", "rt")

    ansTable = answers.read()
    answers.close()

    ans = ansTable.splitlines()

    ans1 = str(ans[0])
    ans2 = str(ans[2])
    ans3 = str(ans[4])

    ans1Score = 0
    ans2Score = 0
    ans3Score = 0

    links = []

    soup = BeautifulSoup(text, 'lxml')

    for r in soup.find_all(class_='r'):

        linkRaw = str(r)

        link = re.search("(?P<url>https?://[^\s]+)", linkRaw).group("url")

        if '&' in link:

            finalLink = link.split('&')
            link = str(finalLink[0])

        links.append(link)

    #print(links)
    #print(' ')

    for g in soup.find_all(class_='g'):

        webBlock = str(g)

        ans1Tally = webBlock.count(ans1)
        ans2Tally = webBlock.count(ans2)
        ans3Tally = webBlock.count(ans3)

        if  ans1 in webBlock:

            ans1Score += ans1Tally

        else:

            ans1Found = False

        if ans2 in webBlock:

            ans2Score += ans2Tally

        else:

            ans2Found = False

        if ans3 in webBlock:

            ans3Score += ans3Tally

        else:

            ans3Found = False

        if ans1Found and ans2Found and ans3Found is False:

            searchLink = str(links[0])

            if searchLink.endswith('pdf'):
                pass

            else:

                text2 = await get(searchLink)
                soup2 = BeautifulSoup(text2, 'lxml')

                for p in soup2.find_all('p'):

                    extraBlock = str(p)

                    extraAns1Tally = extraBlock.count(ans1)
                    extraAns2tally = extraBlock.count(ans2)
                    extraAns3Tally = extraBlock.count(ans3)

                    if ans1 in extraBlock:

                        ans1Score += extraAns1Tally

                    if ans2 in extraBlock:

                        ans2Score += extraAns2Tally

                    if ans3 in extraBlock:

                        ans3Score += extraAns3Tally

                    with open("Results.txt", "w") as results:
                        results.write(newString + '\n\n')    
                        results.write(ans1+": "+str(ans1Score)+'\n')
                        results.write(ans2+": "+str(ans2Score)+'\n')
                        results.write(ans3+": "+str(ans3Score))

        links.pop(0)

        print(' ')
        print('-----')
        print(ans1+": "+str(ans1Score))
        print(ans2+": "+str(ans2Score))
        print(ans3+": "+str(ans3Score))
        print('-----')


if __name__ ==  '__main__':
    loop = asyncio.get_event_loop()
    try:
        loop.run_until_complete(main())
    finally:
        loop.run_until_complete(loop.shutdown_asyncgens())
        loop.close()

<强> UPD:

主要思想是将请求内部循环的逻辑移动到单独的协同程序中,并将这些协同程序的多个传递给asyncio.gather。它会并行化您的请求。

async def main():
    # Her do all that are before the loop.

    coros = [
        process_single_g(g)
        for g
        in soup.find_all(class_='g')
    ]

    results = await asyncio.gather(*coros)  # this function will run multiple tasks concurrently
                                            # and return all results together.

    for res in results:
        ans1Score, ans2Score, ans3Score = res

        print(' ')
        print('-----')
        print(ans1+": "+str(ans1Score))
        print(ans2+": "+str(ans2Score))
        print(ans3+": "+str(ans3Score))
        print('-----')



async def process_single_g(g):
    # Here do all things you inside loop for concrete g.

    text2 = await get(searchLink)

    # ...

    return ans1Score, ans2Score, ans3Score