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()