bulk.py

import sublime
import sublime_plugin
import os
from . import util
from . import processor


class CloseJobsCommand(sublime_plugin.WindowCommand):
    def __init__(self, *args, **kwargs):
        super(CloseJobsCommand, self).__init__(*args, **kwargs)

    def run(self):
        self.window.show_input_panel("Input JobId (Separated By ','): ", "", self.on_input, None, None)

    def on_input(self, input):
        if not input:
            sublime.error_message("Please input job ids that is separated by ','")
            return

        job_ids = input.split(",")
        processor.handle_close_jobs_thread(job_ids)

class BulkExportSingleCommand(sublime_plugin.WindowCommand):
    def __init__(self, *args, **kwargs):
        super(BulkExportSingleCommand, self).__init__(*args, **kwargs)

    def run(self, export_soql=False):
        self.export_soql = export_soql
        sobjects_describe = util.populate_sobjects_describe()
        self.sobjects = sorted(sobjects_describe.keys())
        self.window.show_quick_panel(self.sobjects, self.on_done)

    def on_done(self, index):
        if index == -1: return
        self.chosen_sobject = self.sobjects[index]
        if not self.export_soql:
            processor.handle_backup_sobject_thread(self.chosen_sobject)
        else:
            self.window.show_input_panel("Input SOQL: ", "", self.on_input, None, None)

    def on_input(self, soql):
        if not soql or not soql.upper().startswith("SELECT"):
            sublime.error_message("Invalid SOQL")
            return

        processor.handle_backup_sobject_thread(self.chosen_sobject, soql)

class BulkExportAllCommand(sublime_plugin.WindowCommand):
    def __init__(self, *args, **kwargs):
        super(BulkExportAllCommand, self).__init__(*args, **kwargs)

    def run(self):
        processor.handle_backup_all_sobjects_thread()

class BulkOperationCommand(sublime_plugin.WindowCommand):
    def __init__(self, *args, **kwargs):
        super(BulkOperationCommand, self).__init__(*args, **kwargs)

    def run(self, operation=None):
        self.operation = operation
        sobjects_describe = util.populate_sobjects_describe()
        self.sobjects = sorted(sobjects_describe.keys())
        self.window.show_quick_panel(self.sobjects, self.on_done)

    def on_done(self, index):
        if index == -1: return
        self.sobject = self.sobjects[index]
        path = sublime.get_clipboard()
        if not os.path.isfile(path): path = ""
        self.window.show_input_panel("Input CSV Path: ", 
            path, self.on_input, None, None)

    def on_input(self, file_path):
        if not file_path.endswith('csv'): 
            sublime.error_message("Input file must be CSV")
            return

        if not os.path.exists(file_path):
            sublime.error_message(file_path + " is not valid file")
            return

        processor.handle_bulk_operation_thread(self.sobject, file_path, self.operation)

class BulkUpsertCommand(sublime_plugin.WindowCommand):
    def __init__(self, *args, **kwargs):
        super(BulkUpsertCommand, self).__init__(*args, **kwargs)

    def run(self):
        sublime.message_dialog("This command is ongoing")