Merge pull request #574 from Dipsomaniac/master

Support encoding in Postgres
coleifer · Apr 17, 2015 · 2f232b0 · 2f232b0
2 parents fa356ce + 96c89ab
commit 2f232b0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/peewee.py b/peewee.py
@@ -3250,13 +3250,15 @@ class PostgresqlDatabase(Database):
 
     register_unicode = True
 
-    def _connect(self, database, **kwargs):
+    def _connect(self, database, encoding=None, **kwargs):
         if not psycopg2:
             raise ImproperlyConfigured('psycopg2 must be installed.')
         conn = psycopg2.connect(database=database, **kwargs)
         if self.register_unicode:
             pg_extensions.register_type(pg_extensions.UNICODE, conn)
             pg_extensions.register_type(pg_extensions.UNICODEARRAY, conn)
+        if encoding:
+            conn.set_client_encoding(encoding)
         return conn
 
     def _get_pk_sequence(self, model):