Import python venv for stability

2026-02-15 21:24:16 -08:00
parent 1343e93a59
commit 7d784705c9
4997 changed files with 1628270 additions and 0 deletions
@@ -0,0 +1,133 @@
+try:
+    from urlparse import parse_qsl, unquote, urlparse
+except ImportError:
+    from urllib.parse import parse_qsl, unquote, urlparse
+
+from peewee import *
+from playhouse.cockroachdb import CockroachDatabase
+from playhouse.cockroachdb import PooledCockroachDatabase
+from playhouse.pool import PooledMySQLDatabase
+from playhouse.pool import PooledPostgresqlDatabase
+from playhouse.pool import PooledPsycopg3Database
+from playhouse.pool import PooledSqliteDatabase
+from playhouse.pool import PooledSqliteExtDatabase
+from playhouse.psycopg3_ext import Psycopg3Database
+from playhouse.sqlite_ext import SqliteExtDatabase
+
+
+schemes = {
+    'cockroachdb': CockroachDatabase,
+    'cockroachdb+pool': PooledCockroachDatabase,
+    'crdb': CockroachDatabase,
+    'crdb+pool': PooledCockroachDatabase,
+    'mysql': MySQLDatabase,
+    'mysql+pool': PooledMySQLDatabase,
+    'postgres': PostgresqlDatabase,
+    'postgresql': PostgresqlDatabase,
+    'postgres+pool': PooledPostgresqlDatabase,
+    'postgresql+pool': PooledPostgresqlDatabase,
+    'psycopg3': Psycopg3Database,
+    'psycopg3+pool': PooledPsycopg3Database,
+    'sqlite': SqliteDatabase,
+    'sqliteext': SqliteExtDatabase,
+    'sqlite+pool': PooledSqliteDatabase,
+    'sqliteext+pool': PooledSqliteExtDatabase,
+}
+
+def register_database(db_class, *names):
+    global schemes
+    for name in names:
+        schemes[name] = db_class
+
+def parseresult_to_dict(parsed, unquote_password=False, unquote_user=False):
+
+    # urlparse in python 2.6 is broken so query will be empty and instead
+    # appended to path complete with '?'
+    path = parsed.path[1:]  # Ignore leading '/'.
+    query = parsed.query
+
+    connect_kwargs = {'database': path}
+    if parsed.username:
+        connect_kwargs['user'] = parsed.username
+        if unquote_user:
+            connect_kwargs['user'] = unquote(connect_kwargs['user'])
+    if parsed.password:
+        connect_kwargs['password'] = parsed.password
+        if unquote_password:
+            connect_kwargs['password'] = unquote(connect_kwargs['password'])
+    if parsed.hostname:
+        connect_kwargs['host'] = parsed.hostname
+    if parsed.port:
+        connect_kwargs['port'] = parsed.port
+
+    # Adjust parameters for MySQL.
+    if parsed.scheme == 'mysql' and 'password' in connect_kwargs:
+        connect_kwargs['passwd'] = connect_kwargs.pop('password')
+    elif 'sqlite' in parsed.scheme and not connect_kwargs['database']:
+        connect_kwargs['database'] = ':memory:'
+
+    # Get additional connection args from the query string
+    qs_args = parse_qsl(query, keep_blank_values=True)
+    for key, value in qs_args:
+        if value.lower() == 'false':
+            value = False
+        elif value.lower() == 'true':
+            value = True
+        elif value.isdigit():
+            value = int(value)
+        elif '.' in value and all(p.isdigit() for p in value.split('.', 1)):
+            try:
+                value = float(value)
+            except ValueError:
+                pass
+        elif value.lower() in ('null', 'none'):
+            value = None
+
+        connect_kwargs[key] = value
+
+    return connect_kwargs
+
+def parse(url, unquote_password=False, unquote_user=False):
+    parsed = urlparse(url)
+    return parseresult_to_dict(parsed, unquote_password, unquote_user)
+
+def connect(url, unquote_password=False, unquote_user=False, **connect_params):
+    parsed = urlparse(url)
+    connect_kwargs = parseresult_to_dict(parsed, unquote_password, unquote_user)
+    connect_kwargs.update(connect_params)
+    database_class = schemes.get(parsed.scheme)
+
+    if database_class is None:
+        if database_class in schemes:
+            raise RuntimeError('Attempted to use "%s" but a required library '
+                               'could not be imported.' % parsed.scheme)
+        else:
+            raise RuntimeError('Unrecognized or unsupported scheme: "%s".' %
+                               parsed.scheme)
+
+    return database_class(**connect_kwargs)
+
+# Conditionally register additional databases.
+try:
+    from playhouse.pool import PooledPostgresqlExtDatabase
+except ImportError:
+    pass
+else:
+    register_database(
+        PooledPostgresqlExtDatabase,
+        'postgresext+pool',
+        'postgresqlext+pool')
+
+try:
+    from playhouse.apsw_ext import APSWDatabase
+except ImportError:
+    pass
+else:
+    register_database(APSWDatabase, 'apsw')
+
+try:
+    from playhouse.postgres_ext import PostgresqlExtDatabase
+except ImportError:
+    pass
+else:
+    register_database(PostgresqlExtDatabase, 'postgresext', 'postgresqlext')