Python3 changes

pg_view/models/collector_partition.py

@@ -1,14 +1,18 @@
+import os
 import time
 from multiprocessing import Process
 
-import os
 import psutil
+import sys
 
 from pg_view.consts import RD, TICK_LENGTH, SECTOR_SIZE
 from pg_view.formatters import StatusFormatter, FnFormatter
 from pg_view.models.collector_base import BaseStatCollector, logger
 from pg_view.models.displayers import COLALIGN
 
+if sys.hexversion >= 0x03000000:
+    long = int
+
 
 class PartitionStatCollector(BaseStatCollector):
     """Collect statistics about PostgreSQL partitions """

pg_view/models/collector_pg.py

@@ -27,8 +27,8 @@
 
 def dbversion_as_float(server_version):
     version_num = server_version
-    version_num /= 100
-    return float('{0}.{1}'.format(version_num / 100, version_num % 100))
+    version_num //= 100
+    return float('{0}.{1}'.format(version_num // 100, version_num % 100))
 
 
 def process_sort_key(process):

setup.py

@@ -15,7 +15,7 @@
 def read_module(path):
     data = {}
     with open(path, 'r') as fd:
-        exec (fd.read(), data)
+        exec(fd.read(), data)
     return data
 
 

tests/common.py

@@ -1,4 +1,7 @@
-from StringIO import StringIO
+try:
+    from io import StringIO
+except ImportError:
+    from StringIO import StringIO
 
 import os
 

tests/test_db_client.py

@@ -10,7 +10,7 @@
 
 class DbClientUtilsTest(TestCase):
     @mock.patch('pg_view.models.clients.logger')
-    @mock.patch('__builtin__.open')
+    @mock.patch('pg_view.models.clients.open', create=True)
     def test_read_postmaster_pid_should_return_none_when_error(self, mocked_open, mocked_logger):
         mocked_open.side_effect = Exception
         data = read_postmaster_pid('/var/lib/postgresql/9.3/main', 'default')
@@ -20,7 +20,7 @@ def test_read_postmaster_pid_should_return_none_when_error(self, mocked_open, mo
             expected_msg.format(name='default', wd='/var/lib/postgresql/9.3/main'))
 
     @mock.patch('pg_view.models.clients.logger')
-    @mock.patch('__builtin__.open')
+    @mock.patch('pg_view.models.clients.open', create=True)
     def test_read_postmaster_pid_should_return_none_when_error_strip(self, mocked_open, mocked_logger):
         mocked_open.return_value.readline.return_value = []
         data = read_postmaster_pid('/var/lib/postgresql/9.3/main', 'default')
@@ -29,7 +29,7 @@ def test_read_postmaster_pid_should_return_none_when_error_strip(self, mocked_op
         mocked_logger.error.assert_called_with(
             expected_msg.format(name='default', wd='/var/lib/postgresql/9.3/main'))
 
-    @mock.patch('__builtin__.open')
+    @mock.patch('pg_view.models.clients.open', create=True)
     def test_read_postmaster_pid_should_return_pid_when_read_file(self, mocked_open):
         mocked_open.return_value.readline.return_value = '123 '
         data = read_postmaster_pid('/var/lib/postgresql/9.3/main', 'default')
@@ -217,11 +217,9 @@ def test_establish_user_defined_connection_should_raise_error_when_cant_connect(
         with self.assertRaises(NotConnectedError):
             client.establish_user_defined_connection('instance', [])
 
-        mocked_logger.error.assert_has_calls([
-            mock.call(
-                "failed to establish connection to instance via {'host': 'localhost', 'database': 'db', 'port': '5432', 'user': 'user'}"),
-            mock.call('PostgreSQL exception: ')]
-        )
+        expected_msg = "failed to establish connection to instance via {0}".format(
+            client.connection_builder.build_connection())
+        mocked_logger.error.assert_has_calls([mock.call(expected_msg), mock.call('PostgreSQL exception: ')])
 
     @mock.patch('pg_view.models.clients.logger')
     @mock.patch('pg_view.models.clients.psycopg2')
@@ -238,7 +236,8 @@ def test_establish_user_defined_connection_should_raise_error_when_not_pid_postm
         with self.assertRaises(NoPidConnectionError):
             client.establish_user_defined_connection('default', [])
 
-        expected_msg = "failed to read pid of the postmaster on {'host': 'localhost', 'database': 'db', 'port': '5432', 'user': 'user'}"
+        expected_msg = "failed to read pid of the postmaster on {0}".format(
+            client.connection_builder.build_connection())
         mocked_logger.error.assert_called_once_with(expected_msg)
 
     @mock.patch('pg_view.models.clients.logger')

tests/test_models_pg.py

@@ -158,7 +158,7 @@ def test__read_proc_should_return_data_when_process_ok(self, mocked_psutil_proce
 
             'guest_time': 0.0,
             'starttime': datetime.datetime.fromtimestamp(1480777289.0),
-            'delayacct_blkio_ticks': 0L,
+            'delayacct_blkio_ticks': 0,
             'cmdline': 'backend'
         }
         self.assertEqual(expected_proc_stats, proc_stats)
@@ -285,8 +285,8 @@ def test_refresh_should_return_results_when_ok(self, mocked__do_refresh, mocked_
             'priority': 10,
             'vsize': 252428288,
             'guest_time': 0.0,
-            'starttime': 911L,
-            'delayacct_blkio_ticks': 1L,
+            'starttime': 911,
+            'delayacct_blkio_ticks': 1,
             'cmdline': 'backend'
         }
 
@@ -310,14 +310,14 @@ def test_refresh_should_return_results_when_ok(self, mocked__do_refresh, mocked_
             'status': 'status',
             'write_bytes': 1,
             'vsize': 252428288,
-            'delayacct_blkio_ticks': 1L,
+            'delayacct_blkio_ticks': 1,
             'pid': 1049,
             'priority': 10,
             'cmdline': 'backend',
             'read_bytes': 655,
             'uss': 10,
             'stime': 0.0001,
-            'starttime': 911L,
+            'starttime': 911,
             'utime': 0.0002,
             'type': 'unknown',
             'guest_time': 0.0,