Add unzip

geosentinel/ui_utils.py

@@ -35,7 +35,7 @@ Functions:
 
 from argparse import ArgumentParser
 
-import os
+import os,sys
 import zipfile
 
 from collections import OrderedDict
@@ -84,16 +84,21 @@ SENTINEL_UNZIP_ARGUMENTS.add_argument("product_type", help="sentinel product typ
 
 
 def find_files(args):
+    check_existing_directory(args.folder, 2)
     folder = args.folder
     satelite = args.plattform
     files = os.listdir(folder)
-    if(satelite=="Sentinel-1"):
-        namfil1 = 'S1A_'+ args.product_type
-        namfil2 = 'S1B_'+ args.product_type
-    if(satelite=="Sentinel-2"):
-        namfil1 = 'S2A_'+ args.product_type
-        namfil2 = 'S2B_'+ args.product_type
-    files = [f for f in files if f.startswith(namfil1) or f.startswith(namfil2)]
+    if(satelite!="Sentinel-1")or(satelite!="sentinel-1")or(satelite=="Sentinel-2")or(satelite=="sentinel-2"):
+        print satelite + " not exist"
+        sys.exit()
+    else:
+        if(satelite=="Sentinel-1")or(satelite=="sentinel-1"):
+            namfil1 = 'S1A_'+ args.product_type
+            namfil2 = 'S1B_'+ args.product_type
+        if(satelite=="Sentinel-2")or(satelite=="sentinel-2"):
+            namfil1 = 'S2A_'+ args.product_type
+            namfil2 = 'S2B_'+ args.product_type
+        files = [f for f in files if f.startswith(namfil1) or f.startswith(namfil2)]
     if(len(files)==0):
         print"Files not found..."
     return files
@@ -106,13 +111,19 @@ def unzip(list_files,args):
         zip_ref.extractall(args.folder) # extract file to dir
         zip_ref.close() # close file
 
-def check_existing_directory(folder) : 
-    
-    if not os.path.exists(folder):
-        os.makedirs(folder)
-        print folder + " was created"
-    else: 
-        print folder + " exists "
+def check_existing_directory(folder,flag):
+
+    if(flag==1):
+        if not os.path.exists(folder):
+            os.makedirs(folder)
+            print folder + " was created"
+        else:
+            print folder + " exists "
+    if(flag == 2):
+        if not os.path.exists(folder):
+            print folder + " not exist"
+            sys.exit()
+
 
 
 def select_products( products_list , products_df, user_footprint, threshold ) :
@@ -144,7 +155,7 @@ def select_products( products_list , products_df, user_footprint, threshold ) :
 def download_sentinel_mages(args):
     
             
-    check_existing_directory(args.output_folder)
+    check_existing_directory(args.output_folder,1)
     
     products = SentinelData()