Update stubs

This commit is contained in:
Trinh Anh Ngoc
2022-10-14 23:31:24 +07:00
parent aa693d6561
commit 6609be1d16

View File

@@ -1,12 +1,13 @@
from collections import defaultdict from collections import defaultdict
from typing import Any, Callable, Generator, Iterator, KeysView, Literal, Mapping, Optional, TypeVar, Union from typing import Any, Callable, Collection, Generator, Iterable, Iterator, KeysView, Literal, Mapping, Optional, \
TypeVar, Union
from weakref import WeakSet from weakref import WeakSet
from .fields import Field from .fields import Field
from .models import BaseModel from .models import BaseModel
from .modules.registry import Registry from .modules.registry import Registry
from .sql_db import Cursor from .sql_db import Cursor
from .tools import frozendict, StackMap from .tools import OrderedSet, frozendict, StackMap
_T = TypeVar('_T') _T = TypeVar('_T')
_ModelT = TypeVar('_ModelT', bound=BaseModel) _ModelT = TypeVar('_ModelT', bound=BaseModel)
@@ -84,6 +85,9 @@ class Environment(Mapping[str, BaseModel]):
def lang(self) -> str: ... def lang(self) -> str: ...
def clear(self) -> None: ... def clear(self) -> None: ...
def clear_upon_failure(self): ... def clear_upon_failure(self): ...
def invalidate_all(self, flush: bool = ...) -> None: ...
def _recompute_all(self) -> None: ...
def flush_all(self) -> None: ...
def is_protected(self, field: Field, record: BaseModel) -> bool: ... def is_protected(self, field: Field, record: BaseModel) -> bool: ...
def protected(self, field: Field) -> BaseModel: ... def protected(self, field: Field) -> BaseModel: ...
def protecting(self, what, records: Optional[BaseModel] = ...) -> Generator[None, None, None]: ... def protecting(self, what, records: Optional[BaseModel] = ...) -> Generator[None, None, None]: ...
@@ -102,7 +106,6 @@ class Transaction:
cache: Cache cache: Cache
protected: StackMap[Field, set[int]] protected: StackMap[Field, set[int]]
tocompute: defaultdict[Field, set[int]] tocompute: defaultdict[Field, set[int]]
towrite: defaultdict[str, dict[int, dict[str, Any]]]
def __init__(self, registry: Registry) -> None: ... def __init__(self, registry: Registry) -> None: ...
def flush(self) -> None: ... def flush(self) -> None: ...
def clear(self) -> None: ... def clear(self) -> None: ...
@@ -113,13 +116,18 @@ EMPTY_DICT: frozendict
class Cache: class Cache:
_data: defaultdict[Field, dict] _data: defaultdict[Field, dict]
_dirty: defaultdict[Field, OrderedSet[int]]
def __init__(self) -> None: ... def __init__(self) -> None: ...
def __repr__(self) -> str: ...
def _get_field_cache(self, model: BaseModel, field: Field) -> dict: ... def _get_field_cache(self, model: BaseModel, field: Field) -> dict: ...
def _set_field_cache(self, model: BaseModel, field: Field): ... def _set_field_cache(self, model: BaseModel, field: Field): ...
def contains(self, record: BaseModel, field: Field) -> bool: ... def contains(self, record: BaseModel, field: Field) -> bool: ...
def contains_field(self, field: Field) -> bool: ...
def get(self, record: BaseModel, field: Field, default=...): ... def get(self, record: BaseModel, field: Field, default=...): ...
def set(self, record: BaseModel, field: Field, value) -> None: ... def set(self, record: BaseModel, field: Field, value, dirty: bool = ..., check_dirty: bool = ...) -> None: ...
def update(self, records: BaseModel, field: Field, values) -> None: ... def update(self, records: BaseModel, field: Field, values, dirty: bool = ..., check_dirty: bool = ...) -> None: ...
def update_raw(self, records: BaseModel, field: Field, values, dirty: bool = ..., check_dirty: bool = ...) -> None: ...
def insert_missing(self, records: BaseModel, field: Field, values) -> None: ...
def remove(self, record: BaseModel, field: Field) -> None: ... def remove(self, record: BaseModel, field: Field) -> None: ...
def get_values(self, records: BaseModel, field: Field) -> Iterator: ... def get_values(self, records: BaseModel, field: Field) -> Iterator: ...
def get_until_miss(self, records: BaseModel, field: Field) -> list: ... def get_until_miss(self, records: BaseModel, field: Field) -> list: ...
@@ -127,5 +135,16 @@ class Cache:
def get_fields(self, record: BaseModel) -> Iterator[Field]: ... def get_fields(self, record: BaseModel) -> Iterator[Field]: ...
def get_records(self, model: _ModelT, field: Field) -> _ModelT: ... def get_records(self, model: _ModelT, field: Field) -> _ModelT: ...
def get_missing_ids(self, records: BaseModel, field: Field) -> Iterator[int]: ... def get_missing_ids(self, records: BaseModel, field: Field) -> Iterator[int]: ...
def get_dirty_fields(self) -> 'set[Field]': ...
def get_dirty_records(self, model: _ModelT, field: Field) -> _ModelT: ...
def has_dirty_fields(self, records: BaseModel, fields: Iterable[Field] | None = ...) -> bool: ...
def clear_dirty_field(self, field: Field) -> Collection[int]: ...
def invalidate(self, spec: list[tuple[Field, tuple | None]] | None = ...) -> None: ... def invalidate(self, spec: list[tuple[Field, tuple | None]] | None = ...) -> None: ...
def clear(self) -> None: ...
def check(self, env: Environment) -> None: ... def check(self, env: Environment) -> None: ...
class Starred:
__slots__ = ['value']
value: Any
def __init__(self, value) -> None: ...
def __repr__(self) -> str: ...