-
-
Save muxuezi/e529f9a526885282ad5f to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import asyncio | |
import aiohttp | |
import bs4 | |
import tqdm | |
@asyncio.coroutine | |
def get(*args, **kwargs): | |
response = yield from aiohttp.request('GET', *args, **kwargs) | |
return (yield from response.read_and_close(decode=True)) | |
@asyncio.coroutine | |
def wait_with_progress(coros): | |
for f in tqdm.tqdm(asyncio.as_completed(coros), total=len(coros)): | |
yield from f | |
def first_magnet(page): | |
soup = bs4.BeautifulSoup(page) | |
a = soup.find('a', title='Download this torrent using magnet') | |
return a['href'] | |
@asyncio.coroutine | |
def print_magnet(query): | |
url = 'http://thepiratebay.se/search/{}/0/7/0'.format(query) | |
with (yield from sem): | |
page = yield from get(url, compress=True) | |
magnet = first_magnet(page) | |
print('{}: {}'.format(query, magnet)) | |
distros = ['archlinux', 'ubuntu', 'debian'] | |
sem = asyncio.Semaphore(5) | |
loop = asyncio.get_event_loop() | |
f = asyncio.wait([print_magnet(d) for d in distros]) | |
loop.run_until_complete(f) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment