sjo d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlm Z d dl
m Z d dl m
Z
d dlmZmZ d d lmZ e e Zd
Zd efd Z G d
d Z G d de Z G d de Z G d de Zd Zd Z dS ) N)Path)Dict) WposError)
USER_WPOS_DIR)
setup_logging)ALL_OPTIMIZATION_FEATURES)home_diris_run_under_user)write_file_via_tempfilez.progress-statusreturnc P t t t t S )N)r r r PROGRESS_STATUS_FILE_NAME py/clwpos/user/progress_check.pyget_progress_file_pathr s
M+D E EEr c e Zd ZU dZeed< fdZd Zdee ef fdZ
d
dZ d
d Z d
d
Z
d
d Zedee ef fd Z xZS )CommandProgresszK
Class to represent user's AccelerateWP utility commands progress.
total_stagesc d | D }||vr4t d| dt | || }t t
| | }|S )Nc i | ]
}|j | S r )command).0subclasss r
z+CommandProgress.__new__..$ s*
+3H h
r zInternal Error: command "z&" is not allowed, available commands: )__subclasses__r listkeyssuperr __new__)clsr command_subclass_mapr instance __class__s r r zCommandProgress.__new__# s
7:7I7I7K7K
. . . W W W378L8Q8Q8S8S3T3T W W X X
X'0 (33;;HEEr c t st d t | _ d| _ d| _ d S )Nz;Internal Error: trying to use CommandProgress class as rootFr )r
r r
progress_file in_progresscompleted_stages)selfargss r __init__zCommandProgress.__init__. sD "" [ YZZ Z 3 5 5 ! r r c | j | j dS )zF
Return number of total and completed stages as dict.
r r( r- r) s r as_dictzCommandProgress.as_dict5 s
! - $ 5
r Nc d| _ d| _ | j j ddd | dS )z=
Start tracking the progress of the command.
Tr i )modeparentsexist_okN)r' r( r& parentmkdir_update_progress_filer. s r startzCommandProgress.start> sM ! !''UD4'PPP ""$$$$$r c | j sdS | xj dz
c_ | j | j k rt d | dS )zg
Update progress status by incrementing
the number of completed stages by one.
N zYInternal Error: the number of completed stages is greater than the total number of stages)r' r( r r r6 r. s r updatezCommandProgress.updateG sj
F " 4#4 4 4=
""$$$$$r c | j sdS d| _ | j | _ | j r| j dS dS )zF
Complete progress, delete file with progress status.
NF)r' r r( r& existsunlinkr. s r completezCommandProgress.completeV sc F $ 1 $ $ & & ( % % ' ' ' ' ' ( (r c z t t j | | j d d S )Ni )r jsondumpsr/ r& r. s r r6 z%CommandProgress._update_progress_fileb sA Jt||~~ & &(:E C C C C Cr c t } ddd}| s|S t | d 5 }t j | }ddd n # 1 swxY w Y t
|d t
|d dS # t $ r'}t d| |cY d}~S d}~ww xY w) zu
Return current progress status of the command execution,
getting it from from special file.
r9 r r- rNr r( z$Can't parse progress status json: %s)
r r< openr@ loadsreadint Exceptionloggererror)r& defaultsfstatus_from_filees r
get_statuszCommandProgress.get_statusi s- /00
$%1== ##%% O mS))
8Q#':affhh#7#7
8
8
8
8
8
8
8
8
8
8
8
8
8
8
8 !$$4^$D E E$'(89K(L$M$M
LL@! D D DOOOOOO s@ B$ 'A." B$ .A22B$ 5A26-B$ $
C. C
CC)r N)__name__
__module__ __qualname____doc__rG __annotations__r r+ r strr/ r7 r: r> r6 staticmethodrO
__classcell__)r$ s @r r r s " " "
c3h
% % % %
%
%
%
% ( ( ( (C C C C S#X \ r r c 0 e Zd ZU dZdZeed< dZeed< dS )EnableCommandProgresszS
Class to represent user's AccelerateWP utility "enable" command progress.
enabler r N rP rQ rR rS r rU rT r rG r r r rY rY s> GSL#r rY c 0 e Zd ZU dZdZeed< dZeed< dS )DisableCommandProgresszL
Class to represent user's WPOS utility "disable" command progress.
disabler r Nr\ r r r r^ r^ s> GS L#r r^ c < e Zd ZU dZdZeed< dZeed< de fdZ
dS ) GetCommandProgresszH
Class to represent user's WPOS utility "get" command progress.
getr
r user_infoc d}t | }t d | D }||z }||t t z z | _ dS )a"
Recalculate and change total number of progress stages depending on
the number of user's docroots and wordpresses.
so we could update progress after checking every docroot and wordpress.
Example of user_info structure:
{
'docroot1':
{'wps': [{'path': '', 'version': '5.9'},
{'path': 'wp2', 'version': '5.9'}], ... },
'docroot2':
{'wps': [{'path': '', 'version': '5.9'}], ... },
}
c 3 @ K | ]}t |d V dS )wpsN)len)r
doc_root_infos r zHGetCommandProgress.recalculate_number_of_total_stages.. s@ "
"
*7C
e$ % %"
"
"
"
"
"
r N)rj sumvaluesr r )r) re number_of_initial_stagesnumber_of_docrootsnumber_of_wordpresstotal_checks_per_modules r "recalculate_number_of_total_stagesz5GetCommandProgress.recalculate_number_of_total_stages s $% ^^ ! "
"
;D;K;K;M;M"
"
"
#57J"J 4%,E(F(FF Gr N) rP rQ rR rS r rU rT r rG dictrs r r r rb rb sd GSL#GD G G G G G Gr rb c F t j fd }|S )zX
Decorator for the WPOS utility commands
we want to track the progress for.
c t j d | _ | j | g|R i || j S # | j w xY w)N_)r rP stripcommand_progressr7 r> )r) r* kwargsfuncs r wrapperz'track_command_progress..wrapper s /
0C0CC0H0H I I ##%%% -4.t...v.. ! * * , , , ,D ! * * , , , ,s A- -B functoolswrapsr{ r| s ` r track_command_progressr s:
_T- - - - - Nr c F t j fd }|S )zu
Decorator for the WPOS utility methods after which
the counter of completed operations must be updated.
c r | g|R i |}t | d | j |S )Nry )getattrry r: )r) r* rz resultr{ s r r| z(update_command_progress..wrapper sM d,T,,,V,, 4+ , , 8 ! ( ( * * *
r r} r s ` r update_command_progressr s:
_T
Nr ) r~ r@ pathlibr typingr clwpos.cl_wpos_exceptionsr clwpos.constantsr clwpos.logsetupr clwpos.optimization_featuresr clwpos.utilsr r
secureior rP rI r r r rY r^ rb r r r r r r s / / / / / / * * * * * * ) ) ) ) ) ) B B B B B B 4 4 4 4 4 4 4 4 , , , , , , x . F F F F Fa a a a a a a a H O _ G G G G G G G G>
r