o ޭh@sddlmZddlZddlmZddlmZmZddlm Z ej dkr*ddl m Z n ddl m Z d"ddZ er=ddlmZd#ddZd$ddZd%d&d d!ZdS)') annotationsN)Iterable) TYPE_CHECKINGAny)logger) batched)isliceiterable Iterable[Any]nintreturnIterable[tuple[Any]]ccs<t|}tt||}r|Vtt||}sdSdSN)itertupler )r ritbatchrS/var/www/html/stock_analysis/be/venv/lib/python3.10/site-packages/tortoise/utils.pyr s r )BaseDBAsyncClientclientrsafeboolstrcCs||}||S)z Generates the SQL schema for the given client. :param client: The DB client to generate Schema SQL for :param safe: When set to true, creates the table only when it does not already exist. )schema_generatorget_create_schema_sql)rr generatorrrrget_schema_sqls  r!Nonecs>||}t||}td||r||IdHdSdS)z Generates and applies the SQL schema directly to the given client. :param client: The DB client to generate Schema SQL for :param safe: When set to true, creates the table only when it does not already exist. zCreating schema: %sN)rr!rdebuggenerate_from_string)rrr schemarrrgenerate_schema_for_client#s   r& instances batch_size int | NoneIterable[Iterable[Any]]ccs$|s|VdSt||EdHdS)zB Generate iterable chunk by batch_size # noqa: DAR301 Nr )r'r(rrrchunk1s r+)r r rrrr)rrrrrr)rrrrrr"r)r'r r(r)rr*) __future__rsyscollections.abcrtypingrr tortoise.logr version_info itertoolsr r tortoise.backends.base.clientrr!r&r+rrrrs