mirror of
https://github.com/neondatabase/neon.git
synced 2025-12-27 16:12:56 +00:00
5bd8e2363a
This will help to keep us from using deprecated Python features going forward. Signed-off-by: Tristan Partin <tristan@neon.tech>
32 lines
644 B
Python
Executable File
32 lines
644 B
Python
Executable File
#! /usr/bin/env python3
|
|
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import os
|
|
|
|
import asyncpg
|
|
|
|
|
|
async def run(**kwargs) -> asyncpg.Record:
|
|
conn = await asyncpg.connect(
|
|
**kwargs,
|
|
statement_cache_size=0, # Prepared statements doesn't work pgbouncer
|
|
)
|
|
rv = await conn.fetchrow("SELECT 1")
|
|
await conn.close()
|
|
|
|
return rv
|
|
|
|
|
|
if __name__ == "__main__":
|
|
kwargs = {
|
|
k.removeprefix("NEON_").lower(): v
|
|
for k in ("NEON_HOST", "NEON_DATABASE", "NEON_USER", "NEON_PASSWORD")
|
|
if (v := os.environ.get(k, None)) is not None
|
|
}
|
|
|
|
row = asyncio.run(run(**kwargs))
|
|
|
|
print(row[0])
|