?HAMA

Navigation

Breadcrumbs

Create

Edit File Size: 1.96 KB

//opt/cloudlinux/venv/lib/python3.11/site-packages/sqlalchemy/dialects/postgresql/provision.py

Text Base64

import time

from ... import exc
from ... import text
from ...testing.provision import create_db
from ...testing.provision import drop_db
from ...testing.provision import log
from ...testing.provision import temp_table_keyword_args

@create_db.for_db("postgresql")
def _pg_create_db(cfg, eng, ident):
    template_db = cfg.options.postgresql_templatedb

with eng.connect().execution_options(isolation_level="AUTOCOMMIT") as conn:
        try:
            _pg_drop_db(cfg, conn, ident)
        except Exception:
            pass
        if not template_db:
            template_db = conn.scalar("select current_database()")

attempt = 0
        while True:
            try:
                conn.execute(
                    "CREATE DATABASE %s TEMPLATE %s" % (ident, template_db)
                )
            except exc.OperationalError as err:
                attempt += 1
                if attempt >= 3:
                    raise
                if "accessed by other users" in str(err):
                    log.info(
                        "Waiting to create %s, URI %r, "
                        "template DB %s is in use sleeping for .5",
                        ident,
                        eng.url,
                        template_db,
                    )
                    time.sleep(0.5)
            except:
                raise
            else:
                break

@drop_db.for_db("postgresql")
def _pg_drop_db(cfg, eng, ident):
    with eng.connect().execution_options(isolation_level="AUTOCOMMIT") as conn:
        conn.execute(
            text(
                "select pg_terminate_backend(pid) from pg_stat_activity "
                "where usename=current_user and pid != pg_backend_pid() "
                "and datname=:dname"
            ),
            dname=ident,
        )
        conn.execute("DROP DATABASE %s" % ident)

@temp_table_keyword_args.for_db("postgresql")
def _postgresql_temp_table_keyword_args(cfg, eng):
    return {"prefixes": ["TEMPORARY"]}

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Name	Size	Perms	Modified	Actions
__pycache__ DIR	-	drwxr-xr-x	2026-02-06 08:01:46	Edit Download recursive recursive
array.py	11.50 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
base.py	126.22 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
dml.py	7.61 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
ext.py	7.56 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
hstore.py	12.13 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
json.py	9.87 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
pg8000.py	9.49 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
provision.py	1.96 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
psycopg2.py	36.88 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
psycopg2cffi.py	1.62 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
pygresql.py	7.94 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
pypostgresql.py	2.85 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
ranges.py	4.51 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
zxjdbc.py	1.38 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive
__init__.py	2.40 KB	lrw-r--r--	2026-01-20 13:01:47	Edit Download recursive recursive

Directory Contents