gMAdZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl m Zddl mZdd l mZdd lmZdd lmZejd vrdd lmZejZdZnddlZejZdZGddejZGddeZ Gdde Z!GddeZ"GddeZ#GddeZ$Gddej%Z&dZ'dZ(d Z)d!Z*d"d#d$d%Z+Gd&d'e j,Z,dS)(z Create, control, and destroy PostgreSQL clusters. postgresql.cluster provides a programmer's interface to controlling a PostgreSQL cluster. It provides direct access to proper signalling interfaces. NNamedTemporaryFile)api) configfile) installation) exceptions)driver)get_python_name) close_fds)win32win64) signal1_mswc&tdd|S)NFw)deletemodeencodingrrs H/granary/edev/quantum/lib/python3.11/site-packages/postgresql/cluster.py namedtemprs U3 J J JJc$td|S)Nr)rrrrs rrr!s 3 : : ::rceZdZdZdZdZdS) ClusterErrorz General cluster error. z-C000CLUSTERN__name__ __module__ __qualname____doc__codesourcerrrr$s$  rrceZdZdZdZdS)ClusterInitializationErrorz&General cluster initialization failurez-CiniNrrr r!r"r$rrr&r&*s))rr&ceZdZdZdZdS) InitDBErrorz4A non-zero result was returned by the initdb commandz-CidbNr'r$rrr)r)-s77rr)ceZdZdZdZdS)ClusterStartupErrorzCluster startup failedz-CbotNr'r$rrr+r+0rr+ceZdZdZdZdS)ClusterNotRunningErrorzCluster is not runningz-CdwnNr'r$rrr.r.3r,rr.ceZdZdZdZdS)ClusterTimeoutErrorzCluster operation timed outz-CoutNr'r$rrr0r06srr0ceZdZdZdZdZdS)ClusterWarningz$Warning issued by cluster operationsz-CwrnrNrr$rrr2r2:s'' rr2utf-8zpostgresql.confz pg_hba.confzpostmaster.pidz-Ez-Az-U)rauthenticationuserceZdZdZejZdZdZe Z e Z e Z e Z e dZdZe dZdZe dZe dZe ejjfd Z d0dZdjfdZdZdZ d1 d2dZdZ d3 d4dZd5dZ dZ!dZ"d Z#d!Z$d"Z%d#Z&d$Z'd%Z(d&Z)d'Z*d(Z+ d6 d7d.Z, d6 d7d/Z-dS)8ClusteraO Interface to a PostgreSQL cluster. Provides mechanisms to start, stop, restart, kill, drop, and initalize a cluster(data directory). Cluster does not strive to be consistent with ``pg_ctl``. This is considered to be a base class for managing a cluster, and is intended to be extended to accommodate for a particular purpose. Nc~|rdStj|jsdSdS)Nrunningvoidstopped)r9ospathexistsdata_directoryselfs rstatez Cluster.state`s: \\^^ ) + , , & rc#VK|j}dd|zdzfV|dkr d|jfVdSdS)N[]r9pid)rB)rArBs r_e_metaszCluster._e_metashsS *% sU{S !!!! i  rc2|jjp |jjS)z; Path to the executable to use to startup the cluster. )r postmasterpostgresr@s r daemon_pathzCluster.daemon_pathns   % C):)CCrcf tj|j|j}t |5}t |cdddS#1swxYwYdS#t$r0}|j tj tj fvrYd}~dSYd}~dSd}~wwxYw)z3 The current pid from the postmaster.pid file. N) r<r=joinr?DEFAULT_PID_FILENAMEopenintreadlineIOErrorerrnoEIOENOENT)rAr=fes rget_pid_from_filezCluster.get_pid_from_fileus ',,t*D,E F F4 T a qzz||   g%)U\*** 44444+*****s:9A6!A) A6)A--A60A-1A66 B0B++B0cR|}||j}||jS|S)z; If we have the subprocess, use the pid on the object. )rXdaemon_processrF)rArFds rrFz Cluster.pids3   #[ 1m 5L *rclt|dstj|j|_|jS)N _settings)hasattrr ConfigFilepgsql_dot_confr]r@s rsettingszCluster.settingss1 { # #?)$*=>>4> rc`|jd||j|jS)z. The path to the HBA file of the cluster. hba_file)ragetr?DEFAULT_HBA_FILENAME)rArMs rrczCluster.hba_files6    4T677  rrinstallation objectr?path to the data directoryc||_tj||_tj|j|j|_d|_d|_ dSN) rr<r=abspathr?rMDEFAULT_CONFIG_FILENAMEr`rZdaemon_command)rArr?s r__init__zCluster.__init__s^#$77$ $$$rz{mod}.{name}({ins!r}, {dir!r})cz|t|jt|j|j|jSri)typerrrr?)rAformats r__repr__zCluster.__repr__s: ::::   rcV|||S)z3 Start the cluster and wait for it to startup. )startwait_until_startedr@s r __enter__zCluster.__enter__s(**,,, +rcV||dS)z3 Stop the cluster and wait for it to shutdown. N)stopwait_until_stopped)rAtypvaltbs r__exit__zCluster.__exit__s)))+++rpassword>Password to assign to the cluster's superuser(`user` keyword).c j|jj}||jjdf}n|f}d|vrtdddi||d|jg}|D]]}|dvr|t vrtd ||t ||||^| d p tj }td | d d D}d }d} d} |Wtt|d} | || d| jzf}|d|jfzt|z|z|z} tj| t(dtj |tj } | j | } | j| jnv#t2$r } | jt4jkrYd} ~ nd} ~ wwxYw | j| jn&#| j| jwwxYw| dkr| j} |d}nK#t@$r>tBj"#d|$D}YnwxYwtKd| ||d| | -| j| j| jfD]}||| P| j}| tBj&'|rtCj(|dSdSdS#| -| j| j| jfD]}||| O| j}| tBj&'|rtCj(|wwwxYw)z Create the cluster at the given `data_directory` using the provided keyword parameters as options to the command. `command_option_map` provides the mapping of keyword arguments to command options. Ninitdbz4unable to find executable for cluster initializationdetailz4The installation does not have 'initdb' or 'pg_ctl'.detailscreatorr)logfileextra_argumentsz#got an unexpected keyword argument rc,g|]}t|Sr$)str.0xs r z Cluster.init..s+3q66rrr$rz --pwfile=-Di)r bufsizestdinstdoutstderrTrr3c<g|]}t|ddS))reprrs rrz Cluster.init..s3d1ggadmrz"initdb exited with non-zero status)commandrr))rrpg_ctlr& setdefaultDEFAULT_CLUSTER_ENCODINGinitdb_option_map TypeErrorappendrdspPIPEtuplerr writeflushnamer?Popenr rclosewaitrOSErrorrSEINTRrreadstripdecodeUnicodeDecodeErrorr<lineseprM splitlinesr)r=r>unlink)rAr}kwroptsrr extra_args supw_filesupw_tmppcmdrcrWrmsgns rinitz Cluster.inits   #& ^   %x 166 I6 V^^ #: F    -- D9::: $ a )))    )aaA B BB;; #$$$;;r!u FF9   ('66+R00*) ( !COBzN$C$CDDDH NN8 NNx},.I 4,- - Dkk   3  x G  W17==???  &&((R hnn     5;    hnn hnn   Agg  A 88G  SS  :??\\^^  SSS )    *mh * ggiii A NN w~~aYq\\\\\ mh * ggiii A NN w~~aYq\\\\ so CN/H )!N/ H4H/*I/H44I8!N/#I<<5N/2KN/AL N/LN//BP2c|r| |np#t$rc| |n6#t$r)t dddi|YnwxYwYnwxYwtj|j dD]t\}}}|D]4}tj tj ||5|D]4}tj tj ||5utj |j dS)z8 Stop the cluster and remove it from the filesystem z%cluster failed to shutdown after killhintz#Shared memory may have been leaked.rF)topdownN)r9shutdownrxr0killr2emitr<walkr?remover=rMrmdir)rArootdirsfilesrs rdropz Cluster.drop6sx \\^^ ==???     IIKKK  -?@ tvvvvv  74#6%HHH''dD%((tIbgll4&&''''''tHRW\\$ % %&&&&'(4 s3?B,A32B,30B&#B,%B&&B,+B,rWhere to send stderrraMapping of runtime parametersc|rdS|jd|jf}|Qt|D]/\}}|d||0tj|td| tj n|tj tj }||j |j ||_||_dS)z Start the cluster. Nrz --{k}={v})kvi)r rrrr)r9rKr?dictitemsrrprrr rSTDOUTrrrrZrl)rArrarrrrs rrsz Cluster.startPs \\^^  6 4!45# 8nn""$$,,sqJJ{!!A!**++++h BGGG I 7! _8>>'--///$$r c&|r*||||rtd||||||dS)a Restart the cluster gracefully. This provides a higher level interface to stopping then starting the cluster. It will perform the wait operations and block until the restart is complete. If waiting is not desired, .start() and .stop() should be used directly. )timeoutzfailed to shutdown clusterr)rraN)r9rwrxrrsrt)rArrars rrestartzCluster.restartms \\^^.99;;;W--- \\^^     **w8*444G,,,,,rc|j}|K t|tjdS#t$r!}|jt jkrYd}~dSd}~wwxYwdS)z: Signal the cluster to reload its configuration file. N)rFpg_killsignalSIGHUPrrSESRCHrArFrWs rreloadzCluster.reloady #_  C    w%+  _' AA  Ac|j}|K t|tjdS#t$r!}|jt jkrYd}~dSd}~wwxYwdS)zM Stop the cluster gracefully waiting for clients to disconnect(SIGTERM). N)rFrrSIGTERMrrSrrs rrwz Cluster.stopsy #_  C        w%+  _rc|j}|K t|tjdS#t$r!}|jt jkrYd}~dSd}~wwxYwdS)zF Shutdown the cluster as soon as possible, disconnecting clients. N)rFrrSIGINTrrSrrs rrzCluster.shutdownrrc|j}|K t|tjdS#t$r!}|jt jkrYd}~dSd}~wwxYwdS)zM Stop the cluster immediately(SIGKILL). Does *not* wait for shutdown. N)rFrrSIGKILLrrSrrs rrz Cluster.killsy #_  C        w%+  _rctj|jrhtj|jrDtjtj|jdrdSdS)zH Whether or not the data directory *appears* to be a valid cluster. baseTF)r<r=isdirr?r>r`rMr@s r initializedzCluster.initializedskW]]4&'''..$%%'-- T0&99:: $ rcx|jP|j}|3|}|d|_|SdSdS|}|dS t |t jn.#t$r!}|jtj krYd}~dSd}~wwxYwdS)zm Whether or not the postmaster is running. This does *not* mean the cluster is accepting connections. NFT) rZpollrXr9rrSIG_DFLrrSr)rArrFrWs rr9zCluster.runnings  $ !!1m  " "C T LLNN 5 4   ! !3 k 5 C    w%+ 55555 $s1B B7B22B7c b|\}}|jjd|pd|pdd|S)z Create a postgresql.driver connector based on the given keywords and listen_addresses and port configuration in settings. localhosti8)hostportr$)addressr fit)rArrrs r connectorzCluster.connectorsP ||~~*$    + ,$    rc ,|jdi|S)zD Create a connection object to the cluster, but do not connect. r$)r)rArs r connectionzCluster.connections$   "    rc |std||jdi|}||S)zV Create an established connection from the connector. Cluster must be running. z(cannot connect if cluster is not runningrr$)r9r.rconnect)rArrs rrzCluster.connects\  .   do!))+++ (rcJ|jd}|d\|dd}d|vsd|vrd}nd|vrd}nd|vrd}n |d }nd}||d fS) z4 Get the host-port pair from the configuration. )listen_addressesrrN,r*z 127.0.0.1z::1rr)ragetsetrdlowersplit)rAr[addrsrs rrzCluster.addresss m!UU * 55# $ $ * * , , 2 23 7 75UcUll DDu DD  DD 8DD 4 f rc|sdSd}|\}}|jd||dd} |n#t j$r}|jjD]}|j }|j j dddkr.t|t j t jfrYd}~d Sn!t|t jrYd}~d St|t jt jfr|t|t jr|}n|}Yd}~nd}~wwxYw||nd S) z~ If the daemon is running, and is not in startup mode. This only works for clusters configured for TCP/IP connections. FNz -*- ping -*- template1disable)r5rrdatabasesslmoder)rT)r9rr rrpg_excClientCannotConnectErrorrfailureserrorr version_info isinstanceUndefinedObjectError AuthenticationSpecificationErrorCFError ProtocolErrorServerNotReadyError)rArWrrrerrattemptrs rready_for_connectionszCluster.ready_for_connectionss  % !||~~*$t{  t  *     (   ,'  w A %bqb)E111 ! -   TTTTTT  1f=>> TTTTTT!fnf&:;<< W!V/00 Q U  A) .m%s&A--E >D ikkEW$$    :e $r)rrfr?rgri)r}r~)NN)rrrar)NNr)rr)rrrr).rrr r! pg_driverdefaultr rr?rrkrNrepropertyrBrGrKrXrFrar<r=rMrcrmrprqrur|rrrsrrrwrrrr9rrrrrrtrxr$rrr7r7Ms    42,,  (  DD (D        (    (  7< ( & /    >D-1s*ssssj   6&*/3 " ,:----*                :         .)&)&)&XFH262 B2 /2222jFH26 B /rr7)-r!sysr<rSr subprocessrtempfilerrpg_apirrpg_innr rr rencodings.aliasesr python.osr platformrrrrrrErrorrr&r)r+r.r0Warningr2rrkrerNrr7r$rrr-s  ''''''$$$$$$""""""!!!!!!...... <%%%(((((( ;KKKK 7;;;6< ,,\,V^ #+$' ~~~~~fn~~~~~r