#! /usr/bin/python3

#  Create/view a "txt" file using the ansible playbook
# "generate-updates-uptimes-per-host-file.yml" which records the number of rpms
# available to be upgraded for ansible hosts, and the uptime of those hosts.
#  This is very helpful when doing upgrade+reboot runs, as we can easily see
# what has/hasn't been upgraded and/or rebooted.

# $0 update       = create the file and/or do backups
# $0 diff [x] [y] = see the difference between the current state and backups
# $0 uptime [x]   = see the current state, can be filtered for uptime >= x
# $0 info [x]     = see the current state, in long form, can be filtered by name
# $0 host [x]     = see the current state of a host(s), can be filtered by name
# $0 list [x]     = see the current state, can be filtered by name
# $0 history      = see history
# $0 history-keep = clenaup old history
# $0 stats [x]    = see stats, can specify a backup

# Examples:
# $0 update ... run it daily or so as root.
# $0 diff ... see what changed.
# $0 list '*.stg.*' ... see what staging looks like.
# $0 list '*copr*' ... see what copr looks like.
# $0 history-keep 4 ... keep four days of history (including today)
# $0 uptime 1d ... see what hasn't been rebooted in the last 24 hours.
# $0 uptime 25w ... see what hasn't been rebooted in too damn long

import os
import sys

import fnmatch
import glob
import locale
import shutil
import time

#  If we try to update this seconds since the file changed, flush the
# ansible FACT cache.
conf_dur_flush_cache = (60*60*8)

#  How many hosts to show in tier 4 updates/uptimes...
conf_stat_4_hosts = 4

#  Do we use a shorter duration by default (drop minutes/seconds)
conf_short_duration = True

#  Do we want a small osinfo in diff/list/etc.
conf_small_osinfo = True

#  Try to print OS/ver even nicer (when small) ... but includes spaces.
conf_align_osinfo_small = True

#  Allow 9,999,999 updates, or try to work out the correct size.
conf_fast_width_history = True

#  Remove suffix noise in names.
conf_suffix_dns_replace = {
  '.fedoraproject.org'    : '.<FP>.org',
  '.fedorainfracloud.org' : '.<FIC>.org',
}
_suffix_dns_replace = {}
for x in conf_suffix_dns_replace:
    _suffix_dns_replace[x] = False

#  Dir. where we put, and look for, the files...
conf_path = "/var/log/"

# Have nice "plain" numbers...
def _ui_int(num):
    return locale.format_string('%d', int(num), grouping=True)
try:
    locale.setlocale(locale.LC_ALL, '')
except locale.Error:
    # default to C locale if we get a failure.
    print(' Warning: Failed to set locale, defaulting to C', file=sys.stderr)
    os.environ['LC_ALL'] = 'C'
    locale.setlocale(locale.LC_ALL, 'C')


fname = conf_path + "ansible-list-updates-uptime.txt"
backup_today = time.strftime("%Y-%m-%d", time.gmtime())
fname_today = fname + '.' + backup_today

backups = sorted(x.removeprefix(fname + '.') for x in glob.glob(fname + '.*'))

tm_yesterday = int(time.time()) - (60*60*24)
backup_yesterday = time.strftime("%Y-%m-%d", time.gmtime(tm_yesterday))
fname_yesterday = fname + '.' + backup_yesterday

if len(backups) < 1 or backups[-1] != backup_today:
    fname_today = None
if len(backups) < 2 or backups[-2] != backup_yesterday:
    if fname_today is None and backups and backups[-1] == backup_yesterday:
        pass # Just missing today
    else:
        fname_yesterday = None

if len(sys.argv) >= 2:
    if '-v' in sys.argv:
        sys.argv.remove('-v') # In theory sys.argv[0] but meh
        conf_small_osinfo = False
        conf_short_duration = False
        conf_stat_4_hosts *= 4
        conf_suffix_dns_replace = {}

_max_len_osnm = 0 # osname_small
_max_len_osvr = 0 # osvers ... upto the first '.'
class Host():
    """ Class for holding the Host data from a line in the files. """

    __slots__ = ['name', 'rpms', 'uptime', 'date', 'osname', 'osvers',
                 'osname_small']

    def __init__ (self, data):
        global _max_len_osnm
        global _max_len_osvr

        self.name   = data['name']
        self.rpms   = data['rpms']
        self.uptime = data['uptime']
        self.date   = data['date']
        self.osname = data['osname']
        self.osvers = data['osvers']

        if False: pass
        elif self.osname == 'CentOS':
            osname_small = 'EL'
        elif self.osname == 'RedHat':
            osname_small = 'EL'
        elif self.osname == 'Fedora':
            osname_small = 'F'
        else:
            osname_small = self.osname[:3]
        self.osname_small = osname_small

        _max_len_osnm = max(len(osname_small), _max_len_osnm)
        vers = self.osvers
        off = vers.find('.')
        if off != -1:
            vers = vers[:off]
        _max_len_osvr = max(len(vers), _max_len_osvr)

    def __str__(self):
        return self.name

    def __eq__(self, other):
        if self.name != other.name:
            return False
        if self.rpms != other.rpms:
            return False
        if self.osname != other.osname:
            return False
        if self.osvers != other.osvers:
            return False
        return True

    def __gt__(self, other):
        if self.name > other.name:
            return True
        if self.name != other.name:
            return False

        if self.rpms > other.rpms:
            return True
        if self.rpms != other.rpms:
            return False

        if self.osname > other.osname:
            return True
        if self.osname != other.osname:
            return False

        if self.osvers > other.osvers:
            return True
        return False

    # Pretend to be a dict...
    def __getitem__(self, key):
        if key not in self.__slots__:
            raise KeyError()
        return getattr(self, key)

    @property
    def osinfo(self):
        return "%s/%s" % (self.osname, self.osvers)

    @property
    def osinfo_small(self):
        if conf_align_osinfo_small:
            vers = self.osvers
            rest = ''
            off = vers.find('.')
            if off != -1:
                rest = vers[off:]
                vers = vers[:off]
            return "%*s/%*s%s" % (_max_len_osnm, self.osname_small,
                                  _max_len_osvr, vers, rest)
        return "%s/%s" % (self.osname_small, self.osvers)


cmd = "diff"
if len(sys.argv) >= 2:
    if sys.argv[1] in ("backups", "backups-keep",
                       "hist", "history", "history-keep",
                       "diff", "diff-u",
                       "help",
                       "host", "info",
                       "list",
                       "stats",
                       "update", "update-fast", "update-flush",
                       "update-daily", "update-daily-refresh",
                       "uptime",):
        cmd = sys.argv.pop(1)

_tm_d = {'d' : 60*60*24, 'h' : 60*60, 'm' : 60, 's' : 1,
         'w' : 60*60*24*7,
         'q' : 60*60*24*7*13}
def parse_duration(seconds):
    if seconds is None:
        return None
    if seconds.isdigit():
        return int(seconds)

    ret = 0
    for mark in ('w', 'd', 'h', 'm', 's'):
        pos = seconds.find(mark)
        if pos == -1:
            continue
        val = seconds[:pos]
        seconds = seconds[pos+1:]
        if not val.isdigit():
            # dbg("!isdigit", val)
            return None
        ret += _tm_d[mark]*int(val)
    if seconds.isdigit():
        ret += int(seconds)
    elif seconds != '':
        # dbg("!empty", seconds)
        return None

    return ret


def _add_dur(dur, ret, nummod, suffix, static=False):
    mod = dur % nummod
    dur = dur // nummod
    if mod > 0 or (static and dur > 0):
        ret.append(suffix)
        if static and dur > 0:
            ret.append("%0*d" % (len(str(nummod)), mod))
        else:
            ret.append(str(mod))
    return dur

def format_duration(seconds, short=False, static=False):
    if seconds is None:
        seconds = 0
    dur = int(seconds)

    ret = []
    dur = _add_dur(dur, ret, 60, "s", static=static)
    dur = _add_dur(dur, ret, 60, "m", static=static)
    if short:
        if dur == 0 and not static:
            return '<1h'
        ret = []
    dur = _add_dur(dur, ret, 24, "h", static=static)
    dur = _add_dur(dur, ret,  7, "d", static=static)
    if dur > 0:
        ret.append("w")
        ret.append(str(dur))
    return "".join(reversed(ret))

# Duration in UI for lists/etc.
def _ui_dur(dur):
    return format_duration(dur, short=conf_short_duration, static=True)

def _main_file_recent():
    f1 = os.stat(fname)

    if (int(time.time()) - f1.st_mtime) > (60*60*24):
        return False
    return True

def _backup_today_identical():
    if fname_today is None:
        return False

    b = backup_today
    f1 = os.stat(fname)
    f2 = os.stat(fname + '.' + b)
    if f1.st_size != f2.st_size:
        return False
    if (f1.st_mtime - f2.st_mtime) > 64: # seconds, just a copy
        return False
    return True

cmp_arg = False
cmp = None
#  This does arguments for a bunch of commands, like stats/list/etc.
# by using fname1() after, which looks at cmp_arg.
# But also does diff arguments.
def _cmp_arg():
    global cmp
    global cmp_arg

    if len(sys.argv) < 2 or sys.argv[1] == "main":
        if len(sys.argv) >= 2:
            sys.argv.pop(1)
        cmp = backups[-1] # Most recent
        if len(backups) > 1 and _backup_today_identical():
            # Eg. if you just do one update a day, you want to cmp vs.
            # the previous day, not today.
            cmp = backups[-2]
    elif sys.argv[1] == "today" and fname_today is not None:
        cmp = backup_today
        cmp_arg = True
    elif sys.argv[1] == "yesterday" and fname_yesterday is not None:
        cmp = backup_yesterday
        cmp_arg = True
    elif sys.argv[1] not in backups:
        _usage()
        print("Backups:", ", ".join(backups))
        sys.exit(1)
    else:
        cmp = sys.argv[1]
        cmp_arg = True

_max_len_osnm = 0 # osname_small
_max_len_osvr = 0 # osvers ... upto the first '.'
def line2data(line):
    global _max_len_osnm
    global _max_len_osvr

    name, rpms, uptime, date = line.split(' ', 3)
    osname = "Unknown"
    osvers = "?"
    if ' ' in date:
        date, osname, osvers = date.split(' ', 2)

    rpms = int(rpms)
    uptime = int(uptime)

    return Host(locals())

def lines2datas(lines):
    return (line2data(line) for line in lines)

# Filter datas using name as a filename wildcard match.
def filter_name_datas(datas, name):
    for data in datas:
        if not fnmatch.fnmatch(data.name, name):
            continue
        yield data

# Filter datas using uptime as a minium.
def filter_uptime_datas(datas, uptime):
    for data in datas:
        if data.uptime < uptime:
            continue
        yield data

# Sub. suffix of DNS names for UI
def _ui_name(name):
    for suffix in conf_suffix_dns_replace:
        if name.endswith(suffix):
            _suffix_dns_replace[suffix] = True
            return name[:-len(suffix)] + conf_suffix_dns_replace[suffix]
    return name

def _ui_osinfo(data):
    if conf_small_osinfo:
        return data.osinfo_small
    return data.osinfo

# Reset the usage after _max_update()
def _reset_ui_name():
    for suffix in sorted(_suffix_dns_replace):
        _suffix_dns_replace[suffix] = False

# Explain if we used any suffix subs.
def _explain_ui_name():
    done = False
    pre = "* NOTE:"
    for suffix in sorted(_suffix_dns_replace):
        if _suffix_dns_replace[suffix]:
            print("%s %12s = %s" % (pre,conf_suffix_dns_replace[suffix],suffix))
            pre = "      :"
            done = True
    if done:
        print("      : Use -v to show full names.")

def fname2lines(fname):
    return [x.strip() for x in open(fname).readlines()]

def bfname2lines(b):
    return fname2lines(fname + '.' + b)

def fname1():
    if cmp_arg:
        return bfname2lines(cmp)
    return fname2lines(fname)

_max_len_name = 0
_max_len_rpms = 0 # Number of rpm updates via. _ui_int().
_max_len_upts = 0 # Uptime duration with short=True
_max_len_date = 0 # 2025-08-04 = 4+1+2+1+2
_max_terminal_width = shutil.get_terminal_size().columns
if _max_terminal_width < 20:
    _max_terminal_width = 80
_max_terminal_width -= 14
def _max_update(datas):
    for data in datas:
        _max_update_data(data)

def _max_update_data(data):
    global _max_len_name
    global _max_len_rpms
    global _max_len_upts
    global _max_len_date

    name = _ui_name(data.name)
    if len(name) > _max_len_name:
        _max_len_name = len(name)

    rpms = _ui_int(data.rpms)
    if len(rpms) > _max_len_rpms:
        _max_len_rpms = len(rpms)

    upts_len = len(_ui_dur(data.uptime))
    if upts_len > _max_len_upts:
        _max_len_upts = upts_len

    if len(data.date) > _max_len_date:
        _max_len_date = len(data.date)

def _max_update_correct(prefix):
    global _max_len_name
    mw = _max_terminal_width - len(prefix)
    while _max_len_name + _max_len_rpms + _max_len_upts + _max_len_date >= mw:
        _max_len_name -= 1

# Return stats for updates added/deleted between two data sets.
def _diffstats(data1, data2):
  uadd, udel = 0, 0

  data1 = list(sorted(data1))
  data2 = list(sorted(data2))
  while len(data1) > 0 or len(data2) > 0:
    if len(data1) <= 0:
        d2 = data2.pop(0)
        uadd += d2.rpms
        continue
    if len(data2) <= 0:
        d1 = data1.pop(0)
        udel -= d1.rpms
        continue

    d1 = data1[0]
    d2 = data2[0]

    if d1.name < d2.name:
        udel -= d1.rpms
        data1.pop(0)
        continue

    if d1.name > d2.name:
        uadd += d2.rpms
        data2.pop(0)
        continue

    if d1 == d2:
        data1.pop(0)
        data2.pop(0)
        continue

    if d1.osinfo != d2.osinfo:
        udel -= d1.rpms
        uadd += d2.rpms
        data1.pop(0)
        data2.pop(0)
        continue

    # Now name is eq and osinfo is eq
    #  So either new updates arrived, or we installed some and they went
    # down ... alas. we can't tell if both happened.
    if d1.rpms > d2.rpms:
        udel -= d1.rpms - d2.rpms
    if d1.rpms < d2.rpms:
        uadd += d2.rpms - d1.rpms
    data1.pop(0)
    data2.pop(0)

  # diffstat returns...
  return uadd, udel

def _ui_diffstats(data1, data2):
    cmpds = _diffstats(data1, data2)
    return _ui_int(cmpds[0]), _ui_int(cmpds[1])



# This is the real __main__ start ...
def _usage():
  prog = "updates+uptime"
  if sys.argv:
      prog = os.path.basename(sys.argv[0])
  pl = " " * len(prog)
  print("""
 Usage: %s <cmd>
   Cmds:
        help
            = This message.

        diff         [backup1] [backup2]
            = See the difference between the current state and backups.
        diff-u       [backup1] [backup2]
            = Shows before/after instead of modified (like diff -u).

        history
            = Show history data.
        history-keep [days]
            = Cleanup old history.

        host         [host*] [backup]
            = See the current state of a host(s), can be filtered by name.
        info         [host*] [backup]
            = See the current state, in long form, can be filtered by name.
        list         [host*] [backup]
            = See the current state, can be filtered by name.

        stats        [backup]
            = Show stats.

        update
            = Create the file and/or do backups.
        update-fast
            = Create the file.
        update-flush
            = Create the file, after flushing ansible caches.
        update-daily
            = update-flush and do backups.
        update-daily-refresh
            = update-daily with new main file.

        uptime       [duration] [backup]
            = See the current state, can be filtered for uptime >= duration.
""" % (prog,))

if cmd == "help":
  _usage()

def _backup_suffix(backup):
  suffix = ''
  if backup == backup_today:
      if ident:
          suffix = ' (today, is identical)'
      else:
          suffix = ' (today)'
  if backup == backup_yesterday:
    suffix = ' (yesterday)'
  return suffix

if cmd in ("backups", "hist", "history"):
  ident = _backup_today_identical()
  print("History:")
  last_name = "main"
  last_data = list(sorted(lines2datas(fname2lines(fname))))
  last_suff = ""

  #  We _could_ open+read+etc each file, just to find out the max updates for
  # all hist ... but len("Updates")+2=9 which means 9,999,999 updates)
  hl = len("Hosts")
  ul = len("Updates") + 2
  if conf_fast_width_history:
      ul += 2
  else:
  #  Whatever, it's less memory than holding all history at once if you want
  # to enable it..
      for backup in reversed(backups):
          data = list(sorted(lines2datas(bfname2lines(backup))))
          updates = _ui_int(sum(d.rpms for d in data))
          hl = max(hl, len(_ui_int(len(data))))
          ul = max(ul, len(updates))

  print(" %10s %*s %*s  %*s %*s" % ("Day", hl, "Hosts",
        ul, "Updates", ul, "Avail", ul, "Inst."))

  for backup in reversed(backups):
      data = list(sorted(lines2datas(bfname2lines(backup))))
      updates = _ui_int(sum(d.rpms for d in last_data))
      ul = max(ul, len(updates))
      cmpds = _ui_diffstats(data.copy(), last_data.copy())
      print(' %10s %*s %*s, %*s %*s, %s' % (last_name,
            hl, _ui_int(len(last_data)),
            ul, updates, ul, cmpds[0], ul+1, cmpds[1], last_suff))
      last_name = backup
      last_data = data
      last_suff = _backup_suffix(backup)
  updates = _ui_int(sum(d.rpms for d in last_data))
  print(' %10s %*s %*s %s' % (last_name, hl, _ui_int(len(last_data)),
        ul, updates, last_suff))

if cmd in ("backups-keep", "history-keep"):
  keep = 8
  if len(sys.argv) >= 2:
      keep = int(sys.argv.pop(1))
  if keep <= 0:
      _usage()
      sys.exit(1)
  while keep < len(backups):
      # We just keep the newest N
      b = backups.pop(0)
      print("Removing:", b)
      fn = fname + '.' + b
      os.unlink(fn)

if cmd == "update":
  cmd = "update-flush"
  if not os.path.exists(fname):
    cmd = "update-daily" # This does the sorting etc.
  elif fname_today is None:
    cmd = "update-daily"
  else:
    mtime = os.path.getmtime(fname)
    if (int(time.time()) - mtime) > conf_dur_flush_cache:
        cmd = "update-fast"

if cmd == "update-flush": # Get the latest uptime.
  os.chdir("/srv/web/infra/ansible/playbooks")
  os.system("ansible-playbook generate-updates-uptimes-per-host-file.yml -t updates --flush-cache")
if cmd == "update-fast": # Use ansible FACT cache for uptime.
  os.chdir("/srv/web/infra/ansible/playbooks")
  os.system("ansible-playbook generate-updates-uptimes-per-host-file.yml -t updates")
if cmd == "update-daily-refresh": # Also recreate the main file.
  if os.path.exists(fname):
    os.unlink(fname)
  cmd = "update-daily"
if cmd == "update-daily": # Also create backup file.
  os.chdir("/srv/web/infra/ansible/playbooks")
  os.system("ansible-playbook generate-updates-uptimes-per-host-file.yml --flush-cache")

# Below here are the query commands, stuff needs to exist at this point.
if not os.path.exists(fname):
    print(" Error: No main file. Run update sub-command", file=sys.stderr)
    sys.exit(4)
if not _main_file_recent():
    print(" Warning: Main file is old. Run update sub-command", file=sys.stderr)
if fname_today is None:
    print(" Warning: Backup for today does not exist!", file=sys.stderr)
if fname_yesterday is None:
    print(" Warning: Backup for yesterday does not exist!", file=sys.stderr)

def _cli_match_host(data):
  if len(sys.argv) >= 2:
      host = sys.argv.pop(1)
      print("Matching:", host)
      data = filter_name_datas(data, host)
      data = list(data)
      if not data:
          print("Not host(s) matched:", host)
          sys.exit(2)
  return data

if cmd == "stats":
  _cmp_arg()
  data = lines2datas(fname1())
  if cmp_arg:
      sys.argv.pop(1)
  data = list(_cli_match_host(data))
  # Basically we have hosts/updates/uptime and we want 4 tiers of data:
  # 1. All. 2. OS name (Eg. Fedora). 3. OS name+version (Eg. Fedora 42).
  # 4. For updates/uptime a "few" hosts with the biggest numbers.
  osdata = {'hosts' : {}, 'updates' : {}, 'uptimes' : {}, 'vers' : {}}
  updates = 0 # total updates
  most = [] # Tier 4, for updates
  awake = 0 # total uptime
  awakest = [] # Tier 4, for uptime

  conf_suffix_dns_replace = {} # Turn off shortened names for stats...
  for d2 in data:
    # Tidy UI for OS names with only one version...
    if d2.osname not in osdata['vers']:
        osdata['vers'][d2.osname] = set()
    osdata['vers'][d2.osname].add(d2.osinfo)

    # Tier 2/3 hosts...
    if d2.osname not in osdata['hosts']:
        osdata['hosts'][d2.osname] = 0
    osdata['hosts'][d2.osname] += 1

    if d2.osinfo not in osdata['hosts']:
        osdata['hosts'][d2.osinfo] = 0
    osdata['hosts'][d2.osinfo] += 1

    updates += d2.rpms
    # Tier 2/3 updates...
    if d2.osname not in osdata['updates']:
        osdata['updates'][d2.osname] = 0
    osdata['updates'][d2.osname] += d2.rpms

    if d2.osinfo not in osdata['updates']:
        osdata['updates'][d2.osinfo] = 0
    osdata['updates'][d2.osinfo] += d2.rpms
    # Tier 4 updates...
    most.append((d2.rpms, d2.uptime, d2))
    most.sort()
    while len(most) > conf_stat_4_hosts:
        most.pop(0)

    awake += d2.uptime
    # Tier 2/3 uptimes...
    if d2.osname not in osdata['uptimes']:
        osdata['uptimes'][d2.osname] = 0
    osdata['uptimes'][d2.osname] += d2.uptime

    if d2.osinfo not in osdata['uptimes']:
        osdata['uptimes'][d2.osinfo] = 0
    osdata['uptimes'][d2.osinfo] += d2.uptime
    # Tier 4 uptimes...
    awakest.append((d2.uptime, d2.rpms, d2))
    awakest.sort()
    while len(awakest) > conf_stat_4_hosts:
        awakest.pop(0)

  # Print "stats"
  # _max_update(data)
  # Do this by hand...
  _max_len_name = max((len(d.name) for d in data))
  _max_len_rpms = max(len("Updates"), len(_ui_int(updates)))
  _max_len_upts = len(_ui_dur(awake))
  _max_len_date = 0
  _max_update_correct(' ')
  print("%-16s  %6s %*s %*s" % ("OS", "Hosts", 
        _max_len_rpms, "Updates", _max_len_upts, "Uptime"))
  print("-" * (16+2+6+1+_max_len_rpms+1+_max_len_upts))
  print("%-16s: %6s %*s %*s" % ("All", _ui_int(len(data)),
        _max_len_rpms, _ui_int(updates), _max_len_upts, _ui_dur(awake)))
  subprefix = ''
  subplen   = 12
  for osi in sorted(osdata['hosts']):
    if '/' not in osi:
        if len(osdata['vers'][osi]) == 1:
            subprefix = ''
            subplen   = 14
            continue
        subprefix = '  '
        subplen   = 12
        print("  %-14s: %6s %*s %*s" % (osi, _ui_int(osdata['hosts'][osi]),
              _max_len_rpms, _ui_int(osdata['updates'][osi]),
              _max_len_upts, _ui_dur(osdata['uptimes'][osi])))
    if '/' in osi:
        print("  %s%-*s: %6s %*s %*s" % (subprefix, subplen, osi,
              _ui_int(osdata['hosts'][osi]),
              _max_len_rpms, _ui_int(osdata['updates'][osi]),
              _max_len_upts, _ui_dur(osdata['uptimes'][osi])))
  print("-" * (16+2+6+1+_max_len_rpms+1+_max_len_upts))
  # Redo the lengths, because it's real hostname data now...
  _max_update(data)
  _max_len_date = 0
  _max_update_correct(' ')
  if most:
    # print("")
    print("Hosts with the most Updates:")
  for m in most:
    print("  %-*s %*s %*s %s" % (_max_len_name, m[2],
          _max_len_rpms, _ui_int(m[0]), _max_len_upts, _ui_dur(m[1]),
          _ui_osinfo(m[2])))
  if awakest:
    # print("")
    print("Hosts with the most Uptime:")
  for a in awakest:
    print("  %-*s %*s %*s %s" % (_max_len_name, a[2],
          _max_len_rpms, _ui_int(a[1]), _max_len_upts, _ui_dur(a[0]),
          _ui_osinfo(a[2])))
  _explain_ui_name()

def _print_info(host, lines):
  hosts = []
  for x in lines:
      x = line2data(x)
      if fnmatch.fnmatch(x.name, host):
          hosts.append(x)
  if not hosts:
      print("Not host(s) matched:", host)
      sys.exit(2)
  for host in hosts:
    print("Host:", host.name)
    print("        OS:", host.osinfo)
    print("   Updates:", _ui_int(host.rpms))
    print("    Uptime:", format_duration(host.uptime)) # !ui_dur
    print("   Checked:", host.date)

if cmd in ("host", "info"):
  if cmd == "host":
      host = "batcave*"
  else:
      host = "*"
  if len(sys.argv) >= 2:
      host = sys.argv.pop(1)
  if len(sys.argv) >= 2 and sys.argv[1] == "all":
      for b in backups:
          print("Backup:", b)
          _print_info(host, bfname2lines(b))
      sys.argv = [sys.argv[0]]
      print("Main:")
  _cmp_arg()
  _print_info(host, fname1())

def _print_line(prefix, data):
    print("%s%-*s %*s %*s %*s %s" % (prefix,
          _max_len_name, _ui_name(data.name),
          _max_len_rpms, _ui_int(data.rpms),
          _max_len_upts, _ui_dur(data.uptime),
          _max_len_date, data.date, _ui_osinfo(data)))


if cmd == "list":
  host = "*"
  if len(sys.argv) >= 2:
      host = sys.argv.pop(1)

  _cmp_arg()
  data = lines2datas(fname1())
  data = list(filter_name_datas(data, host))
  _max_update(data)
  _max_update_correct('')
  for d1 in data:
      _print_line('', d1)
  _explain_ui_name()

if cmd == "uptime":
  age = 0
  if len(sys.argv) >= 2:
      age = parse_duration(sys.argv.pop(1))

  _cmp_arg()
  data = lines2datas(fname1())
  data = list(filter_uptime_datas(data, age))
  _max_update(data)
  _max_update_correct('')
  for d1 in data:
      _print_line('', d1)
  _explain_ui_name()

if cmd in ("diff", "diff-u"):
  _cmp_arg()
  fn1 = fname + '.' + cmp
  fn2 = fname
  data1 = fname2lines(fn1)
  if len(sys.argv) >= 3:
    # Doing a diff. between two backups...
    if sys.argv[2] == 'today' and fname_today is not None:
        fn2 = fname_today
    if sys.argv[2] == 'yesterday' and fname_yesterday is not None:
        fn2 = fname_yesterday
    if sys.argv[2] in backups:
        fn2 = fname + '.' + sys.argv[2]
  data2 = fname2lines(fn2)
  print("diff %s %s" % (fn1, fn2), file=sys.stderr)
  data1 = list(sorted(lines2datas(data1)))
  data2 = list(sorted(lines2datas(data2)))
  hosts = _ui_int(len(data2))
  updates = _ui_int(sum(d.rpms for d in data2))
  ul = len(updates)
  cmpds = _ui_diffstats(data1.copy(), data2.copy())
  _max_update(data1)
  _max_update(data2)
  _max_update_correct(' ')
  while len(data1) > 0 or len(data2) > 0:
    if len(data1) <= 0:
        _print_line('+', data2[0])
        data2.pop(0)
        continue
    if len(data2) <= 0:
        _print_line('-', data1[0])
        data1.pop(0)
        continue

    d1 = data1[0]
    d2 = data2[0]

    if d1.name < d2.name:
        _print_line('-', d1)
        data1.pop(0)
        continue

    if d1.name > d2.name:
        _print_line('+', d2)
        data2.pop(0)
        continue

    if d1 == d2:
        _print_line(' ', d2)
        data1.pop(0)
        data2.pop(0)
        continue

    if cmd == "diff-u":
        _print_line('-', d1)
        data1.pop(0)
        _print_line('+', d2)
        data2.pop(0)
        continue

    # diff
    data1.pop(0)
    _print_line('!', d2)
    data2.pop(0)

    continue
  print('hosts=%s updates=%s (a=%s i=%s)' % (hosts, updates, cmpds[0],cmpds[1]))
  _explain_ui_name()