Преглед изворни кода

AMBARI-10026: Download logs are too verbose (Eugene Chekanskiy via jluniya)

Jayush Luniya пре 10 година
родитељ
комит
1ea0ffbfae
1 измењених фајлова са 5 додато и 5 уклоњено
  1. 5 5
      ambari-common/src/main/python/ambari_commons/inet_utils.py

+ 5 - 5
ambari-common/src/main/python/ambari_commons/inet_utils.py

@@ -26,13 +26,13 @@ from exceptions import *
 from logging_utils import *
 from logging_utils import *
 from os_check import OSCheck
 from os_check import OSCheck
 
 
-def download_file(link, destination, chunk_size=16 * 1024):
+def download_file(link, destination, chunk_size=16 * 1024, progress_func = None):
   print_info_msg("Downloading {0} to {1}".format(link, destination))
   print_info_msg("Downloading {0} to {1}".format(link, destination))
   if os.path.exists(destination):
   if os.path.exists(destination):
       print_warning_msg("File {0} already exists, assuming it was downloaded before".format(destination))
       print_warning_msg("File {0} already exists, assuming it was downloaded before".format(destination))
       return
       return
 
 
-  force_download_file(link, destination, chunk_size)
+  force_download_file(link, destination, chunk_size, progress_func = progress_func)
 
 
 
 
 def download_progress(file_name, downloaded_size, blockSize, totalSize):
 def download_progress(file_name, downloaded_size, blockSize, totalSize):
@@ -73,7 +73,7 @@ def find_range_components(meta):
   return (file_size, seek_pos)
   return (file_size, seek_pos)
 
 
 
 
-def force_download_file(link, destination, chunk_size = 16 * 1024, progress_func = download_progress):
+def force_download_file(link, destination, chunk_size = 16 * 1024, progress_func = None):
   request = urllib2.Request(link)
   request = urllib2.Request(link)
 
 
   if os.path.exists(destination) and not os.path.isfile(destination):
   if os.path.exists(destination) and not os.path.isfile(destination):
@@ -127,8 +127,8 @@ def force_download_file(link, destination, chunk_size = 16 * 1024, progress_func
 
 
         file_size_dl += len(buffer)
         file_size_dl += len(buffer)
         f.write(buffer)
         f.write(buffer)
-
-        progress_func(file_name, file_size_dl, chunk_size, file_size)
+        if progress_func is not None:
+          progress_func(file_name, file_size_dl, chunk_size, file_size)
     finally:
     finally:
       f.close()
       f.close()