aashraychegu commited on
Commit
323d7c9
·
1 Parent(s): 48982db

Update load_rock.py

Browse files
Files changed (1) hide show
  1. load_rock.py +5 -5
load_rock.py CHANGED
@@ -4,6 +4,8 @@ import random as r
4
  import requests
5
  from PIL import Image
6
  from io import BytesIO
 
 
7
  headers = { 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3',
8
  "Accept" : "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7",
9
  "Accept-Encoding": "gzip, deflate, br",
@@ -57,8 +59,6 @@ def get_rock_name_url():
57
  path = f"{rs}/assets/img.{url.split('.')[-1]}"
58
  print(path)
59
  # system(f'{curl_invoker} -A "Mozilla/5.0 (platform; rv:geckoversion) Gecko/geckotrail Firefox/firefoxversion" {url}')
60
- with open(path, "wb") as f:
61
- img = requests.get(url,headers = headers).content
62
- print(img)
63
- f.write(img)
64
- return name, Image.fromstring(BytesIO(img))
 
4
  import requests
5
  from PIL import Image
6
  from io import BytesIO
7
+ import binascii
8
+
9
  headers = { 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3',
10
  "Accept" : "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7",
11
  "Accept-Encoding": "gzip, deflate, br",
 
59
  path = f"{rs}/assets/img.{url.split('.')[-1]}"
60
  print(path)
61
  # system(f'{curl_invoker} -A "Mozilla/5.0 (platform; rv:geckoversion) Gecko/geckotrail Firefox/firefoxversion" {url}')
62
+ img = requests.get(url,headers = headers).content
63
+
64
+ return name, Image.open(BytesIO(binascii.unhexlify(img)))