Python typing

From wikinotes
Revision as of 15:05, 3 April 2022 by Will (talk | contribs) (→‎Example)

Type annotations in python.

Example

from typing import Dict, List, Tuple
from typing import Optional, Union

class Project:
    # type/return val
    def add_user(user: User) -> Bool
        pass

    # type w/ default value
    def set_active(status: bool=False) -> Bool:
        pass

    # generics
    def update_metadata(data: List[Dict]) -> List[Dict]:
        pass

    # optional
    def grade() -> Optional[str]:  # nullable type
        pass

    # one type or another
    def set_id(id: Union[str, int]) -> None:
        pass

Common Types

dates/times

import datetime

def foo() -> datetime.datetime:
    return datetime.datetime.now()

def foo() -> datetime.date:
    return datetime.date.today()

open file objects

from typing import IO

# either bytes/str
def foo() -> IO:
    return open('foo.txt', 'r')

def foo() -> IO[bytes]:
    return open('foo.txt', 'rb')

def foo() -> IO[str]:
    return open('foo.txt', 'r')

context managers

@contextlib.contextmanager
def foo() -> Iterator[str]:
    print('foo')
    yield 'bar'
    print('baz')