@j d dl mZ d dlmZmZ d dlmZmZ d dlm Z m
Z
d dl m Z d dl
mZ d dlmZ d dlmZ G d d
e Zd S ) )uuid4)change_settingsload)config
lveinfolib)history history_x60) history_gov)incident)servers)userc N e Zd ZdZdZdZddZd Zd Zd Z d Z
dd Z d Z d
Z
dS )UpdateOrCreateServerIDzeChange old-style server ID 'localhost' to auto-generated uuid
User can set his own server ID
) localhost
Nc : || _ |p t j | _ dS )zi
:type engine: sqlalchemy.engine.base.Engine
:type config_path: Union[str, None]
N)enginer GLOBAL_CONFIG_LOCATION config_path)selfr r s /builddir/build/BUILDROOT/lve-stats-5.0.5-1.el8.x86_64/opt/cloudlinux/venv/lib/python3.11/site-packages/lvestats/lib/server_id.py__init__z UpdateOrCreateServerID.__init__ s!
&G&*G c t | j }| d S # t $ r }t d| Y d}~dS d}~ww xY w)*
:rtype: Union[None, str]
)path server_idError: N)r r get Exceptionprint)r cfges r _get_current_server_id_cfgz1UpdateOrCreateServerID._get_current_server_id_cfg" sl D ,---C77;'' ' -A-- 44444 s ),
AAAc t d|i| j |S # t $ r }t d| Y d}~dS d}~ww xY w)zmSets option 'server_id' in config file
:type server_id: str
:rtype: Union[None, str]
r r N)r r r r! )r r r# s r _set_server_id_cfgz)UpdateOrCreateServerID._set_server_id_cfg- se
[) 4d6F G G G -A-- 44444 s
A=Ac |j |j |k }| | d|i dS )a
updates server_id in one of the tables
:type conn: sqlalchemy.engine.base.Connection
:type table: Union[history, history_gov, incident, user, history_x60]
:type current_server_id: str
:type new_server_id: str
r N) __table__updatewherer executevalues)r conntablecurrent_server_id
new_server_id
sql_updates r _update_server_id_tablez.UpdateOrCreateServerID._update_server_id_table9 sn _++--33 O 0 0
Z&&
- (
r c 6 t t t t t g}t j | j | }| j }| } |D ]}|
|||| | t j
||d | t j
t j |k | | |S # t* $ rE}| | t/ d| Y d}~dS d}~ww xY w)z
updates server_id for all tables in the database
:type current_server_id: str
:type new_server_id: str
:rtype: Union[None, str]
)r lve_versionr N)r r
r r
r r get_lve_versionr connectbeginr2 r+ r r( insertr, deleter* r commitcloser rollbackr! ) r r/ r0
sql_tablesr4 r- txr. r# s r _update_server_id_dbz+UpdateOrCreateServerID._update_server_id_dbH sy {HdKH
0>OPP {""$$
ZZ\\ !#
\
\,,T5:K][[[[
LL*1133:: +KHH
LL*113399':KO`:`aa b b b
IIKKK JJLLL KKMMM JJLLL -A-- 44444 s )B6E
F:FFTc ( t t d| j }|rt d| d }t | |r| }|| j v rt d| d dS ||k rt d| d |S t | dk rt d dS | }|
|s || j vr|S | |p|pd | r|
| S |S )
zw
:type current_server_id: Union[None, str]
:type prompt: bool
:rtype: Union[None, str]
NzEnter new server ID ['z']: Server ID cannot be ''z Server ID is already set to ' z(Server ID length should be less than 255r ) strr server_id_lengthinputr! striprestricted_server_id_listlenr$ r? r& )r promptr/ r res
cur_server_ids r _resolve_localhostz)UpdateOrCreateServerID._resolve_localhostf sN LL!7$"7!78
@@@@AAC #JJJ
( IIKK D ::::i:::;;;t ---BiBBBCCC 9~~##@AAAt 7799
$V $ T%CCC $ $%6%V-%V;Xa b b 6**955 5r c | }t |t r.| | j v rt d| d | | S )r rA rB )r/ )r$
isinstancerD rG rH r! rM )r cur_cfg_server_ids r rJ z UpdateOrCreateServerID.prompt sw ! ; ; = = ' - - @!''))T-KKK >*;>>> ? ? ?&&9J&KKKr c . | d S )r F)rJ )rM )r s r autozUpdateOrCreateServerID.auto s &&e&444r )N)TN)__name__
__module__ __qualname____doc__rH rE r r$ r& r2 r? rM rJ rR r r r r s !/ H H H H
! ! !< : L L L5 5 5 5 5r r N)uuidr clcommon.clconfparsr r lvestats.libr r lvestats.orm.historyr r lvestats.orm.history_govr
lvestats.orm.incidentr lvestats.orm.serversr lvestats.orm.userr
objectr rW r r ra s 5 5 5 5 5 5 5 5 + + + + + + + + 5 5 5 5 5 5 5 5 0 0 0 0 0 0 * * * * * * ( ( ( ( ( ( " " " " " "@5 @5 @5 @5 @5V @5 @5 @5 @5 @5r