mirror of
https://github.com/neondatabase/neon.git
synced 2026-01-08 05:52:55 +00:00
762a8a7bb5
Ruff has "B" class of lints, including B018 which will nag on useless expressions, related to #4719. Enable such lints and fix the existing issues. Most notably: - https://beta.ruff.rs/docs/rules/mutable-argument-default/ - https://beta.ruff.rs/docs/rules/assert-false/ --------- Co-authored-by: Alexander Bayandin <alexander@neon.tech>
30 lines
608 B
Python
Executable File
30 lines
608 B
Python
Executable File
#! /usr/bin/env python3
|
|
|
|
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])
|