FormatSQL.py

from __future__ import absolute_import
import sublime
import sublime_plugin

try:
    from .sqlparse import format
except ValueError:
    from sqlparse import format


class FormatSqlCommand(sublime_plugin.TextCommand):
    def run(self, edit):
        view = self.view
        regions = view.sel()
        # if there are more than 1 region or region one and it's not empty
        if len(regions) > 1 or not regions[0].empty():
            for region in view.sel():
                if not region.empty():
                    s = view.substr(region)
                    s = self._run(s)
                    view.replace(edit, region, s)
        else:  # format all text
            alltextreg = sublime.Region(0, view.size())
            s = view.substr(alltextreg)
            s = self._run(s)
            view.replace(edit, alltextreg, s)

    def _run(self, s):
        settings = self.view.settings()
        #indent_char = " " if settings.get("translate_tabs_to_spaces") else "\t"
        indent_char = " " #TODO indent by TAB (currently not supported in python-sqlparse)
        indent_size = int(settings.get("tab_size")) if indent_char == " " else 1
        s = s.encode("utf-8")
        return format(
            s, keyword_case="upper", reindent=True, indent_width=indent_size
        )