Вернуть NULL при несовпадении - PullRequest
0 голосов
/ 01 марта 2019

Приведенный ниже код извлекает входные данные и разбивает входные данные со значением «HttpOnly», а затем, если выполняется условие «если», возвращает значение как таковое.

Как сделать это значениевернуть как NULL или "123", если условие не выполнено в самой split ()?

from soaptest.api import *
from com.parasoft.api import *

def getHeader(input, context):

    headerNew = ""
    strHeader = str(input).split("HttpOnly")
    for i in strHeader:
        if "com.abc.mb.SSO_GUID" in i:
            Application.showMessage(i)
            headerNew = i

    return headerNew

EDIT

Ввод - "abcdefgHttpOnly"

Выход - «abcdefg»

Вход - «abcdefg»

Выход - «123»

1 Ответ

0 голосов
/ 01 марта 2019

Вы можете просто проверить, является ли «HttpOnly» in первым, и вернуть «123».

def getHeader(input):
    if 'HttpOnly' not in str(input):
        return '123'

    headerNew = ""
    strHeader = str(input).split("HttpOnly")

    # Not using i as variable since it is usually used as an index
    for header in strHeader:
        if "com.abc.mb.SSO_GUID" in header:
            # Application.showMessage(header)
            headerNew = header

    return headerNew
print(getHeader('com.abc.mb.SSO_GUIDabcdefgHttpOnly')) # com.abc.mb.SSO_GUIDabcdefg
print(getHeader('com.abc.mb.SSO_GUIDabcdefg')) # 123
...