def test_get_file(mocker, session): source = factory.Source() file = factory.File(source=source) session.add(source) session.add(file) result = get_file(session, file.uuid) assert result == file
def on_file_download_success(self, uuid: Any) -> None: """ Called when a file has downloaded. """ self.session.commit() file_obj = storage.get_file(self.session, uuid) file_obj.download_error = None storage.update_file_size(uuid, self.data_dir, self.session) self.file_ready.emit(file_obj.source.uuid, uuid, file_obj.filename)
def get_file(self, file_uuid: str) -> db.File: file = storage.get_file(self.session, file_uuid) self.session.refresh(file) return file