ruạṛ
a �u:j/A � @ s� U d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZ d dl m Z d dl m Z d dl mZ d dlT d dlmZ d d lmZmZmZmZmZmZmZmZ daee d < d Z d Z e e e d �dd�Z!ee e ee d�dd�Z"ee e ee d�dd�Z#ed4e e e$eee e%d�dd��Z&ed5e e$e%d�dd��Z'd d � Z(e(d6e e$e$e eee e%e%e%d �d d ��Z)d!d"� Z*d#d$� Z+e e d%�d&d'�Z,e d(�d)d*�Z-e e d+�d,d-�Z.e e d.�d/d0�Z/d1d2� Z0e1d3k�r�e0� dS )7� N)�Path)�List�Optional)�bart_jobs_folder)� ErrorCode)� BartError)�*)�JobActivityLogger)� measure_time�cleanup_backup_folder�run_sudo_command_str�run_sudo_command�create_backup_tar_cli� read_homedir�does_path_exists� is_directory�activity_loggerzYmysqldump --no-create-info --no-data --databases <db_name> > <output_dir><db_name>.createzJmysqldump --no-create-db --databases <db_name> > <output_dir><db_name>.sql)�username�cbs_id�returnc C s� t � d|� d| � d|� �}tj�|d�}tj�|d�}tj|dd� tj|dd� t�d|� �� t�d |� �� t�d |� �� |S ) Nz/job_�/Z _partial_�homedir�mysqlT��exist_okzCreated backup folder: zCreated homedir folder: zCreated mysql folder: )r �os�path�join�makedirsr �success)r r � backup_folderZhomedir_folderZ mysql_folder� r! �N/opt/bart/bart_venv/lib/python3.9/site-packages/app/backup_realtime_partial.py�create_empty_folder s r# )� exclude_paths�baser c C st g }|p d� d�}| pg D ]T}|s$q|�d�}|r`||ksH|�|d �r`|t|�d� �d�}n|}|�|� q|S )a. Make excludes relative to the given base path. Examples: base='public_html', exclude='public_html/wp-content' -> 'wp-content' base='public_html', exclude='public_html' -> '' (exclude entire base) base='public_html/blog', exclude='public_html/blog/wp-admin' -> 'wp-admin' � r N)�strip�lstrip� startswith�len�append)r$ r% �normedZ base_clean�eZe_cleanZe_relr! r! r" � _normalize_excludes_for_base' s r. c C s^ t | pg |�}g }|s|S |D ]:}|dkr:|�ddg� q |�d|� d�d|� d�g� q |S )z� Build rsync-specific exclude flags anchored at the transfer root. If normalized exclude is '' -> exclude everything under the base. r& z--exclude=/**/*z --exclude=/z/***r )r. �extend)r$ r% r, Z rsync_argsr- r! r! r" �_build_rsync_ex_args= s r0 F)r r �pathsr$ �skip_missing_pathsc C s� g }g }|pg D �]�}|� d�}tj�||�} tj�| d|�} t| �s�d| � d�} t�| � |�|� |r�t�d|� d�� qn t� | � d�� t t j | ��tj tj�| �dd � t|p�g |�} | r�t�d d �| �� �� ztt| ��rd d d|g| �| � d��| � d��} t| � nd d d|| | g} t| � t�d|� d| � �� |�|� W q t�y� } z8d| � d| � d|� �} t� | � t t j| ��W Y d }~qd }~0 0 q||fS )Nr r zPath z! does not exist in the filesystemzSkipping missing path � and continuing backupz , exiting.Tr zExcludes (rsync): � �rsyncz-az --homedirzCopied z to zFailed to copy u → �: )r' r r r r r �warnr+ �info�errorr r ZPATH_NOT_FOUNDr �dirnamer0 r r r � ExceptionZRSYNC_FAILED_DURING_BACKUP)r r r1 r$ r2 Zsuccessful_paths� skipped_pathsr r% � source_pathZdestination_path�msgZrsync_exZ rsync_cmdr- r! r! r" � copy_pathsR sV ���� � $r? )r � databases�skip_missing_databasesc C sr t j�| d�d }t j|dd� g }g }|D �]:}t�d|� �� t�d|��d|�}t�d|��d|�}z4t |� t |� t� d|� d |� �� |� |� W q, t �yf } z�d t | �v } d |� d | � �} | �rt�d|� d �� |� |� |�rt�d|� d�� W Y d } ~ q,tj} ntj} t�| � d�� t| � t�d| � �� t| | ��W Y d } ~ q,d } ~ 0 0 q,||fS )Nr r Tr zCreating dump for db z <db_name>z <output_dir>z Database z dumped to zUnknown databasezFailed to dump database r6 z not found on serverzSkipping missing database r3 z*. Cleaning up the jobs folder and exiting.�Cleaned backup folder )r r r r r �note�MYSQL_COMMAND_DB_CREATE�replace� MYSQL_COMMAND_TABLE_CREATE_DUMPr r r+ r; �strr7 r8 r Z DB_NOT_FOUNDZSQL_DUMP_CMD_FAILEDr9 r r ) r r@ rA � output_dirZsuccessful_dbs� skipped_dbs�dbZmysqldump_db_create_cmdZ mysqldump_table_create_dump_cmdr- Z is_unknown_dbr> Z error_coder! r! r" �copy_databases� sF ��� "rK c s � fdd�}|S )Nc s> t � � }� | i |��}t � � }|| }t�d|d�d�� |S )Nz![TOTAL TIME] Script completed in z.3fz seconds)�timer r8 )�args�kwargs� start_time�resultZend_timeZ elapsed_time��funcr! r" �wrapper� s z log_time_taken.<locals>.wrapperr! )rR rS r! rQ r" �log_time_taken� s rT )r r1 r@ r r$ rA r2 �is_clic C s$ t ||� t�d� t| |�}t| �} | sLd| � d�} t�| � ttj| ��t || |||d�\} } t |||d�\} }| r�t� dd� | �� �� | r�t� dd� | �� �� |r�t� d d� |�� �� | r�t� d d� | �� �� t|| � t|�}t� d |� �� t|� t� d |� �� || | | |fS ) Nz4<======== Starting Realtime Partial Backup ========>zUser z does not exist on the server.)r$ r2 )rA zSkipped paths: z, z Successfully backed up paths: zSkipped databases: z"Successfully backed up databases: zCreated tarfile rB )�set_job_loggerr r8 r# r r9 r r ZUSER_NOT_FOUNDr? rK r7 r r �generate_mysql_sql_filer r )r r1 r@ r r$ rA r2 rU r r � error_msgZbacked_up_pathsr< Z backed_up_dbsrI Zoutput_tarfile_pathr! r! r" �backup� s0 rY c C sz t j�| d�}z0t�d� t||d� t�d|� d|� �� W n6 tyt } z t�d|� d�� W Y d }~n d }~0 0 d S )Nz mysql.sqlz Starting generating mysql.sql��cpuser� output_pathzGenerated mysql.sql for z location: z Failed to generate mysql.sql: z+. Continuing backup without mysql.sql file.)r r r r r8 �generate_mysql_sqlr; r7 )r r Zmysql_sql_pathr- r! r! r" rW � s rW c C s t | |d�ad S )N)Zecho_to_stdout)r r )Zjob_idrU r! r! r" rV � s rV )�liner c C s dd� }d}t �||| �S )a Escape underscores in database names inside backticks, exactly like cPanel. Example: `vqolkate_WPFVR` -> `vqolkate\_WPFVR` Only database names (before .*) are escaped. Normalizes escaping: first unescapes any existing escapes, then escapes all underscores. c S sL | � d�}t�dd|�}d}|D ] }|dkr6|d7 }q ||7 }q d|� d�S )N� z\\+_�_r& z\_�`)�group�re�sub)�matchZdbnameZdbname_normalizedZdbname_escaped�charr! r! r" �repl s z$escape_dbname_in_grant.<locals>.replz`([^`]+)`(?=\.\*))rc rd )r^ rg �patternr! r! r" �escape_dbname_in_grant� s ri )r[ c C s� d| � d�}z*t dd|g�}|du r4t�d� g W S W n8 tyn } z t�d|� �� g W Y d}~S d}~0 0 g }|�� }|r�d|d v r�d nd }||d� D ]@}|�d �}t|�d kr�|\} } } |�| �� | �� | �� f� q�|S ) z� Return list of (user, host, password_hash) tuples for all MySQL users belonging to the cPanel account. Uses MySQL wrapper for database commands. zJSELECT User, Host, authentication_string FROM mysql.user WHERE User LIKE 'z%';r �-eNz-Error querying mysql.user: No output returnedzError querying mysql.user: ZUserr r_ � � ) r r r9 r; � splitlines�splitr* r+ r' ) r[ �queryrP r- �users�linesZ start_idxr^ �parts�user�host�pw_hashr! r! r" �get_cp_mysql_users s$ rv )rs rt c C s z>t ddd| � d|� d�g�}|du r<td| � d|� d ���W n> ty| } z&td| � d|� d |� ���W Y d}~n d}~0 0 g }|�� D ]~}|�� }|s�q�|�� �d �r�q�|�� �d �s�q�|�� �d �r�|�|� q�t|�}t � dd|�}|� d�s�|d7 }|�|� q�|S )z� Return a list of GRANT statements for this user@host, properly formatted for cPanel. Uses MySQL wrapper for database commands. r rj zSHOW GRANTS FOR '�'@'�';NzError fetching grants for �@z: No output returnedr6 z GRANTS FORZGRANT� GRANT USAGEz`([^`]+)`@`([^`]+)`z '\1'@'\2'�;) r � RuntimeErrorr; rm r' �upperr) r+ ri rc rd �endswith)rs rt rP r- �grantsr^ r! r! r" �build_grants_for_user5 s0 0 r� rZ c C s$ g }t | �}|s&t�d| � d�� dS |D ]�\}}}|rV|�d|� d|� d|� d�� z2t||�}|D ] }|�� �d�rzqf|�|� qfW q* ty� } z4t�d |� d |� d | � �� W Y d} ~ q*W Y d} ~ q*d} ~ 0 0 q*|�rt|�� d � |�d � t� d |� d| � �� nt�d| � d�� dS )zy Generate mysql.sql file with user grants for the cPanel user. Uses MySQL wrapper for all database commands. zNo MySQL users found for z , skipping mysql.sql generationNzGRANT USAGE ON *.* TO 'rw z' IDENTIFIED BY PASSWORD 'rx rz zFailed to fetch grants for ry r6 � zCreated mysql.sql at z for cPanel user zNo grants found for z, mysql.sql not created) rv r r7 r+ r� r} r) r; r � write_textr r ) r[ r\ �outputrp rs rt ru r �gr- r! r! r" r] ` s. � $r] c C s� t jdd�} | jdtddd� | jddd d � | jd dd d � | jd ddd � | jdtddd� | �� }|j}|j}|j}t |� t |� |r�t |� t |j |||j |ddd� d S )Nz@Backup script for docroot and database data for realtime backup.)� descriptionz --usernameTzSpecify the username)�type�required�helpz--paths�+z=Space-separated paths (relative to homedir), e.g. public_html)�nargsr� z--exclude-pathszsSpace-separated paths to exclude (relative to each path you copy), e.g. public_html/wp-admin public_html/wp-contentz --databasesz Space-separated database namesz--cbs_idzSpecify the CBS IdF)r$ rA rU )�argparse�ArgumentParser� add_argumentZvalidate_usernameZvalidate_cbs_id� parse_argsr1 r@ r$ Zvalidate_docrootsZvalidate_databasesrY r r )�parserrM r1 r@ r$ r! r! r" �main� s" �r� �__main__)NF)F)NFFF)2r� r rc rL �pathlibr �typingr r Z app.configr Zapp.errors.error_codesr Zapp.errors.exceptionsr Zapp.utils.args_validationsZ app.utils.job_activity_loggerr Zapp.utils.utility_functionsr r r r r r r r r �__annotations__rD rF rG �intr# r. r0 �list�boolr? rK rT rY rW rV ri rv r� r] r� �__name__r! r! r! r" �<module> sZ ( ";- � �' +&
cải xoăn