From 73c9eddca61a5bd84f2d38deddcc6b866cd7d2e3 Mon Sep 17 00:00:00 2001
From: space-nuko <24979496+space-nuko@users.noreply.github.com>
Date: Mon, 15 May 2023 17:48:57 -0500
Subject: [PATCH] Use python downloader instead of calls to wget

---
 disco_utils.py    | 21 ++++++++++++++++++---
 model_settings.py |  2 +-
 2 files changed, 19 insertions(+), 4 deletions(-)

diff --git a/disco_utils.py b/disco_utils.py
index 027cd64..89e834f 100644
--- a/disco_utils.py
+++ b/disco_utils.py
@@ -18,6 +18,9 @@ import torchvision.transforms.functional as TF
 import subprocess
 from importlib import util as importlibutil
 import numpy as np
+import os
+import requests
+from urllib.parse import urlparse
 
 import comfy.model_management
 
@@ -200,6 +203,18 @@ def pipie(modulestr):
   res = subprocess.run(['git', 'install', '-e', modulestr], stdout=subprocess.PIPE).stdout.decode('utf-8')
   print(res)
 
-def wget(url, outputdir):
-  res = subprocess.run(['wget', url, '-P', f'{outputdir}'], stdout=subprocess.PIPE).stdout.decode('utf-8')
-  print(res)
+def pyget(url, path=None, filename=None):
+    try:
+        response = requests.get(url)
+        response.raise_for_status()
+        parsed_url = urlparse(url)
+        filename = filename if filename else os.path.basename(parsed_url.path)
+        path = os.path.join(path, filename) if path else filename
+        with open(path, 'wb') as file:
+            file.write(response.content)
+        if os.path.exists(path):
+            return True
+        else:
+            print(f"Unable to save file to: {path}")
+    except requests.exceptions.HTTPError as errh:
+        print(f"HTTP Error: ({url}): {errh}")
diff --git a/model_settings.py b/model_settings.py
index 339fdcd..5fbb59f 100644
--- a/model_settings.py
+++ b/model_settings.py
@@ -136,7 +136,7 @@ class ModelSettings:
 
             if not diff_model_map[diffusion_model_name]['downloaded']:
                 for model_uri in diff_model_map[diffusion_model_name]['uri_list']:
-                    disco_utils.wget(model_uri, self.model_path)
+                    disco_utils.pyget(model_uri, self.model_path)
                     if os.path.exists(model_local_path):
                         diff_model_map[diffusion_model_name]['downloaded'] = True
                         return