sort table in report tool

Author: vadv

mamonsu/tools/report/pgsql.py

@@ -2,7 +2,7 @@
 
 import logging
 import time
-
+from collections import OrderedDict
 from mamonsu.plugins.pgsql.pool import Pooler
 from mamonsu.tools.report.format import header_h1, key_val_h1, humansize
 
@@ -144,7 +144,9 @@ class PostgresInfo(object):
  else round(100*io.heap_blks_hit::float8/(io.heap_blks_read+io.heap_blks_hit)::float8) end as "heap_hit_perc",
 
  case coalesce(io.idx_blks_read + io.idx_blks_hit, 0) when 0 then 0
- else round(100*io.idx_blks_hit::float8/(io.idx_blks_read + io.idx_blks_hit)::float8) end as "idx_hit_perc"
+ else round(100*io.idx_blks_hit::float8/(io.idx_blks_read + io.idx_blks_hit)::float8) end as "idx_hit_perc",
+
+ b.size as "size_b"
 
 from
  pg_catalog.pg_stat_all_tables as s
@@ -301,17 +303,21 @@ def _collect_connections(self):
  return result
 
  def _collect_biggest(self):
- result = {}
+ result, sizes, sorted_result = {}, {}, OrderedDict({})
  for info_dbs in Pooler.query('select datname \
  from pg_catalog.pg_database where datistemplate = false'):
  try:
  for info in Pooler.query(self.BigTableInfo[0], info_dbs[0]):
  table_name = '{0}.{1}'.format(info_dbs[0], info[0])
  result[table_name] = ''
- for val in info[1:]:
+ values = info[1:] # remove first elements (table name with schema)
+ sizes[table_name] = values.pop() # size in bytes in last element
+ for val in values:
  result[table_name] = "{0}\t\t{1}".format(
  result[table_name], val)
  except Exception as e:
  logging.error("Connect to db {0} error: {1}".format(
  info_dbs[0], e))
- return result
+ for table_name in sorted(result, key=sizes.__getitem__, reverse=True):
+ sorted_result[table_name] = result[table_name]
+ return sorted_result