Set session timezone to local timezone on startup

AUTHORS

@@ -136,6 +136,7 @@ Contributors:
     * saucoide
     * Chris Rose (offbyone/offby1)
     * Chris Novakovic
+    * Max Smolin (maximsmol)
 
 Creator:
 --------

changelog.rst

@@ -4,6 +4,7 @@ Dev
 Features
 --------
 * Add a `--ping` command line option; allows pgcli to replace `pg_isready`
+* The session time zone setting is set to the system time zone by default
 
 Bug fixes:
 ----------

pgcli/main.py

@@ -23,6 +23,7 @@
 from cli_helpers.utils import strip_ansi
 from .explain_output_formatter import ExplainOutputFormatter
 import click
+import tzlocal
 
 try:
     import setproctitle
@@ -1624,6 +1625,8 @@ def cli(
     else:
         pgcli.connect(database, host, user, port)
 
+    pgcli.pgexecute.set_timezone(tzlocal.get_localzone_name())
+
     if list_databases:
         cur, headers, status = pgcli.pgexecute.full_databases()
 

pgcli/pgexecute.py

@@ -881,3 +881,10 @@ def casing(self):
 
     def explain_prefix(self):
         return "EXPLAIN (ANALYZE, COSTS, VERBOSE, BUFFERS, FORMAT JSON) "
+
+    def set_timezone(self, timezone: str):
+        query = psycopg.sql.SQL("set time zone {}").format(
+            psycopg.sql.Identifier(timezone)
+        )
+        with self.conn.cursor() as cur:
+            cur.execute(query)

setup.py

@@ -17,6 +17,7 @@
     "sqlparse >=0.3.0,<0.6",
     "configobj >= 5.0.6",
     "cli_helpers[styles] >= 2.2.1",
+    "tzlocal >= 5.2",
 ]