Using an if statement in function argument when calling

Question

How to use an if statement or similar in function arguments to reduce blocks of code?

    if versionid is None:
        s3_response = s3_client.head_object(
              Bucket=bucket
            , Key=key
            )
    else:
        s3_response = s3_client.head_object(
              Bucket=bucket
            , Key=key
            , VersionId=versionid
            )

to something like this:

        s3_response = s3_client.head_object(
              Bucket=bucket
            , Key=key
            , if versionid is not None: VersionId=versionid
            )

Not able to get this working and can't find any examples so assume not possible.

You can use the conditional operator to choose different values of the argument based on a condition, but not whether to provide the argument or not. — Barmar
– Barmar, Commented Jun 24, 2022 at 14:41

chepner · Accepted Answer · 2022-06-24 14:46:02Z

3

You can't use a statement where an expression is expected. Unless you know what the default value (if any) for the VersionId parameter is, you can do something like this:

args = {'Bucket': bucket, 'Key': key}
if versionid is not None:
    args['VersionId'] = versionid

s3_response = s3_client.head_object(**args)

If you do know the appropriate default, you can use the conditional expression:

# FOO is whatever the appropriate default is
s3_response = s3_client.head_object(Bucket=bucket, Key=key, Versionid=FOO if versonid is None else versionid)

(Assuming a non-None versionid will be a truthy value, you can shorten this to versionid or FOO, but be aware that 0 or FOO is 0, not FOO.)

You could write something like

s3_response = s3_client.head_object(
                Bucket=bucket,
                Key=key,
                **({} if versionid is None else {'VersionID': versionid})
              )

but I wouldn't recommend it.

edited Jun 24, 2022 at 14:46

answered Jun 24, 2022 at 14:39

chepner

538k77 gold badges594 silver badges746 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Paul Over a year ago

The first example is perfect here because it requires a string if sending, so None will not work, and I have no default. This is all super useful knowledge, cheers.

chepner Over a year ago

Yeah, if using a ** parameter, then the only distinction to make may be whether the keyword argument is present or not, not whether it has one value or another.

Collectives™ on Stack Overflow

Using an if statement in function argument when calling

1 Answer 1

2 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related