import sys
import requests
import traceback
+from time import sleep
# disable warning about unverified https requests
from requests.packages import urllib3
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
-#arg responsecode baseurl ric_base num_rics uuid startid templatepath count pids pid_id
+#arg responsecode baseurl ric_base num_rics uuid startid templatepath count pids pid_id proxy
data_out=""
url_out=""
try:
- if len(sys.argv) < 11:
- print("1Expected 11/14 args, got "+str(len(sys.argv)-1))
+ if len(sys.argv) < 12:
+ print("1Expected 12/15 args, got "+str(len(sys.argv)-1))
print (sys.argv[1:])
sys.exit()
responsecode=int(sys.argv[1])
num_rics=int(sys.argv[4])
uuid=str(sys.argv[5])
start=int(sys.argv[6])
+ httpproxy="NOPROXY"
if ("/v2/" in baseurl):
- if len(sys.argv) != 15:
- print("1Expected 14 args, got "+str(len(sys.argv)-1)+ ". Args: responsecode baseurl ric_base num_rics uuid startid service type transient notification-url templatepath count pids pid_id")
+ if len(sys.argv) != 16:
+ print("1Expected 15 args, got "+str(len(sys.argv)-1)+ ". Args: responsecode baseurl ric_base num_rics uuid startid service type transient notification-url templatepath count pids pid_id proxy")
print (sys.argv[1:])
sys.exit()
count=int(sys.argv[12])
pids=int(sys.argv[13])
pid_id=int(sys.argv[14])
+ httpproxy=str(sys.argv[15])
else:
- if len(sys.argv) != 11:
- print("1Expected 10 args, got "+str(len(sys.argv)-1)+ ". Args: responsecode baseurl ric_base num_rics uuid startid templatepath count pids pid_id")
+ if len(sys.argv) != 12:
+ print("1Expected 11 args, got "+str(len(sys.argv)-1)+ ". Args: responsecode baseurl ric_base num_rics uuid startid templatepath count pids pid_id proxy")
print (sys.argv[1:])
sys.exit()
count=int(sys.argv[8])
pids=int(sys.argv[9])
pid_id=int(sys.argv[10])
-
+ httpproxy=str(sys.argv[11])
+
+ proxydict=None
+ if httpproxy != "NOPROXY":
+ proxydict = {
+ "http" : httpproxy,
+ "https" : httpproxy
+ }
if uuid == "NOUUID":
uuid=""
start=start
stop=count*num_rics+start
+ http_retry_count=0
+ connect_retry_count=0
+
for i in range(start,stop):
if (i%pids == (pid_id-1)):
payload=template.replace("XXX",str(i))
ric_id=(i%num_rics)+1
ric=ric_base+str(ric_id)
- try:
- headers = {'Content-type': 'application/json'}
- if ("/v2/" in baseurl):
- url=baseurl
-
- data={}
- data["ric_id"]=ric
- data["policy_id"]=uuid+str(i)
- data["service_id"]=serv
- if (trans != "NOTRANSIENT"):
- data["transient"]=trans
- if (pt != "NOTYPE"):
- data["policy_type_id"]=pt
+ connect_ok=False
+ retry_cnt=5
+ while(retry_cnt>0):
+ try:
+ headers = {'Content-type': 'application/json'}
+ if ("/v2/" in baseurl):
+ url=baseurl
+
+ data={}
+ data["ric_id"]=ric
+ data["policy_id"]=uuid+str(i)
+ data["service_id"]=serv
+ if (trans != "NOTRANSIENT"):
+ data["transient"]=trans
+ if (pt != "NOTYPE"):
+ data["policytype_id"]=pt
+ else:
+ data["policytype_id"]=""
+ if (noti != "NOURL"):
+ data["status_notification_uri"]=noti
+ data["policy_data"]=json.loads(payload)
+
+ url_out=url
+ data_out=json.dumps(data)
+ else:
+ url=baseurl+"&id="+uuid+str(i)+"&ric="+str(ric)
+ url_out=url
+ data_out=json.dumps(json.loads(payload))
+ if proxydict is None:
+ resp=requests.put(url, data_out, headers=headers, verify=False, timeout=90)
else:
- data["policy_type_id"]=""
- if (noti != "NOURL"):
- data["status_notification_uri"]=noti
- data["policy_data"]=json.loads(payload)
-
- url_out=url
- data_out=json.dumps(data)
- resp=requests.put(url, data_out, headers=headers, verify=False, timeout=90)
- else:
- url=baseurl+"&id="+uuid+str(i)+"&ric="+str(ric)
- url_out=url
- data_out=json.dumps(json.loads(payload))
- resp=requests.put(url, data_out, headers=headers, verify=False, timeout=90)
- except Exception as e1:
- print("1Put failed for id:"+uuid+str(i)+ ", "+str(e1) + " "+traceback.format_exc())
- sys.exit()
- if (resp.status_code == None):
- print("1Put failed for id:"+uuid+str(i)+ ", expected response code: "+str(responsecode)+", got: None")
- sys.exit()
- if (resp.status_code != responsecode):
- print("1Put failed for id:"+uuid+str(i)+ ", expected response code: "+str(responsecode)+", got: "+str(resp.status_code))
- print(url_out)
- print(str(data_out))
- sys.exit()
-
- print("0")
+ resp=requests.put(url, data_out, headers=headers, verify=False, timeout=90, proxies=proxydict)
+ connect_ok=True
+ except Exception as e1:
+ if (retry_cnt > 1):
+ sleep(0.1)
+ retry_cnt -= 1
+ connect_retry_count += 1
+ else:
+ print("1Put failed for id:"+uuid+str(i)+ ", "+str(e1) + " "+traceback.format_exc())
+ sys.exit()
+
+ if (connect_ok == True):
+ if (resp.status_code == None):
+ print("1Put failed for id:"+uuid+str(i)+ ", expected response code: "+str(responsecode)+", got: None")
+ sys.exit()
+
+ if (resp.status_code != responsecode):
+ if (resp.status_code >= 500) and (http_retry_count < 600 ) and (retry_cnt > 1):
+ sleep(0.1)
+ retry_cnt -= 1
+ http_retry_count += 1
+ else:
+ print("1Put failed for id:"+uuid+str(i)+ ", expected response code: "+str(responsecode)+", got: "+str(resp.status_code))
+ print(url_out)
+ print(str(data_out))
+ sys.exit()
+ else:
+ retry_cnt=-1
+
+ print("0 http retries:"+str(http_retry_count) + ", connect retries: "+str(connect_retry_count))
sys.exit()
except Exception as e: