32 lines
861 B
Python
32 lines
861 B
Python
import requests
|
|
import pathlib
|
|
from random import randint
|
|
import urllib.parse
|
|
from threading import Thread
|
|
|
|
api_url = "https://api.waifu.pics/nsfw/trap"
|
|
|
|
|
|
def get(directory="traps", amount=randint(5, 10)):
|
|
path = pathlib.Path(directory)
|
|
if not path.exists():
|
|
path.mkdir()
|
|
for i in range(amount):
|
|
Thread(target=traps, args=(directory,)).start()
|
|
|
|
|
|
def traps(directory):
|
|
req_url = requests.get(api_url)
|
|
url = req_url.json()["url"]
|
|
if not req_url.ok:
|
|
print("error:", req_url)
|
|
return
|
|
filename = urllib.parse.urlparse(url)
|
|
filename = pathlib.Path(directory, pathlib.Path(filename.path).name)
|
|
with open(filename, "wb") as f:
|
|
response = requests.get(url, stream=True)
|
|
for block in response.iter_content(1024):
|
|
if not block:
|
|
break
|
|
f.write(block)
|