Added the downloader for images
This commit is contained in:
58
scripts/downloader.py
Normal file
58
scripts/downloader.py
Normal file
@@ -0,0 +1,58 @@
|
||||
from pathlib import Path
|
||||
from time import sleep
|
||||
import requests
|
||||
import json
|
||||
import os
|
||||
|
||||
if not Path("images/").exists():
|
||||
os.mkdir("images/")
|
||||
|
||||
if not Path("images/weapons/").exists():
|
||||
os.mkdir("images/weapons/")
|
||||
|
||||
if not Path("images/cases/").exists():
|
||||
os.mkdir("images/cases/")
|
||||
|
||||
with open("items.json") as f:
|
||||
items = json.load(f)
|
||||
|
||||
weapon_images = []
|
||||
case_images = []
|
||||
|
||||
for item in items:
|
||||
weapon_images.append(item["img_url"])
|
||||
for case in item["cases"]:
|
||||
case_images.append(case["img_url"])
|
||||
|
||||
weapon_images = list(set(weapon_images))
|
||||
case_images = list(set(case_images))
|
||||
|
||||
for url in weapon_images:
|
||||
print(f"\rDownloading {url}")
|
||||
r = requests.get(url.replace(".png", ".webp"))
|
||||
r.cookies.set("cookie_notice_accepted", "true")
|
||||
while r.status_code == 429:
|
||||
sleep(0.2)
|
||||
r = requests.get(url.replace(".png", ".webp"))
|
||||
r.cookies.set("cookie_notice_accepted", "true")
|
||||
if r.status_code == 200:
|
||||
with open(f"images/weapons/{url.split('/')[-1]}", "wb") as f:
|
||||
r.raw.decode_content = True
|
||||
f.write(r.content)
|
||||
else:
|
||||
print(f"Error {r.reason}, {r.status_code}")
|
||||
|
||||
for url in case_images:
|
||||
print(f"\rDownloading {url}")
|
||||
r = requests.get(url.replace(".png", ".webp"))
|
||||
r.cookies.set("cookie_notice_accepted", "true")
|
||||
while r.status_code == 429:
|
||||
sleep(0.2)
|
||||
r = requests.get(url.replace(".png", ".webp"))
|
||||
r.cookies.set("cookie_notice_accepted", "true")
|
||||
if r.status_code == 200:
|
||||
with open(f"images/cases/{url.split('/')[-1]}", "wb") as f:
|
||||
r.raw.decode_content = True
|
||||
f.write(r.content)
|
||||
else:
|
||||
print(f"Error {r.reason}, {r.status_code}")
|
||||
Reference in New Issue
Block a user