iDdZddlmZmZddlmZddlmZGddZy)z Docker container checks module )DictOptional)Logger) DockerUtilsceZdZdZdedefdZddedefdZdefdZ defd Z dedefd Z defd Z defd Z ddeedefdZddedefdZy ) DockerChecksz"Check and manage Docker containerslogger project_dircz||_t||_|jj|gd|_y)N)postgresredisbackenddaemonfrontendcaddynginxbackup)r r docker_utilsset_project_dir containers)selfr r s :/home/factor_ks/STREAMA2026/tests/modules/docker_checks.py__init__zDockerChecks.__init__ s3 '/ ))+6 auto_fixreturnc|jjd|jsy|jsy|j |sy|j sy|j sy|jjddy)z/Run all Docker checks and optionally fix issuesz Docker ChecksFDOCKERzAll Docker checks passedT)r subheader_check_docker_installation"_check_docker_compose_installation_check_container_statuses_check_network_check_volumessuccess)rrs rrun_all_checkszDockerChecks.run_all_checkss o...0668--h7""$""$ H&@Arc|jjdd|jjr|jj ddy|jj ddy)zCheck if Docker is installedrzChecking Docker installation...zDocker is installedTz)Docker is not installed or not accessibleF)r inforcheck_docker_installedr%errorrs rr z'DockerChecks._check_docker_installation0sY #DE    3 3 5 KK  *? @ KK  h(S Trc|jjdd|jjr|jj ddy|jj ddy)z$Check if Docker Compose is installedrz'Checking Docker Compose installation...zDocker Compose is installedTz1Docker Compose is not installed or not accessibleF)r r(rcheck_docker_compose_installedr%r*r+s rr!z/DockerChecks._check_docker_compose_installation;sY #LM    ; ; = KK  *G H KK  h([ \rc|jjdd|jj}d}|jD]}|j |d}|dk(r"|jj d|d|=|dk(r|jjd|dd}|sf|jjdd |jjr|jj dd |jjdd y|jjd|d||s|jjdd |d |jj|}|sd|}|dk(r_|jj|r!|jj d|d~|jjdd|d}|jj|r!|jj d|d|jjdd|d}|S)z8Check status of all containers and optionally fix issuesrzChecking container statuses...T not_foundrunningz: z%: not found - container may not existFz8Attempting to start containers with docker-compose up...zContainers started successfullyzFailed to start containerszAttempting to fix z...z streama2026-exitedz startedzFailed to start z restartedzFailed to restart ) r r(rget_all_container_statusesrgetr%r* compose_upwarningget_container_name_by_servicestart_containerrestart_container)rrstatusesall_ok containerstatuscontainer_names rr"z&DockerChecks._check_container_statusesFs #CD$$??A# +I\\)[9F" ##H 2fX.FG;& !!(yk9^,_`KK$$X1ik((335 ++H6WX ))(4PQ$ ##H 2fX.FGKK$$X1CI;c/RS%)%6%6%T%TU^%_N)+7 {)C),,<<^L KK//YKx:PQ KK--h:J9+8VW%*F ,,>>~N KK//YKz:RS KK--h:LYK8XY%*FG# +J rc|jjdd|jjdr|jj ddy|jjdd|jj dr|jj ddy|jjddy) zCheck if Docker network existsrzChecking Docker network...zstreama-networkzDocker network existsTz2Docker network not found - attempting to create...zDocker network createdz1Docker network will be created on next compose up)r r(rcheck_network_existsr%create_networkr+s rr#zDockerChecks._check_networkts #?@    1 12C D KK  *A B KK  X'[ \  //0AB ##H.FG   +^_rc|jjddgd}d}|D]^}|jj|r!|jj dd|d?|jj dd|d`y)zCheck if Docker volumes existrzChecking Docker volumes...)streama2026_postgres_datastreama2026_redis_datastreama2026_stream_storagestreama2026_ffmpeg_cacheTzVolume z existsz) not found - will be created on first use)r r(rcheck_volume_existsr%r5)rvolumesr:volumes rr$zDockerChecks._check_volumess #?@  kF  44V< ##Hxw.GH ##Hx?h.ij  k rNservicesc|jjdd|ddg}|jj|r/|jj dddj |y|jj ddy ) z"Rebuild specific containers or allrzRebuilding containers...rrzContainers rebuilt: z, TzFailed to rebuild containersF)r r(rrebuild_containersr%joinr*)rrIs rrKzDockerChecks.rebuild_containerss| #=>  !:.H    / / 9 KK  ,@8AT@U*V W KK  h(F Grtimeoutc|jjdddddd}d}|jD]\}}|jjdd|d |jj ||r |jj d|d b|d k(r |jjd|d |jj d|d d}|S)z=Wait for all containers to become healthy with active pollingHEALTHz-Waiting for health checks (active polling)...zstreama2026-postgreszstreama2026-rediszstreama2026-backend)r r rTz Waiting for z to become healthy...z : healthyrz*: failed to become healthy (but API works)z: failed to become healthyF)r r(itemsrwait_for_healthr%r*)rrMcontainer_namesr:servicer=s rrQzDockerChecks.wait_for_healths #RS/(,  '6'<'<'> # #G^ KK  XgY>S'T U  00I ##H .CDi'KK$$X':d/efKK%%h7);U0VW"F # r)T)N)<)__name__ __module__ __qualname____doc__rstrrboolr&r r!r"r#r$rlistrKintrQrrrr s, v C tt6 D  D ,$,4,\"( 8D> T sDrrN) rXtypingrrtests.utils.loggerrtests.utils.docker_utilsrrr]rrras!"%0uur