Fixed ArcGIS Pro bug #35

opengeos · Mar 11, 2022 · 1878f42 · 1878f42
1 parent afa1da6
commit 1878f42
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/WBT/PRE/automation.py b/WBT/PRE/automation.py
@@ -570,7 +570,7 @@ def get_book_tag(tool_name, category):
             line = "            if os.path.exists('./{}'.format(unzipped_dir_name)):\n"
         if line.strip() == "shutil.rmtree(f'./{unzipped_dir_name}')":
             line = "                shutil.rmtree('./{}'.format(unzipped_dir_name))\n"
-        if "urllib.request" in line:
+        if "urllib.request" in line and "import" not in line:
             line = line.replace("urllib.request", "urlopen")
 
         outlines.append(line)

diff --git a/WBT/whitebox_tools.py b/WBT/whitebox_tools.py
@@ -23,7 +23,7 @@
 if sys.version_info.major == 2:
     import urllib2 as urlopen
 else:
-    import urlopen as urlopen
+    import urllib.request as urlopen
 
 running_windows = platform.system() == 'Windows'