args_parse.py 15.8 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36

# * Licensed to the OpenAirInterface (OAI) Software Alliance under one or more
# * contributor license agreements.  See the NOTICE file distributed with
# * this work for additional information regarding copyright ownership.
# * The OpenAirInterface Software Alliance licenses this file to You under
# * the OAI Public License, Version 1.1  (the "License"); you may not use this file
# * except in compliance with the License.
# * You may obtain a copy of the License at
# *
# *      http://www.openairinterface.org/?page_id=698
# *
# * Unless required by applicable law or agreed to in writing, software
# * distributed under the License is distributed on an "AS IS" BASIS,
# * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# * See the License for the specific language governing permissions and
# * limitations under the License.
# *-------------------------------------------------------------------------------
# * For more information about the OpenAirInterface (OAI) Software Alliance:
# *      contact@openairinterface.org
# */
#---------------------------------------------------------------------
# Python for CI of OAI-eNB + COTS-UE
#
#   Required Python Version
#     Python 3.x
#
#   Required Python Package
#     pexpect
#---------------------------------------------------------------------

#-----------------------------------------------------------
# Import Libs
#-----------------------------------------------------------
import sys		# arg
import re		# reg
import yaml
37
import constants as CONST
38 39 40 41 42 43

#-----------------------------------------------------------
# Parsing Command Line Arguements
#-----------------------------------------------------------


44
def ArgsParse(argvs,CiTestObj,RAN,HTML,EPC,ldpc,CONTAINERS,HELP,SCA,PHYSIM,CLUSTER):
45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81


    py_param_file_present = False
    py_params={}

    while len(argvs) > 1:
        myArgv = argvs.pop(1)	# 0th is this file's name

	    #--help
        if re.match('^\-\-help$', myArgv, re.IGNORECASE):
            HELP.GenericHelp(CONST.Version)
            sys.exit(0)

	    #--apply=<filename> as parameters file, to replace inline parameters
        elif re.match('^\-\-Apply=(.+)$', myArgv, re.IGNORECASE):
            matchReg = re.match('^\-\-Apply=(.+)$', myArgv, re.IGNORECASE)
            py_params_file = matchReg.group(1)
            with open(py_params_file,'r') as file:
          	# The FullLoader parameter handles the conversion from YAML
        	# scalar values to Python dictionary format
                py_params = yaml.load(file,Loader=yaml.FullLoader)
                py_param_file_present = True #to be removed once validated
	    		#AssignParams(py_params) #to be uncommented once validated

	    #consider inline parameters
        elif re.match('^\-\-mode=(.+)$', myArgv, re.IGNORECASE):
            matchReg = re.match('^\-\-mode=(.+)$', myArgv, re.IGNORECASE)
            mode = matchReg.group(1)
        elif re.match('^\-\-eNBRepository=(.+)$|^\-\-ranRepository(.+)$', myArgv, re.IGNORECASE):
            if re.match('^\-\-eNBRepository=(.+)$', myArgv, re.IGNORECASE):
                matchReg = re.match('^\-\-eNBRepository=(.+)$', myArgv, re.IGNORECASE)
            else:
                matchReg = re.match('^\-\-ranRepository=(.+)$', myArgv, re.IGNORECASE)
            CiTestObj.ranRepository = matchReg.group(1)
            RAN.ranRepository=matchReg.group(1)
            HTML.ranRepository=matchReg.group(1)
            ldpc.ranRepository=matchReg.group(1)
82
            CONTAINERS.ranRepository=matchReg.group(1)
83
            SCA.ranRepository=matchReg.group(1)
84
            PHYSIM.ranRepository=matchReg.group(1)
85
            CLUSTER.ranRepository=matchReg.group(1)
86 87 88 89 90 91 92 93 94 95 96
        elif re.match('^\-\-eNB_AllowMerge=(.+)$|^\-\-ranAllowMerge=(.+)$', myArgv, re.IGNORECASE):
            if re.match('^\-\-eNB_AllowMerge=(.+)$', myArgv, re.IGNORECASE):
                matchReg = re.match('^\-\-eNB_AllowMerge=(.+)$', myArgv, re.IGNORECASE)
            else:
                matchReg = re.match('^\-\-ranAllowMerge=(.+)$', myArgv, re.IGNORECASE)
            doMerge = matchReg.group(1)
            ldpc.ranAllowMerge=matchReg.group(1)
            if ((doMerge == 'true') or (doMerge == 'True')):
                CiTestObj.ranAllowMerge = True
                RAN.ranAllowMerge=True
                HTML.ranAllowMerge=True
97
                CONTAINERS.ranAllowMerge=True
98
                SCA.ranAllowMerge=True
99
                PHYSIM.ranAllowMerge=True
100
                CLUSTER.ranAllowMerge=True
101 102 103 104 105 106 107 108 109
        elif re.match('^\-\-eNBBranch=(.+)$|^\-\-ranBranch=(.+)$', myArgv, re.IGNORECASE):
            if re.match('^\-\-eNBBranch=(.+)$', myArgv, re.IGNORECASE):
                matchReg = re.match('^\-\-eNBBranch=(.+)$', myArgv, re.IGNORECASE)
            else:
                matchReg = re.match('^\-\-ranBranch=(.+)$', myArgv, re.IGNORECASE)
            CiTestObj.ranBranch = matchReg.group(1)
            RAN.ranBranch=matchReg.group(1)
            HTML.ranBranch=matchReg.group(1)
            ldpc.ranBranch=matchReg.group(1)
110
            CONTAINERS.ranBranch=matchReg.group(1)
111
            SCA.ranBranch=matchReg.group(1)
112
            PHYSIM.ranBranch=matchReg.group(1)
113
            CLUSTER.ranBranch=matchReg.group(1)
114 115 116 117 118 119 120 121 122
        elif re.match('^\-\-eNBCommitID=(.*)$|^\-\-ranCommitID=(.*)$', myArgv, re.IGNORECASE):
            if re.match('^\-\-eNBCommitID=(.*)$', myArgv, re.IGNORECASE):
                matchReg = re.match('^\-\-eNBCommitID=(.*)$', myArgv, re.IGNORECASE)
            else:
                matchReg = re.match('^\-\-ranCommitID=(.*)$', myArgv, re.IGNORECASE)
            CiTestObj.ranCommitID = matchReg.group(1)
            RAN.ranCommitID=matchReg.group(1)
            HTML.ranCommitID=matchReg.group(1)
            ldpc.ranCommitID=matchReg.group(1)
123
            CONTAINERS.ranCommitID=matchReg.group(1)
124
            SCA.ranCommitID=matchReg.group(1)
125
            PHYSIM.ranCommitID=matchReg.group(1)
126
            CLUSTER.ranCommitID=matchReg.group(1)
127 128 129 130 131 132 133 134 135
        elif re.match('^\-\-eNBTargetBranch=(.*)$|^\-\-ranTargetBranch=(.*)$', myArgv, re.IGNORECASE):
            if re.match('^\-\-eNBTargetBranch=(.*)$', myArgv, re.IGNORECASE):
                matchReg = re.match('^\-\-eNBTargetBranch=(.*)$', myArgv, re.IGNORECASE)
            else:
                matchReg = re.match('^\-\-ranTargetBranch=(.*)$', myArgv, re.IGNORECASE)
            CiTestObj.ranTargetBranch = matchReg.group(1)
            RAN.ranTargetBranch=matchReg.group(1)
            HTML.ranTargetBranch=matchReg.group(1)
            ldpc.ranTargetBranch=matchReg.group(1)
136
            CONTAINERS.ranTargetBranch=matchReg.group(1)
137
            SCA.ranTargetBranch=matchReg.group(1)
138
            PHYSIM.ranTargetBranch=matchReg.group(1)
139
            CLUSTER.ranTargetBranch=matchReg.group(1)
140 141 142 143 144
        elif re.match('^\-\-eNBIPAddress=(.+)$|^\-\-eNB[1-2]IPAddress=(.+)$', myArgv, re.IGNORECASE):
            if re.match('^\-\-eNBIPAddress=(.+)$', myArgv, re.IGNORECASE):
                matchReg = re.match('^\-\-eNBIPAddress=(.+)$', myArgv, re.IGNORECASE)
                RAN.eNBIPAddress=matchReg.group(1)
                ldpc.eNBIpAddr=matchReg.group(1)
145
                CONTAINERS.eNBIPAddress=matchReg.group(1)
146
                SCA.eNBIPAddress=matchReg.group(1)
147
                PHYSIM.eNBIPAddress=matchReg.group(1)
148
                CLUSTER.eNBIPAddress=matchReg.group(1)
149 150 151
            elif re.match('^\-\-eNB1IPAddress=(.+)$', myArgv, re.IGNORECASE):
                matchReg = re.match('^\-\-eNB1IPAddress=(.+)$', myArgv, re.IGNORECASE)
                RAN.eNB1IPAddress=matchReg.group(1)
152
                CONTAINERS.eNB1IPAddress=matchReg.group(1)
153 154 155
            elif re.match('^\-\-eNB2IPAddress=(.+)$', myArgv, re.IGNORECASE):
                matchReg = re.match('^\-\-eNB2IPAddress=(.+)$', myArgv, re.IGNORECASE)
                RAN.eNB2IPAddress=matchReg.group(1)
156
                CONTAINERS.eNB2IPAddress=matchReg.group(1)
157 158 159 160 161
        elif re.match('^\-\-eNBUserName=(.+)$|^\-\-eNB[1-2]UserName=(.+)$', myArgv, re.IGNORECASE):
            if re.match('^\-\-eNBUserName=(.+)$', myArgv, re.IGNORECASE):
                matchReg = re.match('^\-\-eNBUserName=(.+)$', myArgv, re.IGNORECASE)
                RAN.eNBUserName=matchReg.group(1)
                ldpc.eNBUserName=matchReg.group(1)
162
                CONTAINERS.eNBUserName=matchReg.group(1)
163
                SCA.eNBUserName=matchReg.group(1)
164
                PHYSIM.eNBUserName=matchReg.group(1)
165
                CLUSTER.eNBUserName=matchReg.group(1)
166 167 168
            elif re.match('^\-\-eNB1UserName=(.+)$', myArgv, re.IGNORECASE):
                matchReg = re.match('^\-\-eNB1UserName=(.+)$', myArgv, re.IGNORECASE)
                RAN.eNB1UserName=matchReg.group(1)
169
                CONTAINERS.eNB1UserName=matchReg.group(1)
170 171 172
            elif re.match('^\-\-eNB2UserName=(.+)$', myArgv, re.IGNORECASE):
                matchReg = re.match('^\-\-eNB2UserName=(.+)$', myArgv, re.IGNORECASE)
                RAN.eNB2UserName=matchReg.group(1)
173
                CONTAINERS.eNB2UserName=matchReg.group(1)
174 175 176 177 178
        elif re.match('^\-\-eNBPassword=(.+)$|^\-\-eNB[1-2]Password=(.+)$', myArgv, re.IGNORECASE):
            if re.match('^\-\-eNBPassword=(.+)$', myArgv, re.IGNORECASE):
                matchReg = re.match('^\-\-eNBPassword=(.+)$', myArgv, re.IGNORECASE)
                RAN.eNBPassword=matchReg.group(1)
                ldpc.eNBPassWord=matchReg.group(1)
179
                CONTAINERS.eNBPassword=matchReg.group(1)
180
                SCA.eNBPassword=matchReg.group(1)
181
                PHYSIM.eNBPassword=matchReg.group(1)
182
                CLUSTER.eNBPassword=matchReg.group(1)
183 184 185
            elif re.match('^\-\-eNB1Password=(.+)$', myArgv, re.IGNORECASE):
                matchReg = re.match('^\-\-eNB1Password=(.+)$', myArgv, re.IGNORECASE)
                RAN.eNB1Password=matchReg.group(1)
186
                CONTAINERS.eNB1Password=matchReg.group(1)
187 188 189
            elif re.match('^\-\-eNB2Password=(.+)$', myArgv, re.IGNORECASE):
                matchReg = re.match('^\-\-eNB2Password=(.+)$', myArgv, re.IGNORECASE)
                RAN.eNB2Password=matchReg.group(1)
190
                CONTAINERS.eNB2Password=matchReg.group(1)
191 192 193 194 195
        elif re.match('^\-\-eNBSourceCodePath=(.+)$|^\-\-eNB[1-2]SourceCodePath=(.+)$', myArgv, re.IGNORECASE):
            if re.match('^\-\-eNBSourceCodePath=(.+)$', myArgv, re.IGNORECASE):
                matchReg = re.match('^\-\-eNBSourceCodePath=(.+)$', myArgv, re.IGNORECASE)
                RAN.eNBSourceCodePath=matchReg.group(1)
                ldpc.eNBSourceCodePath=matchReg.group(1)
196
                CONTAINERS.eNBSourceCodePath=matchReg.group(1)
197
                SCA.eNBSourceCodePath=matchReg.group(1)
198
                PHYSIM.eNBSourceCodePath=matchReg.group(1)
199
                CLUSTER.eNBSourceCodePath=matchReg.group(1)
200
                EPC.eNBSourceCodePath=matchReg.group(1)
201 202 203
            elif re.match('^\-\-eNB1SourceCodePath=(.+)$', myArgv, re.IGNORECASE):
                matchReg = re.match('^\-\-eNB1SourceCodePath=(.+)$', myArgv, re.IGNORECASE)
                RAN.eNB1SourceCodePath=matchReg.group(1)
204
                CONTAINERS.eNB1SourceCodePath=matchReg.group(1)
205 206 207
            elif re.match('^\-\-eNB2SourceCodePath=(.+)$', myArgv, re.IGNORECASE):
                matchReg = re.match('^\-\-eNB2SourceCodePath=(.+)$', myArgv, re.IGNORECASE)
                RAN.eNB2SourceCodePath=matchReg.group(1)
208
                CONTAINERS.eNB2SourceCodePath=matchReg.group(1)
209 210 211 212 213 214 215 216 217 218 219 220 221 222
        elif re.match('^\-\-EPCIPAddress=(.+)$', myArgv, re.IGNORECASE):
            matchReg = re.match('^\-\-EPCIPAddress=(.+)$', myArgv, re.IGNORECASE)
            EPC.IPAddress=matchReg.group(1)
        elif re.match('^\-\-EPCUserName=(.+)$', myArgv, re.IGNORECASE):
            matchReg = re.match('^\-\-EPCUserName=(.+)$', myArgv, re.IGNORECASE)
            EPC.UserName=matchReg.group(1)
        elif re.match('^\-\-EPCPassword=(.+)$', myArgv, re.IGNORECASE):
            matchReg = re.match('^\-\-EPCPassword=(.+)$', myArgv, re.IGNORECASE)
            EPC.Password=matchReg.group(1)
        elif re.match('^\-\-EPCSourceCodePath=(.+)$', myArgv, re.IGNORECASE):
            matchReg = re.match('^\-\-EPCSourceCodePath=(.+)$', myArgv, re.IGNORECASE)
            EPC.SourceCodePath=matchReg.group(1)
        elif re.match('^\-\-EPCType=(.+)$', myArgv, re.IGNORECASE):
            matchReg = re.match('^\-\-EPCType=(.+)$', myArgv, re.IGNORECASE)
223
            if re.match('OAI', matchReg.group(1), re.IGNORECASE) or re.match('ltebox', matchReg.group(1), re.IGNORECASE) or re.match('OAI-Rel14-Docker', matchReg.group(1), re.IGNORECASE) or re.match('OC-OAI-CN5G', matchReg.group(1), re.IGNORECASE):
224 225
                EPC.Type=matchReg.group(1)
            else:
226
                sys.exit('Invalid EPC Type: ' + matchReg.group(1) + ' -- (should be OAI or ltebox or OAI-Rel14-Docker or OC-OAI-CN5G)')
227 228 229 230 231 232
        elif re.match('^\-\-EPCContainerPrefix=(.+)$', myArgv, re.IGNORECASE):
            matchReg = re.match('^\-\-EPCContainerPrefix=(.+)$', myArgv, re.IGNORECASE)
            EPC.ContainerPrefix=matchReg.group(1)
        elif re.match('^\-\-XMLTestFile=(.+)$', myArgv, re.IGNORECASE):
            matchReg = re.match('^\-\-XMLTestFile=(.+)$', myArgv, re.IGNORECASE)
            CiTestObj.testXMLfiles.append(matchReg.group(1))
Raphael Defosseux's avatar
Raphael Defosseux committed
233
            HTML.testXMLfiles.append(matchReg.group(1))
234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251
            HTML.nbTestXMLfiles=HTML.nbTestXMLfiles+1
        elif re.match('^\-\-UEIPAddress=(.+)$', myArgv, re.IGNORECASE):
            matchReg = re.match('^\-\-UEIPAddress=(.+)$', myArgv, re.IGNORECASE)
            CiTestObj.UEIPAddress = matchReg.group(1)
        elif re.match('^\-\-UEUserName=(.+)$', myArgv, re.IGNORECASE):
            matchReg = re.match('^\-\-UEUserName=(.+)$', myArgv, re.IGNORECASE)
            CiTestObj.UEUserName = matchReg.group(1)
        elif re.match('^\-\-UEPassword=(.+)$', myArgv, re.IGNORECASE):
            matchReg = re.match('^\-\-UEPassword=(.+)$', myArgv, re.IGNORECASE)
            CiTestObj.UEPassword = matchReg.group(1)
        elif re.match('^\-\-UESourceCodePath=(.+)$', myArgv, re.IGNORECASE):
            matchReg = re.match('^\-\-UESourceCodePath=(.+)$', myArgv, re.IGNORECASE)
            CiTestObj.UESourceCodePath = matchReg.group(1)
        elif re.match('^\-\-finalStatus=(.+)$', myArgv, re.IGNORECASE):
            matchReg = re.match('^\-\-finalStatus=(.+)$', myArgv, re.IGNORECASE)
            finalStatus = matchReg.group(1)
            if ((finalStatus == 'true') or (finalStatus == 'True')):
                CiTestObj.finalStatus = True
252 253 254
        elif re.match('^\-\-OCUserName=(.+)$', myArgv, re.IGNORECASE):
            matchReg = re.match('^\-\-OCUserName=(.+)$', myArgv, re.IGNORECASE)
            PHYSIM.OCUserName = matchReg.group(1)
255
            CLUSTER.OCUserName = matchReg.group(1)
256
            EPC.OCUserName = matchReg.group(1)
257 258
        elif re.match('^\-\-OCPassword=(.+)$', myArgv, re.IGNORECASE):
            matchReg = re.match('^\-\-OCPassword=(.+)$', myArgv, re.IGNORECASE)
Mohammed Ismail's avatar
Mohammed Ismail committed
259
            PHYSIM.OCPassword = matchReg.group(1)
260
            CLUSTER.OCPassword = matchReg.group(1)
261
            EPC.OCPassword = matchReg.group(1)
262 263 264
        elif re.match('^\-\-OCProjectName=(.+)$', myArgv, re.IGNORECASE):
            matchReg = re.match('^\-\-OCProjectName=(.+)$', myArgv, re.IGNORECASE)
            PHYSIM.OCProjectName = matchReg.group(1)
265
            CLUSTER.OCProjectName = matchReg.group(1)
266
            EPC.OCProjectName = matchReg.group(1)
267 268 269 270 271 272
        elif re.match('^\-\-OCUrl=(.+)$', myArgv, re.IGNORECASE):
            matchReg = re.match('^\-\-OCUrl=(.+)$', myArgv, re.IGNORECASE)
            CLUSTER.OCUrl = matchReg.group(1)
        elif re.match('^\-\-OCRegistry=(.+)$', myArgv, re.IGNORECASE):
            matchReg = re.match('^\-\-OCRegistry=(.+)$', myArgv, re.IGNORECASE)
            CLUSTER.OCRegistry = matchReg.group(1)
273 274 275
        elif re.match('^\-\-BuildId=(.+)$', myArgv, re.IGNORECASE):
            matchReg = re.match('^\-\-BuildId=(.+)$', myArgv, re.IGNORECASE)
            RAN.BuildId = matchReg.group(1)
276 277 278 279
        else:
            HELP.GenericHelp(CONST.Version)
            sys.exit('Invalid Parameter: ' + myArgv)

280
    return py_param_file_present, py_params, mode