whoami7 - Manager
:
/
proc
/
self
/
root
/
opt
/
hc_python
/
lib
/
python3.8
/
site-packages
/
sqlalchemy
/
dialects
/
sqlite
/
Upload File:
files >> //proc/self/root/opt/hc_python/lib/python3.8/site-packages/sqlalchemy/dialects/sqlite/provision.py
# dialects/sqlite/provision.py # Copyright (C) 2005-2024 the SQLAlchemy authors and contributors # <see AUTHORS file> # # This module is part of SQLAlchemy and is released under # the MIT License: https://www.opensource.org/licenses/mit-license.php # mypy: ignore-errors import os import re from ... import exc from ...engine import url as sa_url from ...testing.provision import create_db from ...testing.provision import drop_db from ...testing.provision import follower_url_from_main from ...testing.provision import generate_driver_url from ...testing.provision import log from ...testing.provision import post_configure_engine from ...testing.provision import run_reap_dbs from ...testing.provision import stop_test_class_outside_fixtures from ...testing.provision import temp_table_keyword_args from ...testing.provision import upsert # TODO: I can't get this to build dynamically with pytest-xdist procs _drivernames = { "pysqlite", "aiosqlite", "pysqlcipher", "pysqlite_numeric", "pysqlite_dollar", } def _format_url(url, driver, ident): """given a sqlite url + desired driver + ident, make a canonical URL out of it """ url = sa_url.make_url(url) if driver is None: driver = url.get_driver_name() filename = url.database needs_enc = driver == "pysqlcipher" name_token = None if filename and filename != ":memory:": assert "test_schema" not in filename tokens = re.split(r"[_\.]", filename) new_filename = f"{driver}" for token in tokens: if token in _drivernames: if driver is None: driver = token continue elif token in ("db", "enc"): continue elif name_token is None: name_token = token.strip("_") assert name_token, f"sqlite filename has no name token: {url.database}" new_filename = f"{name_token}_{driver}" if ident: new_filename += f"_{ident}" new_filename += ".db" if needs_enc: new_filename += ".enc" url = url.set(database=new_filename) if needs_enc: url = url.set(password="test") url = url.set(drivername="sqlite+%s" % (driver,)) return url @generate_driver_url.for_db("sqlite") def generate_driver_url(url, driver, query_str): url = _format_url(url, driver, None) try: url.get_dialect() except exc.NoSuchModuleError: return None else: return url @follower_url_from_main.for_db("sqlite") def _sqlite_follower_url_from_main(url, ident): return _format_url(url, None, ident) @post_configure_engine.for_db("sqlite") def _sqlite_post_configure_engine(url, engine, follower_ident): from sqlalchemy import event if follower_ident: attach_path = f"{follower_ident}_{engine.driver}_test_schema.db" else: attach_path = f"{engine.driver}_test_schema.db" @event.listens_for(engine, "connect") def connect(dbapi_connection, connection_record): # use file DBs in all cases, memory acts kind of strangely # as an attached # NOTE! this has to be done *per connection*. New sqlite connection, # as we get with say, QueuePool, the attaches are gone. # so schemes to delete those attached files have to be done at the # filesystem level and not rely upon what attachments are in a # particular SQLite connection dbapi_connection.execute( f'ATTACH DATABASE "{attach_path}" AS test_schema' ) @event.listens_for(engine, "engine_disposed") def dispose(engine): """most databases should be dropped using stop_test_class_outside_fixtures however a few tests like AttachedDBTest might not get triggered on that main hook """ if os.path.exists(attach_path): os.remove(attach_path) filename = engine.url.database if filename and filename != ":memory:" and os.path.exists(filename): os.remove(filename) @create_db.for_db("sqlite") def _sqlite_create_db(cfg, eng, ident): pass @drop_db.for_db("sqlite") def _sqlite_drop_db(cfg, eng, ident): _drop_dbs_w_ident(eng.url.database, eng.driver, ident) def _drop_dbs_w_ident(databasename, driver, ident): for path in os.listdir("."): fname, ext = os.path.split(path) if ident in fname and ext in [".db", ".db.enc"]: log.info("deleting SQLite database file: %s", path) os.remove(path) @stop_test_class_outside_fixtures.for_db("sqlite") def stop_test_class_outside_fixtures(config, db, cls): db.dispose() @temp_table_keyword_args.for_db("sqlite") def _sqlite_temp_table_keyword_args(cfg, eng): return {"prefixes": ["TEMPORARY"]} @run_reap_dbs.for_db("sqlite") def _reap_sqlite_dbs(url, idents): log.info("db reaper connecting to %r", url) log.info("identifiers in file: %s", ", ".join(idents)) url = sa_url.make_url(url) for ident in idents: for drivername in _drivernames: _drop_dbs_w_ident(url.database, drivername, ident) @upsert.for_db("sqlite") def _upsert( cfg, table, returning, *, set_lambda=None, sort_by_parameter_order=False ): from sqlalchemy.dialects.sqlite import insert stmt = insert(table) if set_lambda: stmt = stmt.on_conflict_do_update(set_=set_lambda(stmt.excluded)) else: stmt = stmt.on_conflict_do_nothing() stmt = stmt.returning( *returning, sort_by_parameter_order=sort_by_parameter_order ) return stmt
Copyright ©2021 || Defacer Indonesia