Update load_rock.py

load_rock.py

@@ -3,6 +3,9 @@ from bs4 import BeautifulSoup as bs
 import random as r
 import requests
 headers = {'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/39.0.2171.95 Safari/537.36'}
+
+
+
 def get_gallery_image_list_from_rock_id(chosen_rock_id):
     system(
         f'{curl_invoker} -A "Mozilla/5.0 (platform; rv:geckoversion) Gecko/geckotrail Firefox/firefoxversion" -o "{rs}/app_data/html/index.html" https://www.mindat.org/gm/{chosen_rock_id}'
@@ -33,9 +36,10 @@ def get_rock_name_url():
     name = rand_rock_key()
     url = url_of_rock(name)
     path = f"./assets/img.{url.split('.')[-1]}"
-    with open(path, "wb") as f:
-        img = requests.get(url,headers = headers).content
-        print(img)
-        f.write(img)
+    system(f'{curl_invoker} -A "Mozilla/5.0 (platform; rv:geckoversion) Gecko/geckotrail Firefox/firefoxversion" {url} -o {path}')
+    # with open(path, "wb") as f:
+    #     img = requests.get(url,headers = headers).content
+    #     print(img)
+    #     f.write(img)
     # url, _ = urlretrieve(url)
     return name, path