Search in sources :

Example 1 with TaskModel

use of org.datatransferproject.types.common.models.tasks.TaskModel in project data-transfer-project by google.

the class RememberTheMilkTasksExporter method exportTask.

private ExportResult exportTask(RememberTheMilkService service, IdOnlyContainerResource resource) {
    String oldListId = resource.getId();
    GetListResponse oldList = null;
    try {
        oldList = service.getList(oldListId);
    } catch (IOException e) {
        return new ExportResult(e);
    }
    List<TaskList> taskLists = oldList.tasks.list;
    List<TaskModel> tasks = new ArrayList<>();
    for (TaskList taskList : taskLists) {
        if (taskList.taskseries != null) {
            for (TaskSeries taskSeries : taskList.taskseries) {
                // TODO: figure out what to do with notes
                String notesStr = taskSeries.notes == null ? "" : taskSeries.notes.toString();
                for (Task task : taskSeries.tasks) {
                    // TODO: How to handle case with multiple tasks in a series?  Is this good enough?
                    Instant completedTime = null;
                    Instant dueTime = null;
                    if (task.completed != null && !Strings.isNullOrEmpty(task.completed)) {
                        completedTime = Instant.parse(task.completed);
                    }
                    if (task.due != null && !Strings.isNullOrEmpty(task.due)) {
                        dueTime = Instant.parse(task.due);
                    }
                    tasks.add(new TaskModel(oldListId, taskSeries.name, notesStr, completedTime, dueTime));
                }
            }
        }
    }
    TaskContainerResource taskContainerResource = new TaskContainerResource(null, tasks);
    // TODO: what do we do with pagination data?
    return new ExportResult(ResultType.CONTINUE, taskContainerResource, null);
}
Also used : GetListResponse(org.datatransferproject.transfer.rememberthemilk.model.tasks.GetListResponse) Task(org.datatransferproject.transfer.rememberthemilk.model.tasks.Task) TaskList(org.datatransferproject.transfer.rememberthemilk.model.tasks.TaskList) Instant(java.time.Instant) ArrayList(java.util.ArrayList) IOException(java.io.IOException) TaskContainerResource(org.datatransferproject.types.common.models.tasks.TaskContainerResource) TaskSeries(org.datatransferproject.transfer.rememberthemilk.model.tasks.TaskSeries) TaskModel(org.datatransferproject.types.common.models.tasks.TaskModel) ExportResult(org.datatransferproject.spi.transfer.provider.ExportResult)

Example 2 with TaskModel

use of org.datatransferproject.types.common.models.tasks.TaskModel in project data-transfer-project by google.

the class GoogleTasksExporter method getTasks.

private ExportResult<TaskContainerResource> getTasks(Tasks tasksService, IdOnlyContainerResource resource, Optional<PaginationData> paginationData) throws IOException {
    Tasks.TasksOperations.List query = tasksService.tasks().list(resource.getId()).setMaxResults(PAGE_SIZE);
    if (paginationData.isPresent()) {
        query.setPageToken(((StringPaginationToken) paginationData.get()).getToken());
    }
    com.google.api.services.tasks.model.Tasks result = query.execute();
    List<TaskModel> newTasks = result.getItems().stream().map(t -> new TaskModel(resource.getId(), t.getTitle(), t.getNotes(), t.getCompleted() != null ? Instant.ofEpochMilli(t.getCompleted().getValue()) : null, t.getDue() != null ? Instant.ofEpochMilli(t.getDue().getValue()) : null)).collect(Collectors.toList());
    PaginationData newPage = null;
    ResultType resultType = ResultType.END;
    if (result.getNextPageToken() != null) {
        newPage = new StringPaginationToken(result.getNextPageToken());
        resultType = ResultType.CONTINUE;
    }
    TaskContainerResource taskContainerResource = new TaskContainerResource(null, newTasks);
    return new ExportResult<>(resultType, taskContainerResource, new ContinuationData(newPage));
}
Also used : ExportResult(org.datatransferproject.spi.transfer.provider.ExportResult) GoogleStaticObjects(org.datatransferproject.datatransfer.google.common.GoogleStaticObjects) ExportInformation(org.datatransferproject.types.common.ExportInformation) TaskContainerResource(org.datatransferproject.types.common.models.tasks.TaskContainerResource) ImmutableList(com.google.common.collect.ImmutableList) StringPaginationToken(org.datatransferproject.types.common.StringPaginationToken) Credential(com.google.api.client.auth.oauth2.Credential) TaskLists(com.google.api.services.tasks.model.TaskLists) GoogleCredentialFactory(org.datatransferproject.datatransfer.google.common.GoogleCredentialFactory) TaskModel(org.datatransferproject.types.common.models.tasks.TaskModel) ResultType(org.datatransferproject.spi.transfer.provider.ExportResult.ResultType) Exporter(org.datatransferproject.spi.transfer.provider.Exporter) IOException(java.io.IOException) Tasks(com.google.api.services.tasks.Tasks) PaginationData(org.datatransferproject.types.common.PaginationData) UUID(java.util.UUID) Instant(java.time.Instant) IdOnlyContainerResource(org.datatransferproject.types.common.models.IdOnlyContainerResource) Collectors(java.util.stream.Collectors) TaskListModel(org.datatransferproject.types.common.models.tasks.TaskListModel) List(java.util.List) TaskList(com.google.api.services.tasks.model.TaskList) Monitor(org.datatransferproject.api.launcher.Monitor) TokensAndUrlAuthData(org.datatransferproject.types.transfer.auth.TokensAndUrlAuthData) Optional(java.util.Optional) ContinuationData(org.datatransferproject.spi.transfer.types.ContinuationData) VisibleForTesting(com.google.common.annotations.VisibleForTesting) PaginationData(org.datatransferproject.types.common.PaginationData) ContinuationData(org.datatransferproject.spi.transfer.types.ContinuationData) ResultType(org.datatransferproject.spi.transfer.provider.ExportResult.ResultType) TaskContainerResource(org.datatransferproject.types.common.models.tasks.TaskContainerResource) TaskModel(org.datatransferproject.types.common.models.tasks.TaskModel) StringPaginationToken(org.datatransferproject.types.common.StringPaginationToken) ExportResult(org.datatransferproject.spi.transfer.provider.ExportResult)

Example 3 with TaskModel

use of org.datatransferproject.types.common.models.tasks.TaskModel in project data-transfer-project by google.

the class TaskContainerResourceTest method verifySerializeDeserialize.

@Test
public void verifySerializeDeserialize() throws Exception {
    ObjectMapper objectMapper = new ObjectMapper();
    objectMapper.registerModule(new JavaTimeModule());
    objectMapper.registerSubtypes(TaskContainerResource.class);
    List<TaskListModel> taskLists = ImmutableList.of(new TaskListModel("id1", "List 1"));
    List<TaskModel> tasks = ImmutableList.of(new TaskModel("id1", "Write Better tests", "Do this soon", null, null), new TaskModel("id2", "Liberate all the data", "do this in stages", null, null));
    ContainerResource data = new TaskContainerResource(taskLists, tasks);
    String serialized = objectMapper.writeValueAsString(data);
    ContainerResource deserializedModel = objectMapper.readValue(serialized, ContainerResource.class);
    Truth.assertThat(deserializedModel).isNotNull();
    Truth.assertThat(deserializedModel).isInstanceOf(TaskContainerResource.class);
    TaskContainerResource deserialized = (TaskContainerResource) deserializedModel;
    Truth.assertThat(deserialized.getLists()).hasSize(1);
    Truth.assertThat(deserialized.getTasks()).hasSize(2);
    Truth.assertThat(deserialized).isEqualTo(data);
}
Also used : TaskContainerResource(org.datatransferproject.types.common.models.tasks.TaskContainerResource) ContainerResource(org.datatransferproject.types.common.models.ContainerResource) JavaTimeModule(com.fasterxml.jackson.datatype.jsr310.JavaTimeModule) TaskContainerResource(org.datatransferproject.types.common.models.tasks.TaskContainerResource) ObjectMapper(com.fasterxml.jackson.databind.ObjectMapper) TaskListModel(org.datatransferproject.types.common.models.tasks.TaskListModel) TaskModel(org.datatransferproject.types.common.models.tasks.TaskModel) Test(org.junit.Test)

Example 4 with TaskModel

use of org.datatransferproject.types.common.models.tasks.TaskModel in project data-transfer-project by google.

the class RememberTheMilkTasksImporter method importItem.

@Override
public ImportResult importItem(UUID jobId, IdempotentImportExecutor idempotentExecutor, AuthData authData, TaskContainerResource data) {
    String timeline;
    try {
        RememberTheMilkService service = getOrCreateService(authData);
        timeline = service.createTimeline();
        for (TaskListModel taskList : data.getLists()) {
            idempotentExecutor.executeAndSwallowIOExceptions(taskList.getId(), taskList.getName(), () -> service.createTaskList(taskList.getName(), timeline).id);
        }
        for (TaskModel task : data.getTasks()) {
            // Empty or blank tasks aren't valid in RTM
            if (!Strings.isNullOrEmpty(task.getText())) {
                idempotentExecutor.executeAndSwallowIOExceptions(Integer.toString(task.hashCode()), task.getText(), () -> {
                    String newList = idempotentExecutor.getCachedValue(task.getTaskListId());
                    return insertTask(task, newList, timeline);
                });
            }
        }
    } catch (Exception e) {
        monitor.severe(() -> "Error importing item", e);
        return new ImportResult(e);
    }
    return new ImportResult(ImportResult.ResultType.OK);
}
Also used : ImportResult(org.datatransferproject.spi.transfer.provider.ImportResult) TaskListModel(org.datatransferproject.types.common.models.tasks.TaskListModel) TaskModel(org.datatransferproject.types.common.models.tasks.TaskModel) IOException(java.io.IOException)

Example 5 with TaskModel

use of org.datatransferproject.types.common.models.tasks.TaskModel in project data-transfer-project by google.

the class GoogleTasksImporter method importItem.

@Override
public ImportResult importItem(UUID jobId, IdempotentImportExecutor idempotentImportExecutor, TokensAndUrlAuthData authData, TaskContainerResource data) throws Exception {
    Tasks tasksService = getOrCreateTasksService(authData);
    for (TaskListModel oldTasksList : data.getLists()) {
        TaskList newTaskList = new TaskList().setTitle("Imported copy - " + oldTasksList.getName());
        idempotentImportExecutor.executeAndSwallowIOExceptions(oldTasksList.getId(), oldTasksList.getName(), () -> tasksService.tasklists().insert(newTaskList).execute().getId());
    }
    for (TaskModel oldTask : data.getTasks()) {
        Task newTask = new Task().setTitle(oldTask.getText()).setNotes(oldTask.getNotes());
        if (oldTask.getCompletedTime() != null) {
            newTask.setCompleted(new DateTime(oldTask.getCompletedTime().toEpochMilli()));
        }
        if (oldTask.getDueTime() != null) {
            newTask.setDue(new DateTime(oldTask.getDueTime().toEpochMilli()));
        }
        // If its not cached that means the task list create failed.
        if (idempotentImportExecutor.isKeyCached(oldTask.getTaskListId())) {
            String newTaskListId = idempotentImportExecutor.getCachedValue(oldTask.getTaskListId());
            idempotentImportExecutor.executeAndSwallowIOExceptions(oldTask.getTaskListId() + oldTask.getText(), oldTask.getText(), () -> tasksService.tasks().insert(newTaskListId, newTask).execute().getId());
        }
    }
    return new ImportResult(ResultType.OK);
}
Also used : Task(com.google.api.services.tasks.model.Task) Tasks(com.google.api.services.tasks.Tasks) ImportResult(org.datatransferproject.spi.transfer.provider.ImportResult) TaskList(com.google.api.services.tasks.model.TaskList) TaskListModel(org.datatransferproject.types.common.models.tasks.TaskListModel) TaskModel(org.datatransferproject.types.common.models.tasks.TaskModel) DateTime(com.google.api.client.util.DateTime)

Aggregations

TaskModel (org.datatransferproject.types.common.models.tasks.TaskModel)5 TaskListModel (org.datatransferproject.types.common.models.tasks.TaskListModel)4 IOException (java.io.IOException)3 TaskContainerResource (org.datatransferproject.types.common.models.tasks.TaskContainerResource)3 Tasks (com.google.api.services.tasks.Tasks)2 TaskList (com.google.api.services.tasks.model.TaskList)2 Instant (java.time.Instant)2 ExportResult (org.datatransferproject.spi.transfer.provider.ExportResult)2 ImportResult (org.datatransferproject.spi.transfer.provider.ImportResult)2 ObjectMapper (com.fasterxml.jackson.databind.ObjectMapper)1 JavaTimeModule (com.fasterxml.jackson.datatype.jsr310.JavaTimeModule)1 Credential (com.google.api.client.auth.oauth2.Credential)1 DateTime (com.google.api.client.util.DateTime)1 Task (com.google.api.services.tasks.model.Task)1 TaskLists (com.google.api.services.tasks.model.TaskLists)1 VisibleForTesting (com.google.common.annotations.VisibleForTesting)1 ImmutableList (com.google.common.collect.ImmutableList)1 ArrayList (java.util.ArrayList)1 List (java.util.List)1 Optional (java.util.Optional)1