Melhorar tratamento de tipo de produto

aws_parser.py

@@ -2,17 +2,7 @@ import urllib.request
 import json
 import json
 import sys
 import sys
 
 
-if len(sys.argv) == 1:
-    print('You need to specify a product type', end='\n\n')
-    print('Supported product types are:')
-    print('- compute')
-    print('- database')
-    exit(1)
-
-PRODUCT_TYPE = sys.argv[1]
-if (PRODUCT_TYPE != 'compute') and (PRODUCT_TYPE != 'database'):
-    print('Product type needs to be either computer or database!')
-    exit(1)
+PRODUCT_TYPE = sys.argv[1] if len(sys.argv) == 2 else 'none'
 
 
 TERM_ONDEMAND = 'JRTCKXETXF' #Ondemand. TODO: Find out where to get this from
 TERM_ONDEMAND = 'JRTCKXETXF' #Ondemand. TODO: Find out where to get this from
 HOURLY_DIMENSION = '6YS6EN2CT7' #Tier1. TODO: Find out where to get this from
 HOURLY_DIMENSION = '6YS6EN2CT7' #Tier1. TODO: Find out where to get this from
@@ -45,6 +35,9 @@ elif PRODUCT_TYPE == 'database':
     FCATALOG = 'examples/aws_rds.json'
     FCATALOG = 'examples/aws_rds.json'
     CATALOG = 'https://pricing.us-east-1.amazonaws.com/offers/v1.0/aws/AmazonRDS/current/index.json'
     CATALOG = 'https://pricing.us-east-1.amazonaws.com/offers/v1.0/aws/AmazonRDS/current/index.json'
     PRODUCT_FAMILY = 'Database Instance'
     PRODUCT_FAMILY = 'Database Instance'
+else:
+    print("Product type not supported")
+    sys.exit(1)
 
 
 with urllib.request.urlopen(CATALOG) as url:
 with urllib.request.urlopen(CATALOG) as url:
     DATA = json.loads(url.read().decode())
     DATA = json.loads(url.read().decode())