def row_count(self: Sqlite3Adapter, data_token: DataToken) -> int:
if self._connection == None:
raise DatabaseAdapterUsageError("row_count")
try:
statement = SqlStatement().SELECT().COUNT().FROM(data_token)
cursor = self._connection.execute(statement.compile())
return cursor.fetchone()[0]
except Exception as e:
raise DatabaseAdapterError("row_count failed", e)
def delete(self: Sqlite3Adapter, data_token: DataToken,
criteria: Query) -> None:
if self._connection == None:
raise DatabaseAdapterUsageError("delete")
try:
compiler = SqlQueryCompiler(quote=True)
query = compiler.compile(criteria)
statement = SqlStatement().DELETE().FROM(data_token).WHERE(query)
self._connection.execute(statement.compile())
except Exception as e:
raise DatabaseAdapterError("delete failed", e)
def _insert_from_values(
self: Sqlite3Adapter,
data_token: DataToken,
data_store: T,
) -> None:
if data_store.metadata is not None:
self._update_group_table_metadata(data_token, data_store.metadata)
if self._connection == None:
raise DatabaseAdapterUsageError("_insert_from_values")
try:
columns = [str(col) for col in data_store.columns]
values = ["?" for _ in range(len(data_store.columns))]
mappings = Sqlite3Type.type_mappings(data_store.dtypes)
mapped_store = data_store.to_pandas().astype(mappings,
errors="ignore")
statement = (SqlStatement().INSERT_INTO(
data_token, *columns).VALUES(values, quote=False).compile())
self._connection.executemany(
statement,
mapped_store.itertuples(index=False),
)
except Exception as e:
raise DatabaseAdapterError("_insert_from_values failed", e)
def _upsert_from_values(
self: Sqlite3Adapter,
data_token: DataToken,
data_store: T,
alignment_columns: list[str],
) -> None:
if data_store.metadata is not None:
self._update_group_table_metadata(data_token, data_store.metadata)
if self._connection == None:
raise DatabaseAdapterUsageError("_upsert_from_values")
try:
columns = [str(col) for col in data_store.columns]
values = ["?" for _ in range(len(data_store.columns))]
alignment_columns = [str(col) for col in alignment_columns]
all_columns = [str(col) for col in data_store.columns]
update_columns = list(set(all_columns) - set(alignment_columns))
statement = (SqlStatement().INSERT_INTO(
data_token,
*columns).VALUES(values, quote=False).UPDATE_CONFLICTS(
alignment_columns, update_columns))
self._connection.executemany(
statement.compile(),
data_store.itertuples(ignore_index=True),
)
except Exception as e:
raise DatabaseAdapterError("_upsert_from_values failed", e)
def _update_from_values(
self: Sqlite3Adapter,
data_token: DataToken,
data_store: T,
alignment_columns: list[str],
) -> None:
if data_store.metadata is not None:
self._update_group_table_metadata(data_token, data_store.metadata)
if self._connection == None:
raise DatabaseAdapterUsageError("_update_from_values")
try:
alignment_columns = [str(col) for col in alignment_columns]
all_columns = [str(col) for col in data_store.columns]
db_alignment_values = ["?" for _ in alignment_columns]
update_columns = list(set(all_columns) - set(alignment_columns))
update_values = ["?" for _ in update_columns]
query = ColumnQuery(EqualsQuery, AndGroupQuery, alignment_columns,
db_alignment_values)
compiler = SqlQueryCompiler(quote=False)
query = compiler.compile(query)
statement = (SqlStatement().UPDATE_FROM_VALUES(
data_token, update_columns,
update_values).WHERE(str(query)).compile())
statement_columns = update_columns + alignment_columns
self._connection.executemany(
statement,
data_store[statement_columns].itertuples(ignore_index=True),
)
except Exception as e:
raise DatabaseAdapterError("_update_from_values failed", e)
def rollback(self, connection: Connection):
self.data_group_path.unlink()
checkpoint_path = self.checkpoint_path()
shutil.copyfile(str(checkpoint_path), str(self.data_group_path))
checkpoint_path.unlink()
statement = SqlStatement().ATTACH_DATABASE(self.data_group).compile()
connection.execute(statement)
def query(
self: Sqlite3Adapter,
data_token: DataToken,
query: Optional[Query] = None,
columns: Optional[list[str]] = None,
) -> list[tuple]:
if self._connection == None:
raise DatabaseAdapterUsageError("query")
try:
statement = SqlStatement()
if columns is not None:
clean = []
for column in columns:
clean.append(column)
statement.SELECT(*clean)
else:
statement.SELECT_ALL()
statement.FROM(str(data_token))
if query is not None:
compiler = SqlQueryCompiler(quote=True)
query = compiler.compile(query)
statement.WHERE(str(query))
statement = statement.compile()
cursor = self._connection.execute(statement)
data_rows = cursor.fetchall()
return data_rows
except Exception as e:
raise DatabaseAdapterError("query failed", e)
def drop_group_table(self: Sqlite3Adapter, data_token: DataToken) -> None:
self._delete_group_table_metadata(data_token)
if self._connection == None:
raise DatabaseAdapterUsageError("drop_group_table")
try:
statement = SqlStatement().DROP_TABLE(f"{data_token}").compile()
self._connection.execute(statement)
except Exception as e:
raise DatabaseAdapterError("drop_group_table failed", e)
def create_group_table(self: "Sqlite3Adapter", data_token: DataToken,
data_store_type: type[DataStore]) -> None:
if self._connection == None:
raise DatabaseAdapterUsageError("create_group_table")
try:
schema = Sqlite3Schema(data_store_type)
statement = (SqlStatement().CREATE_TABLE(f"{data_token}",
str(schema)).compile())
self._connection.execute(statement)
except Exception as e:
raise DatabaseAdapterError("create_group_table failed", e)
def create_index(self: Sqlite3Adapter, data_token: DataToken,
index: Index) -> None:
if self._connection == None:
raise DatabaseAdapterUsageError("create_index")
try:
col_names = [str(col) for col in index.columns]
statement = (SqlStatement().CREATE_INDEX(index.name, data_token,
col_names).compile())
self._connection.execute(statement)
except Exception as e:
raise DatabaseAdapterError("create_index failed", e)
def _update_from_link(
self: Sqlite3Adapter,
data_token: DataToken,
data_store: T,
alignment_columns: list[str],
) -> None:
if self._connection == None:
raise DatabaseAdapterUsageError("_update_from_link")
try:
link_token = data_store.link_token()
alignment_columns = [str(col) for col in alignment_columns]
all_columns = [str(col) for col in data_store.columns]
update_columns = list(set(all_columns) - set(alignment_columns))
statement = SqlStatement().UPDATE_FROM_LINK(
link_token, data_token, update_columns, alignment_columns)
self._connection.execute(statement.compile())
except Exception as e:
raise DatabaseAdapterError("_update_from_link failed", e)
def _insert_from_link(self: Sqlite3Adapter, data_token: DataToken,
data_store: T) -> None:
if self._connection == None:
raise DatabaseAdapterUsageError("_insert_from_link")
try:
link_token = data_store.link_token()
statement = (SqlStatement().INSERT_INTO(
data_token, *data_store.columns).SELECT(
*data_store.columns).FROM(link_token).compile())
self._connection.execute(statement)
except Exception as e:
raise DatabaseAdapterError("_insert_from_link failed", e)
def create_group(self: Sqlite3Adapter, data_group: str) -> None:
if self._connection == None:
raise DatabaseAdapterUsageError("create_group")
try:
statement = (SqlStatement().ATTACH_DATABASE(
self._conn_config.uri(), data_group).AS(data_group).compile())
self._connection.execute(statement)
data_group_path = Path(
self._conn_config.uri()) / f"{data_group}.db"
self._uncommitted.append(
UncommittedGroupCreate(data_group, data_group_path))
except Exception as e:
raise DatabaseAdapterError("create_group failed", e)
def has_group_table(self: Sqlite3Adapter, data_token: DataToken) -> bool:
if self._connection == None:
raise DatabaseAdapterUsageError("has_group_table")
try:
if not self.has_group(data_token.data_group):
return False
statement = (SqlStatement().SELECT("count(name)").FROM(
f"{data_token.data_group}.sqlite_master").WHERE(
f"type='table' AND name='{data_token.table_name}'").
compile())
cursor = self._connection.execute(statement)
return cursor.fetchone()[0] > 0
except Exception as e:
raise DatabaseAdapterError("has_group_table failed", e)
def drop_group(self: Sqlite3Adapter, data_group: str) -> None:
self._delete_group_metadata(data_group)
if self._connection == None:
raise DatabaseAdapterUsageError("drop_group")
try:
statement = SqlStatement().DETACH_DATABASE(data_group).compile()
self._connection.execute(statement)
db_path = Path(self._conn_config.uri()) / f"{data_group}.db"
uncommittment = UncommittedGroupDelete(db_path, data_group,
self._checkpoint_dir)
uncommittment.checkpoint()
db_path.unlink()
self._uncommitted.append(uncommittment)
except Exception as e:
raise DatabaseAdapterError("drop_group failed", e)
def _upsert_from_link(
self: Sqlite3Adapter,
data_token: DataToken,
data_store: T,
alignment_columns: list[str],
) -> None:
if self._connection == None:
raise DatabaseAdapterUsageError("_upsert_from_link")
try:
link_token = data_store.link_token()
alignment_columns = [str(col) for col in alignment_columns]
all_columns = [str(col) for col in data_store.columns]
update_columns = list(set(all_columns) - set(alignment_columns))
statement = (SqlStatement().INSERT_INTO(
data_token,
*data_store.columns).SELECT(*data_store.columns).FROM(
link_token).WHERE("true").UPDATE_CONFLICTS(
alignment_columns, update_columns).compile())
self._connection.execute(statement)
except Exception as e:
raise DatabaseAdapterError("_upsert_from_link failed", e)
def rollback(self, connection: Connection):
self.data_group_path.unlink()
statement = SqlStatement().DETACH_DATABASE(self.data_group).compile()
connection.execute(statement)