ruạṛ
a �u:j; � @ s� d dl mZmZmZ d dlmZmZ d dlmZ d dl m Z d dl m Z d dlmZ d dlmZ d dlmZ d d lmZ d d lmZ d d l m Z d d l m!Z! d d l"m#Z# e� Z$G dd� de�Z%G dd� de�Z&G dd� de�Z'G dd� de�Z(G dd� de�Z)G dd� de�Z*G dd� de�Z+G d d � d e�Z,G d d � d e�Z-e$j.d d!d"�e%d#�d$d%��Z/e$j.d&d'd"�e&ed(�d)d*��Ze$j.d+d,d"�e'ed(�d-d.��Ze$j.d/d0d"�e(ed(�d1d2��Ze$j.d3d4d"�e)ed(�d5d6��Z e$j.d7d8d"�e*ed(�d9d:��Z e$j.d;d<d"�e+ed(�d=d>��Ze$j.d?d@d"�e-ed(�dAdB��Z e!e0dC�dDdE��Z1e!e0dC�dFdG��Z2e!e0dC�dHdI��Z3e!e0dC�dJdK��Z4e!e0dC�dLdM��Z5e!e0dC�dNdO��Z6e!e0dC�dPdQ��Z7dRS )S� )�Any�Dict�Union)� APIRouter�BackgroundTasks)� BaseModel)� download_bulk)�generate_mysql_config)�generate_mysql_grants)�generate_ssh_key)�multi_user_backup)�get_mysql_version)�single_user_backup)� upload_bulk)�background_task_wrapper)�run_threaded_taskc @ s e Zd ZU eeef ed< dS )�MySQLVersionRequest� cbs_job_idN)�__name__� __module__� __qualname__r �str�int�__annotations__� r r �O/opt/bart/bart_venv/lib/python3.9/site-packages/app/api/routes_multi_backups.pyr s r c @ s. e Zd ZU eeef ed< eeef ed< dS )�SingleUserBackupRequestr �backup_requestsN� r r r r r r r r r r r r r r s r c @ sN e Zd ZU eeef ed< eeef ed< eeef ed< eeef ed< dS )�MultiUserBackupRequestr ZvolumesZ zfs_pool_idsr Nr r r r r r s r c @ s6 e Zd ZU eeef ed< eeeeef f ed< dS )�GenerateMysqlGrantsRequestr Z generate_mysql_grants_requestsNr r r r r r $ s r c @ sF e Zd ZU eeef ed< eed< eed< eeeeef f ed< dS )�DownloadBulkRequestr � download_url� object_prefixZdownload_requestsN)r r r r r r r r r r r r r! ) s r! c @ s> e Zd ZU eeef ed< eed< eeeeef f ed< dS )�GenerateMysqlConfigRequestr �server_hostnameZ generate_mysql_config_requestsNr r r r r r$ 0 s r$ c @ s6 e Zd ZU eeef ed< eeeeef f ed< dS )�GenerateSshKeyRequestr Zgenerate_ssh_key_requestsNr r r r r r& 6 s r& c @ s& e Zd ZU eed< eeef ed< dS )�UploadBulkRequestItemZfiles_path_prefix�filesN)r r r r r r r r r r r' ; s r' c @ s> e Zd ZU eeef ed< eed< eed< eeef ed< dS )�UploadBulkRequestr Z upload_urlr# Zupload_requestsN) r r r r r r r r r' r r r r r) @ s r) z/mysql_versionzMySQL server version (sync))�summary��requestc � s dt | �� d�iS )N� mysql_version�� request_body)r �dictr+ r r r r- G s r- z/single_user_backupz6Single-user backup (async; CBS callback on completion)�r, Zbackground_tasksc � s |� t| �� � d| jd�S )NzSingle User Backup accepted��messager )�add_task�run_single_user_backup_taskr0 r r1 r r r r L s r z/multi_user_backupz5Multi-user backup (async; CBS callback on completion)c � s |� t| �� � d| jd�S )NzMulti User Backup acceptedr2 )r4 �run_multi_user_backup_taskr0 r r1 r r r r R s r z/generate_mysql_grantsz1Generate mysql.sql per user (async; CBS callback)c � s |� t| �� � d| jd�S )Nz generate_mysql_grants acceptedr2 )r4 � run_generate_mysql_grants_taskr0 r r1 r r r r X s r z/download_bulkz=Bulk download files from object storage (async; CBS callback)c � s |� t| �� � d| jd�S )Nzdownload_bulk acceptedr2 )r4 �run_download_bulk_taskr0 r r1 r r r r ^ s r z/generate_mysql_configz?Write mysql client/mysqldump config files (async; CBS callback)c � s |� t| �� � d| jd�S )Nz generate_mysql_config acceptedr2 )r4 � run_generate_mysql_config_taskr0 r r1 r r r r d s r z/generate_ssh_keyz1Write SSH private key files (async; CBS callback)c � s |� t| �� � d| jd�S )Nzgenerate_ssh_key acceptedr2 )r4 �run_generate_ssh_key_taskr0 r r1 r r r r j s r z /upload_bulkz9Bulk upload files to object storage (async; CBS callback)c � s |� t| �� � d| jd�S )NzBulk Upload acceptedr2 )r4 �run_upload_bulk_taskr0 r r1 r r r r p s r r. c s8 � fdd�}t d�dd�}� d }tt|�d||� d S )Nc s t � �S �N)�single_user_backup_implr r. r r �taskx s z)run_single_user_backup_task.<locals>.task�Z action_datac S s | S r< r r? r r r �success{ s z,run_single_user_backup_task.<locals>.successr r �r0 r r �r/ r> r@ r r r. r r5 v s r5 c s8 � fdd�}t d�dd�}� d }tt|�d||� d S )Nc s t � �S r< )�multi_user_backup_implr r. r r r> � s z(run_multi_user_backup_task.<locals>.taskr? c S s | S r< r r? r r r r@ � s z+run_multi_user_backup_task.<locals>.successr r rA rB r r. r r6 � s r6 c s8 � fdd�}t d�dd�}� d }tt|�d||� d S )Nc s t � �S r< )�generate_mysql_grants_implr r. r r r> � s z,run_generate_mysql_grants_task.<locals>.taskr? c S s | S r< r r? r r r r@ � s z/run_generate_mysql_grants_task.<locals>.successr r rA rB r r. r r7 � s r7 c s8 � fdd�}t d�dd�}� d }tt|�d||� d S )Nc s t � �S r< )�download_bulk_implr r. r r r> � s z$run_download_bulk_task.<locals>.taskr? c S s | S r< r r? r r r r@ � s z'run_download_bulk_task.<locals>.successr r rA rB r r. r r8 � s r8 c s8 � fdd�}t d�dd�}� d }tt|�d||� d S )Nc s t � �S r< )�generate_mysql_config_implr r. r r r> � s z,run_generate_mysql_config_task.<locals>.taskr? c S s | S r< r r? r r r r@ � s z/run_generate_mysql_config_task.<locals>.successr r rA rB r r. r r9 � s r9 c s8 � fdd�}t d�dd�}� d }tt|�d||� d S )Nc s t � �S r< )�generate_ssh_key_implr r. r r r> � s z'run_generate_ssh_key_task.<locals>.taskr? c S s | S r< r r? r r r r@ � s z*run_generate_ssh_key_task.<locals>.successr r rA rB r r. r r: � s r: c s8 � fdd�}t d�dd�}� d }tt|�d||� d S )Nc s t � �S r< )�upload_bulk_implr r. r r r> � s z"run_upload_bulk_task.<locals>.taskr? c S s | S r< r r? r r r r@ � s z%run_upload_bulk_task.<locals>.successr r rA rB r r. r r; � s r; N)8�typingr r r Zfastapir r Zpydanticr Z app.multi_backups.download_bulkr rE Z'app.multi_backups.generate_mysql_configr rF Z'app.multi_backups.generate_mysql_grantsr rD Z"app.multi_backups.generate_ssh_keyr rG Z#app.multi_backups.multi_user_backupr rC Z app.multi_backups.mysql_versionr Z$app.multi_backups.single_user_backupr r= Z app.multi_backups.upload_bulkr rH Z!app.utils.background_task_wrapperr Zapp.utils.threaded_tasksr Zrouterr r r r r! r$ r& r' r) �postr- r0 r5 r6 r7 r8 r9 r: r; r r r r �<module> sh
cải xoăn