Skip to content

Commit

Permalink
Merge pull request #8830 from khoaguin/show-SQLite-db-path-dev-mode
Browse files Browse the repository at this point in the history
[DX] print out SQLite database path in dev mode
  • Loading branch information
shubham3121 committed May 21, 2024
2 parents d26e911 + 7175a0c commit 20dd090
Showing 1 changed file with 9 additions and 4 deletions.
13 changes: 9 additions & 4 deletions packages/syft/src/syft/node/node.py
Original file line number Diff line number Diff line change
Expand Up @@ -372,10 +372,12 @@ def __init__(

use_sqlite = local_db or (processes > 0 and not is_subprocess)
document_store_config = document_store_config or self.get_default_store(
use_sqlite=use_sqlite
use_sqlite=use_sqlite,
store_type="Document Store",
)
action_store_config = action_store_config or self.get_default_store(
use_sqlite=use_sqlite
use_sqlite=use_sqlite,
store_type="Action Store",
)
self.init_stores(
action_store_config=action_store_config,
Expand Down Expand Up @@ -434,12 +436,15 @@ def runs_in_docker(self) -> bool:
and any("docker" in line for line in open(path))
)

def get_default_store(self, use_sqlite: bool) -> StoreConfig:
def get_default_store(self, use_sqlite: bool, store_type: str) -> StoreConfig:
if use_sqlite:
path = self.get_temp_dir("db")
file_name: str = f"{self.id}.sqlite"
if self.dev_mode:
print(f"{store_type}'s SQLite DB path: {path/file_name}")
return SQLiteStoreConfig(
client_config=SQLiteStoreClientConfig(
filename=f"{self.id}.sqlite",
filename=file_name,
path=path,
)
)
Expand Down

0 comments on commit 20dd090

Please sign in to comment.