From d875f984db0937872f87a694a48c8af882049456 Mon Sep 17 00:00:00 2001
From: Halton Huo <halton.huo@microsoft.com>
Date: Wed, 7 Sep 2022 18:20:54 +0800
Subject: [PATCH] Fix download_dml.py issue and re-style the script.

Fixes #77.
---
 content/browser/ml/webnn/dml/download_dml.py | 81 ++++++++++----------
 1 file changed, 41 insertions(+), 40 deletions(-)

diff --git a/content/browser/ml/webnn/dml/download_dml.py b/content/browser/ml/webnn/dml/download_dml.py
index fc7c120ef1996e..705761786176bb 100644
--- a/content/browser/ml/webnn/dml/download_dml.py
+++ b/content/browser/ml/webnn/dml/download_dml.py
@@ -1,12 +1,12 @@
 #!/usr/bin/env python3
-import argparse
+# Copyright 2022 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+import json
 import os
-import shutil
-import stat
 import sys
-import tarfile
-import tempfile
-import time
+import zipfile
 
 try:
   from urllib2 import HTTPError, URLError, urlopen
@@ -14,58 +14,59 @@
   from urllib.error import HTTPError, URLError
   from urllib.request import urlopen
 
-import zipfile
-import json
-
-
 # The code is based on https://github.com/microsoft/DirectML/blob/master/Python/setup.py that uses
-# the MIT license (https://github.com/microsoft/DirectML/blob/master/LICENSE). 
+# the MIT license (https://github.com/microsoft/DirectML/blob/master/LICENSE).
 
 dml_feed_url = 'https://api.nuget.org/v3/index.json'
 dml_resource_id = 'microsoft.ai.directml'
 dml_resource_version = '1.8.2'
 
-dependency_dir = '../../../../../../third_party'
-dml_bin_path = f'{dependency_dir}/{dml_resource_id}.{dml_resource_version}/bin/x64-win/'
 base_path = os.path.dirname(os.path.realpath(__file__))
-dependency_path = os.path.join(base_path, dependency_dir)
+dependency_dir = os.path.abspath(os.path.join(
+    base_path, os.pardir, os.pardir, os.pardir,
+    os.pardir, os.pardir, 'third_party'))
 
 dml_resource_name = '.'.join([dml_resource_id, dml_resource_version])
-dml_path = '%s\%s' % (dependency_path, dml_resource_name)
+dml_path = '%s\%s' % (dependency_dir, dml_resource_name)
 
 def get_resource_url(feed_url, resource_id, resource_version):
-    index = urlopen(feed_url)
-    resources = json.loads(index.read())['resources']
+  index = urlopen(feed_url)
+  resources = json.loads(index.read())['resources']
 
-    for resource in resources:
-        if resource['@type'] == 'PackageBaseAddress/3.0.0':
-            return resource['@id'] + '/'.join([resource_id, resource_version]) + '/' + '.'.join([resource_id, resource_version]) + '.nupkg'
+  for resource in resources:
+    if resource['@type'] == 'PackageBaseAddress/3.0.0':
+      return resource['@id'] + '/'.join([resource_id, resource_version]) + \
+             '/' + '.'.join([resource_id, resource_version]) + '.nupkg'
 
-    return ''
+  return ''
 
 def download_nupkg(feed_url, resource_id, resource_version, resource_path):
-    if not os.path.exists(resource_path):
-        url = get_resource_url(feed_url, resource_id, resource_version)
-        if url:
-            print('downloading ' + url)
-            # download the package
-            resource_file = resource_path + '.nupkg'
-            with open(resource_file, 'wb') as file:
-                result = urlopen(url)
-                while True:
-                  block = result.read(1024)
-                  if not block:
-                    break
-                  file.write(block)
+  if os.path.exists(resource_path):
+    return
+
+  url = get_resource_url(feed_url, resource_id, resource_version)
+  if not url:
+    return
+
+  print('Downloading ' + url + ' to ' + resource_path)
+  # download the package
+  resource_file = resource_path + '.nupkg'
+  with open(resource_file, 'wb') as file:
+    result = urlopen(url)
+    while True:
+      block = result.read(1024)
+      if not block:
+        break
+      file.write(block)
 
-            if os.path.exists(resource_file):
-                # nupkg is just a zip, unzip it
-                with zipfile.ZipFile(resource_file, "r") as zip_ref:
-                    zip_ref.extractall(resource_path)
-                os.remove(resource_file)
+  if os.path.exists(resource_file):
+    # nupkg is just a zip, unzip it
+    with zipfile.ZipFile(resource_file, "r") as zip_ref:
+      zip_ref.extractall(resource_path)
+    os.remove(resource_file)
 
 def main():
-    download_nupkg(dml_feed_url, dml_resource_id, dml_resource_version, dml_path)
+  download_nupkg(dml_feed_url, dml_resource_id, dml_resource_version, dml_path)
 
 if __name__ == '__main__':
   sys.exit(main())