Add output feature

 Author:    lenrys29 <lenrys29@gmail.com>

Author: Lenrys29

docker_pull.py

@@ -12,7 +12,10 @@
 urllib3.disable_warnings()
 
 # Graphical needs for drawing full line
-console_rows, console_columns = os.popen('stty size', 'r').read().split()
+try:
+	console_rows, console_columns = os.popen('stty size', 'r').read().split()
+except:
+	console_rows, console_columns = 20 , 20
 
 ############# DEFAULTs VAR
 
@@ -23,6 +26,7 @@
 
 username = ""
 password = ""
+output_path = "."
 
 json_manifest_type='application/vnd.docker.distribution.manifest.v2+json'
 json_manifest_type_bis='application/vnd.docker.distribution.manifest.list.v2+json'
@@ -101,7 +105,10 @@ def progress_bar(ublob, nb_traits):
 ############## Check if args < 2
 
 if len(sys.argv) < 2 :
-	print('Usage:\n\tdocker_pull.py [registry/][repository/]image[:tag|@digest] [username] [password]\n')
+	print ('Usage:')
+	print ('\t docker_pull.py [registry/][repository/]image[:tag|@digest] ')
+	print ('\t docker_pull.py [registry/][repository/]image[:tag|@digest] output_path')
+	print ('\t docker_pull.py [registry/][repository/]image[:tag|@digest] username password output_path\n')
 	exit(1)
 
 ############## Get info from arg 
@@ -113,6 +120,12 @@ def progress_bar(ublob, nb_traits):
     username = sys.argv[2] 
     password = sys.argv[3] 
 
+if len(sys.argv) == 3:
+	output_path = sys.argv[2]
+
+if len(sys.argv) == 5:
+	output_path = sys.argv[4]
+
 ############## Get repository url + registry url for auth
 
 if len(imgparts) > 1 and ('.' in imgparts[0] or ':' in imgparts[0]):
@@ -183,7 +196,7 @@ def progress_bar(ublob, nb_traits):
 layers = resp.json()['layers']
 
 # Create tmp folder that will hold the image
-imgdir = '/tmp/tmp_{}_{}'.format(img.replace('/', '.'), tag)
+imgdir = output_path  + '/tmp_{}'.format(sys.argv[1].replace('/', '.').replace(':','@'))
 
 if os.path.exists(imgdir):
     shutil.rmtree(imgdir)
@@ -305,7 +318,7 @@ def progress_bar(ublob, nb_traits):
 file.close()
 
 # Create image tar and clean tmp folder
-docker_tar = repository.replace('/', '_') + '_' + img.replace('/', '_') + '.tar'
+docker_tar = output_path  + "/" + sys.argv[1].replace('/', '_').replace(':','@') + '.tar'
 sys.stdout.write("Creating archive...")
 sys.stdout.flush()
 tar = tarfile.open(docker_tar, "w")