1 # ============LICENSE_START===============================================
2 # Copyright (C) 2020 Nordix Foundation. All rights reserved.
3 # ========================================================================
4 # Licensed under the Apache License, Version 2.0 (the "License");
5 # you may not use this file except in compliance with the License.
6 # You may obtain a copy of the License at
8 # http://www.apache.org/licenses/LICENSE-2.0
10 # Unless required by applicable law or agreed to in writing, software
11 # distributed under the License is distributed on an "AS IS" BASIS,
12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 # See the License for the specific language governing permissions and
14 # limitations under the License.
15 # ============LICENSE_END=================================================
18 # This script create/update policies spread over a number rics
19 # Intended for parallel processing
20 # Returns a string with result, either "0" for ok, or "1<fault description>"
28 # disable warning about unverified https requests
29 from requests.packages import urllib3
31 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
33 #arg responsecode baseurl ric_base num_rics uuid startid templatepath count pids pid_id
38 if len(sys.argv) < 11:
39 print("1Expected 11/14 args, got "+str(len(sys.argv)-1))
42 responsecode=int(sys.argv[1])
43 baseurl=str(sys.argv[2])
44 ric_base=str(sys.argv[3])
45 num_rics=int(sys.argv[4])
47 start=int(sys.argv[6])
48 if ("/v2/" in baseurl):
49 if len(sys.argv) != 15:
50 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")
56 trans=str(sys.argv[9])
57 noti=str(sys.argv[10])
58 templatepath=str(sys.argv[11])
59 count=int(sys.argv[12])
60 pids=int(sys.argv[13])
61 pid_id=int(sys.argv[14])
63 if len(sys.argv) != 11:
64 print("1Expected 10 args, got "+str(len(sys.argv)-1)+ ". Args: responsecode baseurl ric_base num_rics uuid startid templatepath count pids pid_id")
68 templatepath=str(sys.argv[7])
69 count=int(sys.argv[8])
71 pid_id=int(sys.argv[10])
76 with open(templatepath, 'r') as file:
77 template = file.read()
80 stop=count*num_rics+start
82 for i in range(start,stop):
83 if (i%pids == (pid_id-1)):
84 payload=template.replace("XXX",str(i))
86 ric=ric_base+str(ric_id)
89 headers = {'Content-type': 'application/json'}
90 if ("/v2/" in baseurl):
95 data["policy_id"]=uuid+str(i)
96 data["service_id"]=serv
97 if (trans != "NOTRANSIENT"):
98 data["transient"]=trans
100 data["policy_type_id"]=pt
102 data["policy_type_id"]=""
103 if (noti != "NOURL"):
104 data["status_notification_uri"]=noti
105 data["policy_data"]=json.loads(payload)
108 data_out=json.dumps(data)
109 resp=requests.put(url, data_out, headers=headers, verify=False, timeout=90)
111 url=baseurl+"&id="+uuid+str(i)+"&ric="+str(ric)
113 data_out=json.dumps(json.loads(payload))
114 resp=requests.put(url, data_out, headers=headers, verify=False, timeout=90)
115 except Exception as e1:
116 print("1Put failed for id:"+uuid+str(i)+ ", "+str(e1) + " "+traceback.format_exc())
118 if (resp.status_code == None):
119 print("1Put failed for id:"+uuid+str(i)+ ", expected response code: "+str(responsecode)+", got: None")
121 if (resp.status_code != responsecode):
122 print("1Put failed for id:"+uuid+str(i)+ ", expected response code: "+str(responsecode)+", got: "+str(resp.status_code))
130 except Exception as e:
132 traceback.print_exc()