github/datasette
1
0
Fork 0
mirror of https://github.com/simonw/datasette.git synced 2025-12-10 16:51:24 +01:00
Code Issues Projects Releases Packages Wiki Activity

Fix to support replacing a database, closes #2465

Browse source
This commit is contained in:
Simon Willison 2025-02-06 10:32:47 -08:00
commit f95ac19e71
2 changed files with 36 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
datasette/database.py
View file

@ -32,6 +32,7 @@ AttachedDatabase = namedtuple("AttachedDatabase", ("seq", "name", "file"))
class Database:
# For table counts stop at this many rows:
count_limit = 10000
_thread_local_id_counter = 1
def __init__(
self,
@ -43,6 +44,8 @@ class Database:
mode=None,
):
self.name = None
self._thread_local_id = f"x{self._thread_local_id_counter}"
Database._thread_local_id_counter += 1
self.route = None
self.ds = ds
self.path = path
@ -278,11 +281,11 @@ class Database:
# threaded mode
def in_thread():
conn = getattr(connections, self.name, None)
conn = getattr(connections, self._thread_local_id, None)
if not conn:
conn = self.connect()
self.ds._prepare_connection(conn, self.name)
setattr(connections, self.name, conn)
setattr(connections, self._thread_local_id, conn)
return fn(conn)
return await asyncio.get_event_loop().run_in_executor(