| Path : /var/www/html/yedekledb24/DatabaseBackupMasterv2/__pycache__/ |
|
B-Con CMD Config cPanel C-Rdp D-Log Info Jump Mass Ransom Symlink vHost Zone-H |
| Current File : /var/www/html/yedekledb24/DatabaseBackupMasterv2/__pycache__/storage_utils.cpython-310.pyc |
o
���g�1 � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlmZ d dlZdZ zd dl
Z dZ W n e yC e�
e�Ze�d� Y nw e�
e�Zdd � Zd
d � Zd d
� ZdS )� N)�datetime)�FTP)�
SMBConnectionFTz;NFS support not available - nfs4-share module not installedc C s� z�| dkrJt ||d|dd�}|du rd}|�||�}|s W dS |�� }dd � |D �} |�d
�d }
|
| vrCt�d |
� d
| � �� W dS |�� W dS | dkr�tsXt�d� W dS z't� � } t
j |dd�} | � |�}
|
�
| �}t�| � |�� t�| � W W dS ty� } zt�dt|�� �� W Y d}~W dS d}~ww | dkr�|du r�d}t� }|�||� |�||� |r�|�|� |�� W dS t�d| � �� W dS ty� } zt�d| � d|� dt|�� �� W Y d}~dS d}~ww )a�
Test connection to a storage target.
Args:
storage_type (str): Storage type (smb, nfs, ftp)
host (str): Storage host
port (int): Storage port
username (str): Username for storage
password (str): Password for storage
path (str): Storage path
Returns:
bool: True if connection successful, False otherwise
�smb�
backup-appT�� use_ntlm_v2N� Fc S s g | ]}|j �qS � )�name)�.0�sharer
r
�?/var/www/html/yedekledb24/DatabaseBackupMaster/storage_utils.py�
<listcomp>9 s z+test_storage_connection.<locals>.<listcomp>�/r zShare z not found in available shares: �nfs�>NFS support is not available - nfs4-share module not installed�tcp��protozNFS connection error: �ftp� �Unsupported storage type: z Storage connection error for � �: )r �connect�
listShares�split�logger�error�close� NFS_SUPPORT�tempfile�mkdtemp�nfs4�
NFS4Client�
get_export�mount�os�listdir�umount�rmdir� Exception�strr �login�cwd�quit)� storage_type�host�port�username�password�path�conn� connected�shares� share_names�
share_name� mount_dir�client�exportr' �er r
r
r �test_storage_connection sr �
��
��r@ c
C s� �z#| j dkrLt| j| jd| jdd�}| jdur| jnd}|�| j|�}|s*td��| j� d�d }|�
|�}|� � |j |j
}|j|j
}||fW S | j d
kr�ts[t�d � W d S z7t�� }tj| jd
d�} | �| j�}
|
�|�} t�|�} | �� t�|� | j | j }| j | j }||fW W S ty� }
zt�dt |
�� �� W Y d}
~
W dS d}
~
ww | j dk�rzB| jdur�| jnd}t � }|�| j|� |�!| j| j� z|�"d�}t#|� � d �}|d }W n d}d}Y |�$� ||fW W S t�y }
zt�dt |
�� �� W Y d}
~
W dS d}
~
ww t�d| j � �� W dS t�yF }
zt�d| j%� dt |
�� �� W Y d}
~
dS d}
~
ww ) z�
Get available and total space for a storage target.
Args:
storage (StorageTarget): Storage target object
Returns:
tuple: (free_space, total_space) in bytes
r r Tr Nr � Could not connect to SMB sharer r r r )i @l
r r zError getting NFS space: )r � r r zAVBL /rB � i �>l b zError getting FTP space: r z Error getting storage space for r )&r1 r r4 r5 r2 r3 r r, r6 r �getStorageInfor �
free_units�
block_size� total_unitsr! r r r" r# r$ r% r& r' r( �statvfsr* r+ �f_bfree�f_frsize�f_blocksr- r r. �voidcmd�intr0 r )�storager7 r3 r8 r; � disk_info�
free_space� total_spacer< r= r> r' �statsr? r r
r
r �get_storage_spacew s�
�
��
�� ��rS c
C s �zl| j dkrzt| j| jd| jdd�}| jdur| jnd}|�| j|�}|s*td��| j� d�}|d }t
|�d
krCd� |d
d� �nd }|rO|� d|� �} n|} t |d ��}
|�
|| |
� W d � n1 shw Y |�� | j� d|� �W S | j d
kr�ts�t�d� td��z7t�� } tj| jdd�} | �| j�}
|
�| �}tj� | |�}t�||� |�� t� | � | j� d|� �W W S ty� } z t�dt |�� �� � d}~ww | j dk�rf| jdur�| jnd}t � }|�| j|� |� | j| j� | j�r:z|� | j� W n2 | j�!d�� d�}d }|D ] }|d| 7 }z|� |� W �q |�"|� |� |� Y �qY t |d ��}
|�#d|� �|
� W d � n 1 �sTw Y |�$� | j� d|� �W S td| j � ��� t�y� } zt�d| j%� dt |�� �� � d}~ww )a9
Upload a file to the storage target.
Args:
storage (StorageTarget): Storage target object
local_file (str): Path to local file
remote_filename (str): Filename to use on the remote storage
Returns:
str: Path to the uploaded file on the remote storage
r r Tr Nr rA r r rB � �rbr r r r zError uploading to NFS: r r zSTOR r zError uploading to r )&r1 r r4 r5 r2 r3 r r, r6 r �len�join�open� storeFiler r! r r r" r# r$ r% r& r' r( �shutil�copy2r* r+ r- r r. r/ �strip�mkd�
storbinaryr0 r )rN �
local_file�remote_filenamer7 r3 r8 �partsr; �
remote_dir� remote_path�file_objr< r= r> r' � dest_pathr? r �
path_parts� current_path�partr
r
r �upload_to_storage� s�
� " �
��
� � ��ri )r( �loggingr" rZ r �ftplibr �smb.SMBConnectionr r! �
nfs4_sharer$ � ImportError� getLogger�__name__r �warningr@ rS ri r
r
r
r �<module> s(
�
b r