def mock_data_access(monkeypatch, temp_fs):
data_access = SSHDataAccess()
monkeypatch.setattr(data_access, "_ssh", MockConnection())
data_access.root = temp_fs[0]
data_access.local_root = temp_fs[1]
yield data_access
data_access.close()
def get_data_access(make_fs=None):
"""Return a data access instance appropriate for the current
environment.
:param callable make_fs: a function that returns a filesystem instance, or
None to use a default
:return: (:class:`powersimdata.data_access.data_access.DataAccess`) -- a data access
instance
"""
if server_setup.DEPLOYMENT_MODE == DeploymentMode.Server:
if make_fs is None:
make_fs = lambda: None # noqa: E731
return SSHDataAccess(make_fs())
return LocalDataAccess()
def get_data_access(data_loc=None):
"""Return a data access instance appropriate for the current
environment.
:param str data_loc: pass "disk" if using data from backup disk,
otherwise leave the default.
"""
if data_loc == "disk":
root = server_setup.BACKUP_DATA_ROOT_DIR
else:
root = server_setup.DATA_ROOT_DIR
mode = get_deployment_mode()
if mode == DeploymentMode.Server:
return SSHDataAccess(root)
return LocalDataAccess(root)
def get_data_access(data_loc=None):
"""Return a data access instance appropriate for the current
environment.
:param str data_loc: pass "disk" if using data from backup disk,
otherwise leave the default.
:return: (:class:`powersimdata.data_access.data_access.DataAccess`) -- a data access
instance
"""
if data_loc == "disk":
root = server_setup.BACKUP_DATA_ROOT_DIR
else:
root = server_setup.DATA_ROOT_DIR
if server_setup.DEPLOYMENT_MODE == DeploymentMode.Server:
return SSHDataAccess(root)
return LocalDataAccess(root)
def data_access():
data_access = SSHDataAccess()
yield data_access
data_access.close()
def ssh_data_access():
return SSHDataAccess()
def data_access():
return SSHDataAccess()