File: //usr/local/CyberCP/highAvailability/__pycache__/haManager.cpython-310.pyc
o
��h�! � @ s� d dl Zd dlZd dlZej�d� ej�dd� e�� d dl Z d dl
Z
d dl m Z
d dlmZ d dlmZ d dlmZ d dlZd dlZd d lmZ d d
lmZ G d d � d ej�Z d
d� Z e dkrge � dS dS )� Nz/usr/local/CyberCP�DJANGO_SETTINGS_MODULEzCyberCP.settings)�CyberCPLogFileWriter)�httpProc)�ProcessUtilities)�
ACLManager)�FirewallUtilities)�
FirewallRulesc @ sN e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dS )� HAManagerNc C s" t j�| � || _|| _|| _d S )N)�multi�Thread�__init__�request�data�function)�selfr
r r � r �0/usr/local/CyberCP/highAvailability/haManager.pyr s
zHAManager.__init__c
C sp z| j dkr
| �� W d S | j dkr| �� W d S W d S ty7 } zt�t|�d � W Y d }~d S d }~ww )N� setupNode�
addManagerz [HAManager.run])r r �
BaseException�logging� writeToFile�str)r �msgr r r �run s
� ��z
HAManager.runc
C s� z�t �� t jkst �� t jkr d}t�| jd |� W dS | jjd }t �
|�}|d dkr<d}t�| jd |� W dS t�| jd d� | jd }|D ]-}zt j |d d
�}|dkrat�
|d � W qL tyy t�| jd |d � Y W dS w zt�d
dd� tddd
d�}|�� W n Y d}t�| jd |� W d S ty� } zt�
t|�� t�| jd t|�d � W Y d }~d S d }~ww )Nz2Clusters are only supported on Ubuntu 18.04. [404]�tempStatusPathr �userID�adminz.Only administrators can create clusters. [404]z Setting up node in progress..�commandsT��shell� Failed.z Failed. [404]�tcp�2377z 0.0.0.0/0�Docker)�name�port�protoz#Node successfully configured. [200]z. [404])r � decideDistro�centos�cent8r � statusWriterr r
�sessionr � loadedACL�
subprocess�callr r r �addRuler �saver ) r �mesgr �
currentACLr �command�result�fwRuler r r r r $ sH
�
� (��zHAManager.setupNodec
C sn z�t | jd d �}| jjd }t�|�}|d dkr |�dd�W S d}t|�}|�� }|�dd�d }d| }t �
|� d } d }
d
}t � t
�|���d ��� } | D ]} | �d �d
kr`| �d�d } qPd}t � t
�|���d ��� } | D ]} | �d �d
kr�| �d�d }
qri }
| |
d< |
|
d< |�dd |
�W S ty� } zt | jd d �}|�dd t|��W Y d }~S d }~ww )Nr r r �(Only administrators can create clusters.z/etc/cyberpanel/machineIP�
� z(sudo docker swarm init --advertise-addr � z$sudo docker swarm join-token managerzutf-8z--token���� ���z#sudo docker swarm join-token worker� managerToken� workerToken)r r
r, r r- �ajax�open�read�splitr � executionerr. � check_output�shlex�decode�
splitlines�findr r )r �procr r3 �ipFile�f�ipData� ipAddressr4 r> r? �output�itemsr r r r r �fetchManagerTokensO sF
�� ��z HAManager.fetchManagerTokensc
C s� zHt | jd d �}| jjd }t�|�}|d dkr |�dd�W S | jd }| jd }d| d | d }t�|�dkrB|�dd
�W S |�d d �W S t yj } zt | jd d �}|�dd t
|��W Y d }~S d }~ww ) Nr r r r7 �tokenrN z sudo docker swarm join --token r< z:2377zFailed to join as worker.r9 ) r r
r, r r- r@ r r rD r r )r rJ r r3 rR rN r4 r r r r � addWorker} s"
��zHAManager.addWorkerc C s� z\| j jd }t�|�}t| j d d �}|d dkr |�dd�W S | jd }|D ]2}ztj|dd�}|dkr<t �
|d � W q' t yY } zt �
|d � W Y d }~ W dS d }~ww W d S t yv } zt �
t |�� W Y d }~d S d }~ww )
Nr r r r7 r Tr r! zFailed.)
r
r, r r- r r@ r r. r/ r r r r )r r r3 rJ r r4 r5 r r r r �
leaveSwarm� s.
���� ��zHAManager.leaveSwarmc
C s� zM| j jd }t�|�}t| j d d �}|d dkr |�dd�W S d}tj�|�s-t� |� |d }t
|d�}| j d D ]}|� |� q;|�
� |�d d �W S tyu } z t�t|�� t| j d d �}|�dt|��W Y d }~S d }~ww )
Nr r r r7 � /home/cyberpanel/composePath�/docker-compose.yml�w� composeDatar9 )r
r, r r- r r@ �os�path�exists�mkdirrA r �
writelines�closer r r r ) r r r3 rJ � composePath� composeFile�composerP r r r r �
setUpDataNode� s*
��zHAManager.setUpDataNodec
C sZ z�| j jd }t�|�}t| j d d �}|d dkr |�dd�W S d}|d }t|d��� }t|d�}|D ]E}|�d �d
krL|� d t
| j d � d
� q4|�d�d
kr`|� d| j d d
� q4|�d�d
krt|� d| j d d � q4|� |� q4|� � |�dd �W S t
y� } z t�t
| �� t| j d d �}|�dt
| ��W Y d } ~ S d } ~ ww )Nr r r r7 rU rV �rrW �replicasr; z replicas: �
containersr8 �memoryz memory: � containerRamzcpus:z cpus: "� containerCPUz"
r9 )r
r, r r- r r@ rA � readlinesrI r] r r r^ r r r )
r r r3 rJ r_ r` r ra rP r r r r �submitEditCluster� s4
��zHAManager.submitEditCluster)NNN) �__name__�
__module__� __qualname__r r r rQ rS rT rb rj r r r r r s
+. r c C sB t jdd�} | jddd� | jddd� | jdd d� | �� }d S )
NzCyberPanel HA Manager)� descriptionr z Specific a function to call!)�helpz--idzID!z --ipAddressz IP Address!)�argparse�ArgumentParser� add_argument�
parse_args)�parser�argsr r r �main� s
rv �__main__) �os.pathrY �sys�djangorZ �append�environ�
setdefault�setupr. rF � plogical.CyberCPLogFileWriterr r �plogical.httpProcr �plogical.processUtilitiesr � plogical.aclr � threadingr
rp �plogical.firewallUtilitiesr �firewall.modelsr r r rv rk r r r r �<module> s* Y
�