#!/usr/bin/env python2
# -*- mode: python; coding: iso-8859-1 -*-
# Copyright © 2001, 2002 Free Translation Project.
# Copyright © 2000 Progiciels Bourbeau-Pinard inc.
# François Pinard <pinard@iro.umontreal.ca>, 2000.

"""\
Adjust the database holding PO file statistics.

Usage: postats-data [OPTION] [POFILE]...

  -d  Delete entries for which the PO file does not exist
  -i  Re-initialise the whole database (may be long)
  -t  Re-initialise just the template information
  -v  Entertain the user by telling what goes on
"""

import os, sys
p =  os.path.expanduser('~/po/web/lib')
if os.path.exists(p):
    sys.path.insert(0, p)
else:
    p = "/home/ftp/pub/po/web/lib"
    if os.path.exists(p):
         sys.path.insert(0, p)

import getopt, re, UserDict
import config, data, po, registry

def main(*arguments):
    if not arguments:
        sys.stdout.write(__doc__)
        sys.exit(0)
    delete_option = initialize_option = template_option = verbose_option = 0
    options, arguments = getopt.getopt(arguments, 'ditv')
    for option, value in options:
        if option == '-d':
            delete_option = 1
        elif option == '-i':
            initialize_option = 1
        elif option == '-t':
            template_option = 1
        elif option == '-v':
            verbose_option = 1
    assert not initialize_option or not delete_option
    if initialize_option:
        assert len(arguments) == 0
    #elif not delete_option:
    #   assert len(argument) > 0
    if initialize_option:
        postats = UserDict.UserDict()
        os.path.walk('%s/trans' % config.top_directory,
                     walker, (postats, verbose_option))
        gen_potstats(postats, verbose_option)
    elif template_option:
        postats = data.load_postats()
        gen_potstats(postats, verbose_option)
    else:
        postats = data.load_postats()
        if delete_option:
            for key in postats.keys():
                domain, version, team = key
                name = registry.archive_path(domain, version, team)
                if not os.path.exists(name):
                    if verbose_option:
                        sys.stderr.write('d - %s\n' % name)
                    del postats[key]
        for name in arguments:
            update(postats, name, verbose_option)
    data.save_postats(postats)

def walker((postats, verbose), dirname, bases):
    if verbose:
        bases.sort()
    for base in bases:
        if os.path.splitext(base)[1] == '.po':
            update(postats, '%s/%s' % (dirname, base), verbose)

def update(postats, name, verbose):
    hints = registry.hints(name)
    if hints.domain and hints.version and hints.team:
        if verbose:
            sys.stderr.write('a - %s...' % name)
        entries = po.read(name)
        header = po.header(entries)
        translator, mailto = header['last-translator'], None
        match = re.search('(.*[^ ]) <(.*)>', translator)
        if match:
            translator, mailto = match.group(1, 2)
        try:
            unicode(translator,"ascii")
        except UnicodeError:
            charset = None
            for f in header['content-type'].split(";"):
                f = f.strip()
                if f.startswith("charset="):
                    charset = f[8:]
            if charset:
                translator = unicode(translator,charset).encode("utf-8")
        stats = po.stats(entries)
        key = hints.domain.name, hints.version.name, hints.team.name
        total = stats['translated'] + stats['untranslated'] + stats['fuzzy']
        postats[key] = (translator, mailto, stats['translated'], total,
                        stats['translated_length'], stats['total_length'])
        if verbose:
            sys.stderr.write('\n')
    else:
        sys.stderr.write(_('%s: Not enough hints for a key\n') % name)

def gen_potstats(postats, verbose):
    versions = {}
    for f in os.listdir('%s/domains/POT' % config.top_directory):
        try:
            h = registry.hints(f)
        except ValueError:
            continue
        except KeyError:
            print "BAD FILE",f
            continue
        try:
            oldver = versions[h.domain.name][0]
        except KeyError:
            versions[h.domain.name] = h.version, f
            continue
        if h.version > oldver:
            versions[h.domain.name] = h.version, f
    postats.potstats = {}
    for domain, (version, file) in versions.items():
        if verbose:
            sys.stderr.write("%s ... " % file)
        entries = po.read('%s/domains/POT/%s' % (config.top_directory, file))
        stats = po.stats(entries)
        # aegis.pot has translations
	# assert stats['translated']==0,file
        assert stats['fuzzy']==0,file
        postats.potstats[domain] = (file, stats['untranslated'])
        if verbose:
            sys.stderr.write("%d\n" % stats['untranslated'])
            

if __name__ == '__main__':
    apply(main, tuple(sys.argv[1:]))