Only add WSSE headers on first request

[larsbj]

Dear all,

I have use case where I am supposed to add WSSE SOAP header elements in the first request only.

client = Client(wsdl, wsse=UsernameToken(....)
client.service.MyOp() #OK 
client.Service.MyOp2() # Rejected by server, since WSSE only expected in the first call.

The work around is to do this:

client = Client(wsdl, wsse=UsernameToken(....)
client.service.MyOp() #OK 
client.wsse = None
client.Service.MyOp2() #OK

I get a session identifier in return from the first call to MyOp(), and I use that in subsequent calls.

Is there a more elegant way (in a plugin, for example) to either set WSSE to None after the first service call, or something else?

Maybe subclass UsernameToken and add some state there and only set the headers once? It feels a bit like a hack, though :-)

Best regards,

Lars

[larsbj]

I ended up subclassing UsernameToken handler, with a state:

class MyUsernameToken(UsernameToken):
    def __init__(self,
                username: str,
                password: str,
                password_digest: bool=False,
                use_digest: bool=True,
                nonce: str=get_nonce(),
                created: datetime=datetime.now(timezone.utc),
                timestamp_token=None,
                zulu_timestamp=True,
                hash_password=True):
        super().__init__(username, password, password_digest, use_digest,nonce, created, timestamp_token, zulu_timestamp, hash_password)
        self.headers_applied = False

    def apply(self, envelope, headers):
        if not self.headers_applied:
            super().apply(envelope, headers)
            self.headers_applied = True
        return envelope, headers