Added encoding selection support for database

Author: mtxr

SQLToolsConnections.sublime-settings

@@ -1,4 +1,16 @@
 {
-    "connections": {},
+    "connections": {
+        /**
+         * "Connection name": {         // Connection name, used in menu (Display name)
+         *     "database" : "DATABASE",
+         *     "host"     : "HOSTNAME",
+         *     "port"     : "PORT",
+         *     "type"     : "pgsql",    // SGDB to use: (mysql, pgsql, oracle, vertica)
+         *     "username" : "USERNAME",
+         *     "password" : null        // (if you are using postgresql, you must setup a pgpass file and set it to null)
+         *     "encoding" : null        // Your DB encodings. Default: utf-8
+         * },
+         */
+    },
     "default": null
 }

SQLToolsModels.py

@@ -93,6 +93,9 @@ def __init__(self, name, options):
         self.username    = options['username']
         self.database    = options['database']
 
+        if 'encoding' in options:
+            self.encoding    = options['encoding']
+
         if 'password' in options:
             self.password = options['password']
 
@@ -212,10 +215,11 @@ def formatSql(edit):
                 View().replace(edit, region, Utils.formatSql(text))
 
 class Command(threading.Thread):
-    def __init__(self, args, callback, query=None):
+    def __init__(self, args, callback, query=None, encoding='utf-8'):
         self.query    = query
         self.process  = None
         self.args     = args
+        self.encoding = encoding
         self.callback = callback
         threading.Thread.__init__(self)
 
@@ -234,10 +238,10 @@ def run(self):
         results, errors = self.process.communicate()
 
         if errors:
-            self.callback(errors.decode('utf-8', 'replace').replace('\r', ''))
+            self.callback(errors.decode(self.encoding, 'replace').replace('\r', ''))
             return
 
-        self.callback(results.decode('utf-8', 'replace').replace('\r', ''))
+        self.callback(results.decode(self.encoding, 'replace').replace('\r', ''))
 
     def stop(self):
         if not self.process: