#j 6dZddlmZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl mZdZGdd eZGd d eZGd d eZGddeZe GddZd8dZd9dZd:dZddd;dZGd d!Zdd?d/Zd@dAd1ZdBdCd3Zd@dDd4Z dEdFd6Z!eZ"gd7Z#dS)Gz pg0 - Embedded PostgreSQL for Python Usage: from pg0 import Pg0 # Start PostgreSQL pg = Pg0() pg.start() print(pg.uri) pg.stop() # Or use context manager with Pg0() as pg: print(pg.uri) ) annotationsN) dataclass)Path)Optionalz0.1.0ceZdZdZdS)Pg0ErrorzBase exception for pg0 errors.N__name__ __module__ __qualname____doc__T/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/pg0/__init__.pyrr"s((DrrceZdZdZdS)Pg0NotFoundErrorz0pg0 binary not found and could not be installed.Nr rrrrr's::DrrceZdZdZdS)Pg0NotRunningErrorz#PostgreSQL instance is not running.Nr rrrrr,s--DrrceZdZdZdS)Pg0AlreadyRunningErrorz'PostgreSQL instance is already running.Nr rrrrr1s11DrrceZdZUdZded<ded<dZded<dZded <dZd ed <dZd ed <dZ d ed <dZ d ed<dZ d ed<e ddZ dS) InstanceInfoz(Information about a PostgreSQL instance.strnameboolrunningN Optional[int]pidport Optional[str]versionusernamedatabasedata_diruridatadictreturn'InstanceInfo'c ||dd|dd|d|d|d|d|d |d |d  S) NrdefaultrFrrr!r"r#r$r%) rrrrr!r"r#r$r%)get)clsr&s r from_dictzInstanceInfo.from_dictCss&),,HHY..&!!HHY''XXj))XXj))XXj))    r)r&r'r(r))r r r r __annotations__rrr!r"r#r$r% classmethodr.rrrrr6s22 IIIMMMCD!G!!!!"H"""""H"""""H""""C    [    rrr(Optional[Path]cttj}tjdkrdnd}|dz |z }|r|SdS)z5Get the path to the bundled pg0 binary, if it exists.win32pg0.exepg0binN)r__file__parentsysplatformexists) package_dir binary_name bundled_paths r_get_bundled_binaryr?RsSx..'K"|w66))EK&4L 4rrctjdkrLttjdtjdz dz }|dz dz Stjdz dz S)z7Get the directory where pg0 binary should be installed.r3 LOCALAPPDATAAppDataLocalr5r6z.local)r9r:rosenvironr,home)bases r_get_install_dirrH\sc |wBJNN>49;;3JW3TUUVVe|e##y{{X%--rrc"t}|rt|Stjd}|r|St }t jdkrdnd}||z }|rt|Std)z3Find the pg0 binary or raise an error if not found.r5r3r4zpg0 binary not found. Install it with: curl -fsSL https://raw.githubusercontent.com/vectorize-io/pg0/main/install.sh | bash Or download from: https://github.com/vectorize-io/pg0/releases) r?rshutilwhichrHr9r:r;r)bundledpath install_dirr= binary_paths r _find_pg0rPfs"##G7|| <  D  #$$K"|w66))EK +K ;  I  rTcheckargsrRrsubprocess.CompletedProcessct} tjdkrtj5}tj5}t j|g|||}|d|d| dd}| dd}dddn #1swxYwYdddn #1swxYwYt j |g||||}nt j |g|d d }|r|j dkr|j }d |vrt!|d |vsd |vrt#|t%|p d|j |S#t&$rt)dwxYw)zRun a pg0 command.r3)stdoutstderrrzutf-8replace)errorsN)rS returncoderVrWT)capture_outputtextzalready runningzno running instancez not runningzpg0 command failed with code zpg0 binary not found)rPr9r:tempfile TemporaryFile subprocesscallseekreaddecodeCompletedProcessrunrZrWstriplowerrrrFileNotFoundErrorr) rRrSpg0_pathout_ferr_frcrVrWresults r_run_pg0rns{{H(7 <7 " "')) HUH4J4L4L HPU_%%    1  1 ,,WY,GG,,WY,GG H H H H H H H H H H H H H H H H H H H H H H H H H H H H H H 0&& FF ^!D!#F  ^V&!++]((**F FLLNN22,V444&&,,..88MV\\^^<[<[(000v\)\IZ)\)\]]] 77756667sT$G$DBC+ D+C/ /D2C/ 3D6 G$DG$ D CG$$G>ceZdZdZ d'd(dZd)dZd*dZd+d,dZd)dZe d-dZ e d.dZ d/dZ d0d!Z d1d2d#Zd3d%Zd*d&ZdS)4Pg0a Embedded PostgreSQL instance. Args: name: Instance name (allows multiple instances) port: Port to listen on (None = auto-select available port) username: Database username password: Database password database: Database name data_dir: Custom data directory config: Dict of PostgreSQL configuration options Example: # Simple usage (auto-selects available port) pg = Pg0() pg.start() print(pg.uri) pg.stop() # Context manager with specific port with Pg0(port=5433, database="myapp") as pg: print(pg.uri) # Custom config pg = Pg0(config={"shared_buffers": "512MB"}) r+Npostgresrrrrr"passwordr#r$r configOptional[dict[str, str]]cl||_||_||_||_||_||_|pi|_dSN)rrr"rrr#r$rs)selfrrr"rrr#r$rss r__init__z Pg0.__init__s=          l rr(rc dd|jd|jd|jd|jg }|j)|dt |jg|jr|d|jg|j D]!\}}|d |d |g"t|| S) z Start the PostgreSQL instance. Returns: InstanceInfo with connection details Raises: Pg0AlreadyRunningError: If instance is already running Pg0Error: If start fails start--namez --usernamez --passwordz --databaseNz--portz --data-dir-c=) rr"rrr#rextendrr$rsitemsrninfo)rwrSkeyvalues rrzz Pg0.starts  di $- $- $-   9 KK3ty>>2 3 3 3 = 7 KKt}5 6 6 6+++-- 2 2JC KK#////0 1 1 1 1$yy{{rNonec6tdd|jddS)zv Stop the PostgreSQL instance. Note: Does not raise an error if the instance is not running. stopr{FrQNrnrrws rrzPg0.stops# 49E::::::rTforcerc`dd|jg}|r|dt|ddidS)a Drop the PostgreSQL instance (stop if running, delete all data). Args: force: Skip confirmation prompt (default True for programmatic use) Warning: This permanently deletes all data for this instance! dropr{--forcerRFN)rappendrn)rwrrSs rrzPg0.dropsE$),  # KK " " "$$e$$$$$rctdd|jddd}tj|j}t |S)z Get information about the PostgreSQL instance. Returns: InstanceInfo with current status and connection details rr{-ojsonFrQ)rnrrloadsrVrr.)rwrmr&s rrzPg0.infosE&(DItV5QQQz&-((%%d+++rc4|jS)z"Get the connection URI if running.)rr%rs rr%zPg0.urisyy{{rc4|jS)z!Check if the instance is running.)rrrs rrz Pg0.running$syy{{""rrSrTc*tdd|jg|RS)a6 Run psql with the given arguments. Args: *args: Arguments to pass to psql (e.g., "-c", "SELECT 1") Returns: CompletedProcess with stdout/stderr Example: result = pg.psql("-c", "SELECT version();") print(result.stdout) psqlr{r)rwrSs rrzPg0.psql)s $);d;;;;rsqlc<|d|}|jS)z Execute a SQL command and return the output. Args: sql: SQL command to execute Returns: Command output as string Example: output = pg.execute("SELECT version();") r|)rrV)rwrrms rexecutez Pg0.execute9s4%%}rlinescdd|jg}|$|dt|gt|ddi}|jS)z Get PostgreSQL logs for this instance. Args: lines: Number of lines to return (None = all logs) Returns: Log content as string Example: print(pg.logs(50)) # Last 50 lines logsr{N-nrRF)rr~rrnrV)rwrrSrms rrzPg0.logsIsO$),   KKs5zz* + + +4-u--}r'Pg0'c.||S)z*Context manager entry - starts PostgreSQL.)rzrs r __enter__z Pg0.__enter__\s  rcR |dS#t$rYdSwxYw)z(Context manager exit - stops PostgreSQL.N)rr)rwexc_typeexc_valexc_tbs r__exit__z Pg0.__exit__as:  IIKKKKK!    DD s  &&)r+NrqrqrqNN)rrrrr"rrrrr#rr$r rsrt)r(r)r(r)T)rrr(r)r(r )r(r)rSrr(rT)rrr(rrv)rrr(r)r(r)r r r r rxrzrrrpropertyr%rrrrrrrrrrprpsL:"""""&+/#####$>;;;; % % % % % , , , ,X###X#<<<<  & rrplist[InstanceInfo]crtdddd}tj|j}d|DS)zd List all pg0 instances. Returns: List of InstanceInfo for all known instances listrrFrQcBg|]}t|Sr)rr.).0items r z"list_instances..rs& : : :TL " "4 ( ( : : :r)rnrrrV)rmr&s rlist_instancesris?fdF% 8 8 8F :fm $ $D : :T : : ::r list[str]ctdd}|jd}d|DS)z List available PostgreSQL extensions. Returns: List of available extension names Example: extensions = pg0.list_extensions() print(extensions) # ['vector', 'postgis', ...] zlist-extensionsFrQ c^g|]*}||+Sr)rf)rlines rrz#list_extensions..s- ; ; ;Tdjjll ;DJJLL ; ; ;r)rnrVrfsplit)rmrs rlist_extensionsrusL'u 5 5 5F M   ! ! ' ' - -E ; ;U ; ; ;;rr+rqrrrr"rrr#rsc Tt||||||}|S)a Start a PostgreSQL instance (convenience function). Args: name: Instance name port: Port to listen on (None = auto-select available port) username: Database username password: Database password database: Database name **config: PostgreSQL configuration options Returns: InstanceInfo with connection details Example: info = pg0.start(shared_buffers="512MB") # auto-selects port print(info.uri) )rrr"rrr#rs)rprz)rrr"rrr#rspgs rrzrzs:4      B 88::rrc,tdd|ddS)zk Stop a PostgreSQL instance (convenience function). Args: name: Instance name to stop rr{FrQN)rn)rs rrrs! VXt5111111rrcVdd|g}|r|dt|ddidS)a; Drop a PostgreSQL instance (convenience function). Stops the instance if running and deletes all data. Args: name: Instance name to drop force: Skip confirmation prompt (default True for programmatic use) Warning: This permanently deletes all data for this instance! rr{rrRFN)rrn)rrrSs rrrsC Hd #D  I d %     rctdd|ddd}tj|j}t|S)z Get information about a PostgreSQL instance (convenience function). Args: name: Instance name Returns: InstanceInfo with current status rr{rrFrQ)rnrrrVrr.)rrmr&s rrrsCfhdF% H H HF :fm $ $D  ! !$ ' ''rrc~dd|g}|$|dt|gt|ddi}|jS)z Get PostgreSQL logs for an instance (convenience function). Args: name: Instance name lines: Number of lines to return (None = all logs) Returns: Log content as string rr{NrrRF)r~rrnrV)rrrSrms rrrsM Hd #D  T3u::&''' t )5 ) )F =r)rp PostgreSQLrrrrrrrrzrrrrr?)r(r1)r(r)r(r)rSrrRrr(rT)r(r)r(r)r+Nrqrqrq)rrrrr"rrrrr#rrsrr(r)r+)rrr(r)r+T)rrrrr(r)rrr(r)r+N)rrrrr(r)$r __future__rrrDrJr_r]r9 dataclassesrpathlibrtypingr __version__ Exceptionrrrrrr?rHrPrnrprrrzrrrrr__all__rrrrs"#"""""  !!!!!!      y        x                X             6....8(,+7+7+7+7+7+7\vvvvvvvvr ; ; ; ; < < < <" """""J22222!!!!!& ( ( ( ( (&    r