Control Structures¶
Functions¶
Functions are executable units of code within a contract. Functions may only be declared within a contract’s module scope.
@external
def bid():
...
Functions may be called internally or externally depending on their visibility. Functions may accept input arguments and return variables in order to pass values between them.
Visibility¶
You can optionally declare a function’s visibility by using a decorator. There are three visibility levels in Vyper:
@external
: exposed in the selector table, can be called by an external call into this contract
@internal
(default): can be invoked only from within this contract. Not available to external callers
@deploy
: constructor code. This is code which is invoked once in the lifetime of a contract, upon its deploy. It is not available at runtime to either external callers or internal call invocations. At this time, only the __init__() function may be marked as@deploy
.
External Functions¶
External functions (marked with the @external
decorator) are a part of the contract interface and may only be called via transactions or from other contracts.
@external
def add_seven(a: int128) -> int128:
return a + 7
@external
def add_seven_with_overloading(a: uint256, b: uint256 = 3):
return a + b
A Vyper contract cannot call directly between two external functions. If you must do this, you can use an interface.
Note
For external functions with default arguments like def my_function(x: uint256, b: uint256 = 1)
the Vyper compiler will generate N+1
overloaded function selectors based on N
default arguments.
Internal Functions¶
Internal functions (optionally marked with the @internal
decorator) are only accessible from other functions within the same contract. They are invoked via the self object:
def _times_two(amount: uint256) -> uint256:
return amount * 2
@external
def calculate(amount: uint256) -> uint256:
return self._times_two(amount)
Or for internal functions which are defined in imported modules, they are invoked by prefixing the name of the module to the function name:
Note
As of v0.4.0, the @internal
decorator is optional. That is, functions with no visibility decorator default to being internal
.
Note
Please note that for internal
functions which use more than one default parameter, Vyper versions >=0.3.8
are recommended due to the security advisory GHSA-ph9x-4vc9-m39g.
The __init__
Function¶
The __init__()
function, also known as the constructor, is a special initialization function that is only called at the time of deploying a contract. It can be used to set initial values for storage or immutable variables. It must be declared with the @deploy
decorator. A common use case is to set an owner
variable with the creator of the contract:
owner: address
@deploy
def __init__():
self.owner = msg.sender
Additionally, immutable variables may only be set within the constructor.
Mutability¶
You can optionally declare a function’s mutability by using a decorator. There are four mutability levels:
@pure
: does not read from the contract state or any environment variables.
@view
: may read from the contract state, but does not alter it.
@nonpayable
(default): may read from and write to the contract state, but cannot receive Ether.
@payable
: may read from and write to the contract state, and can receive Ether.
@view
@external
def readonly():
# this function cannot write to state
...
@payable
@external
def send_me_money():
# this function can receive ether
...
Functions default to nonpayable
when no mutability decorator is used.
Functions marked with @view
cannot call mutable (payable
or nonpayable
) functions. Any external calls are made using the special STATICCALL
opcode, which prevents state changes at the EVM level.
Functions marked with @pure
cannot call non-pure
functions.
Re-entrancy Locks¶
The @nonreentrant
decorator places a global nonreentrancy lock on a function. An attempt by an external contract to call back into any other @nonreentrant
function causes the transaction to revert.
@external
@nonreentrant
def make_a_call(_addr: address):
# this function is protected from re-entrancy
...
You can put the @nonreentrant
decorator on a __default__
function but we recommend against it because in most circumstances it will not work in a meaningful way.
Nonreentrancy locks work by setting a specially allocated storage slot to a <locked>
value on function entrance, and setting it to an <unlocked>
value on function exit. On function entrance, if the storage slot is detected to be the <locked>
value, execution reverts.
You cannot put the @nonreentrant
decorator on a pure
function. You can put it on a view
function, but it only checks that the function is not in a callback (the storage slot is not in the <locked>
state), as view
functions can only read the state, not change it.
You can view where the nonreentrant key is physically laid out in storage by using vyper
with the -f layout
option (e.g., vyper -f layout foo.vy
). Unless it is overridden, the compiler will allocate it at slot 0
.
Note
A mutable function can protect a view
function from being called back into (which is useful for instance, if a view
function would return inconsistent state during a mutable function), but a view
function cannot protect itself from being called back into. Note that mutable functions can never be called from a view
function because all external calls out from a view
function are protected by the use of the STATICCALL
opcode.
Note
A nonreentrant lock has an <unlocked>
value of 3, and a <locked>
value of 2. Nonzero values are used to take advantage of net gas metering - as of the Berlin hard fork, the net cost for utilizing a nonreentrant lock is 2300 gas. Prior to v0.3.4, the <unlocked>
and <locked>
values were 0 and 1, respectively.
Note
Prior to 0.4.0, nonreentrancy keys took a “key” argument for fine-grained nonreentrancy control. As of 0.4.0, only a global nonreentrancy lock is available.
The __default__
Function¶
A contract can also have a default function, which is executed on a call to the contract if no other functions match the given function identifier (or if none was supplied at all, such as through someone sending it Eth). It is the same construct as fallback functions in Solidity.
This function is always named __default__
. It must be annotated with @external
. It cannot expect any input arguments.
If the function is annotated as @payable
, this function is executed whenever the contract is sent Ether (without data). This is why the default function cannot accept arguments - it is a design decision of Ethereum to make no differentiation between sending ether to a contract or a user address.
event Payment:
amount: uint256
sender: indexed(address)
@external
@payable
def __default__():
log Payment(msg.value, msg.sender)
Considerations¶
Just as in Solidity, Vyper generates a default function if one isn’t found, in the form of a REVERT
call. Note that this rolls back state changes, and thus will not succeed in receiving funds.
Ethereum specifies that the operations will be rolled back if the contract runs out of gas in execution. send
calls to the contract come with a free stipend of 2300 gas, which does not leave much room to perform other operations except basic logging. However, if the sender includes a higher gas amount through a call
instead of send
, then more complex functionality can be run.
It is considered a best practice to ensure your payable default function is compatible with this stipend. The following operations will consume more than 2300 gas:
Writing to storage
Creating a contract
Calling an external function which consumes a large amount of gas
Sending Ether
Lastly, although the default function receives no arguments, it can still access the msg
object, including:
the address of who is interacting with the contract (
msg.sender
)the amount of ETH sent (
msg.value
)the gas provided (
msg.gas
).
Decorators Reference¶
Decorator |
Description |
---|---|
|
Function can only be called externally, it is part of the runtime selector table |
|
Function can only be called within current contract |
|
Function is called only at deploy time |
|
Function does not read contract state or environment variables |
|
Function does not alter contract state |
|
Function is able to receive Ether |
|
Function cannot be called back into during an external call |
if
statements¶
The if
statement is a control flow construct used for conditional execution:
if CONDITION:
...
CONDITION
is a boolean or boolean operation. The boolean is evaluated left-to-right, one expression at a time, until the condition is found to be true or false. If true, the logic in the body of the if
statement is executed.
Note that unlike Python, Vyper does not allow implicit conversion from non-boolean types within the condition of an if
statement. if 1: pass
will fail to compile with a type mismatch.
You can also include elif
and else
statements, to add more conditional statements and a body that executes when the conditionals are false:
if CONDITION:
...
elif OTHER_CONDITION:
...
else:
...
for
loops¶
The for
statement is a control flow construct used to iterate over a value:
for i: <TYPE> in <ITERABLE>:
...
The iterated value can be a static array, a dynamic array, or generated from the built-in range
function.
Array Iteration¶
You can use for
to iterate through the values of any array variable:
foo: int128[3] = [4, 23, 42]
for i: int128 in foo:
...
In the above, example, the loop executes three times with i
assigned the values of 4
, 23
, and then 42
.
You can also iterate over a literal array, as long as the annotated type is valid for each item in the array:
for i: int128 in [4, 23, 42]:
...
Some restrictions:
You cannot iterate over a multi-dimensional array.
i
must always be a base type.You cannot modify a value in an array while it is being iterated, or call to a function that might modify the array being iterated.
Range Iteration¶
Ranges are created using the range
function. The following examples are valid uses of range
:
for i: uint256 in range(STOP):
...
STOP
is a literal integer greater than zero. i
begins as zero and increments by one until it is equal to STOP
. i
must be of the same type as STOP
.
for i: uint256 in range(stop, bound=N):
...
Here, stop
can be a variable with integer type, greater than zero. N
must be a compile-time constant. i
begins as zero and increments by one until it is equal to stop
. If stop
is larger than N
, execution will revert at runtime. In certain cases, you may not have a guarantee that stop
is less than N
, but still want to avoid the possibility of runtime reversion. To accomplish this, use the bound=
keyword in combination with min(stop, N)
as the argument to range
, like range(min(stop, N), bound=N)
. This is helpful for use cases like chunking up operations on larger arrays across multiple transactions. i
, stop
and N
must be of the same type.
Another use of range can be with START
and STOP
bounds.
for i: uint256 in range(START, STOP):
...
Here, START
and STOP
are literal integers, with STOP
being a greater value than START
. i
begins as START
and increments by one until it is equal to STOP
. i
, START
and STOP
must be of the same type.
Finally, it is possible to use range
with runtime start and stop values as long as a constant bound value is provided.
In this case, Vyper checks at runtime that end - start <= bound.
N
must be a compile-time constant. i
, stop
and N
must be of the same type.
for i: uint256 in range(start, end, bound=N):
...