o /wiZ@sUddlmZddlZddlZddlmZddlmZddlmZddlm Z ddl Z ddl Z ddl m Z ddl mZdd lmZdd lmZe rNddlZn dd lmZed Zgd Zded<ddgZdZGdddZdddZdS)) annotationsN) TracebackType)Any)IO) TYPE_CHECKING) BaseStorage)GrpcStorageProxy)JournalFileBackend)NamedTemporaryFilePool) _LazyImportgrpc)inmemorysqlite cached_sqlitejournal journal_redisgrpc_rdbgrpc_journal_filez list[Any] STORAGE_MODESrri,c@s4eZdZdddZdd d ZdddZdddZdS)StorageSupplierstorage_specifierstrkwargsrreturnNonecKs(||_||_d|_d|_d|_d|_dS)N)r extra_argstempfileserverthreadproxy)selfrrr!T/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/optuna/testing/storages.py__init__/s  zStorageSupplier.__init__optuna.storages.InMemoryStorage | optuna.storages._CachedStorage | optuna.storages.RDBStorage | optuna.storages.JournalStorage | optuna.storages.GrpcStorageProxycCs|jdkrt|jdkrtdtjSd|jvrEt|_d |jj }tjj |fdddt iii|j}d |jvrCtj |S|S|jd kratjjd }|jd t|_tj|S|jd kr|jdt|_|jdusxJtjtjj|jj }||Sd|jvr|jdt|_|jdusJt|jj }tj|S|jdkrt|_d |jj }|tj |S|jdkrd|jvsJ||jdSJ)Nr rz.InMemoryStorage does not accept any arguments!rz sqlite:///{} engine_kwargs connect_argstimeoutcachedrzredis://localhostredisrfilerr grpc_proxy base_storage)rlenr ValueErroroptunastoragesInMemoryStorager rformatname RDBStorageSQLITE3_TIMEOUT_CachedStoragerJournalRedisBackendget fakeredisFakeStrictRedis_redisJournalStorager _create_proxy)r url rdb_storagejournal_redis_storagestorage file_storager!r!r" __enter__7s^                 zStorageSupplier.__enter__rArrcCsZt}tjjj|d||_tj|jjd|_ |j t d|d|_ |j j dd|j S)N localhost)target)hostport<)r') _find_free_portr/r0_grpcr make_server threadingThreadstartrrrwait_server_ready)r rArGr!r!r"r=os zStorageSupplier._create_proxyexc_typetype[BaseException]exc_val BaseExceptionexc_tbrcCsh|jr|j|jr|jd|_|jr2|jdusJ|jd|jd|_d|_dSdS)N)rcloserrrstopwaitjoin)r rPrRrTr!r!r"__exit__xs    zStorageSupplier.__exit__N)rrrrrr)rr$)rArrr)rPrQrRrSrTrrr)__name__ __module__ __qualname__r#rCr=rZr!r!r!r"r.s    8 rrintc CsRttjtj}tddD]}z |d|f|WSty$Yq wJd)Ni2i,3rDFzmust not reach here)socketAF_INET SOCK_STREAMrangebindOSError)sockrGr!r!r"rIs  rI)rr^) __future__rr_rLtypesrtypingrrrr9r/optuna.storagesrroptuna.storages.journalr optuna.testing.tempfile_poolr r optuna._importsr r__annotations__STORAGE_MODES_HEARTBEATr5rrIr!r!r!r"s0           \