stats.py

import click
import json
import time
import os

actions="""
    var result="%s";
    document.write(result);
"""
extentions = ['sql.zip', 'pbf', 'shp.zip', 'json', 'kml']

def sizeof_fmt(num, suffix='B'):
    for unit in ['','K','M','G','T','P','E','Z']:
        if abs(num) < 1024.0:
            return "%3.1f%s%s" % (num, unit, suffix)
        num /= 1024.0
    return "%.1f%s%s" % (num, 'Yi', suffix)

@click.command()
@click.option('--files', help='Filenames.')
@click.option('--name', help='Project Name.')
def stats(name, files):
    """Simple program that generates stats for OSM exports"""
    files = files.split(" ")
    data = {}

    out = "<ul>"
    for table in sorted(files):
        
        out = out + "<li><span class='name'>%s</span> " % table.title().replace("_", " ")
        files = {}
        for extention in extentions:
            filename = "%s/%s.%s" % (name, table, extention)
            size = os.path.getsize(filename)
            date = time.ctime(os.path.getmtime(filename))
            download = '%s' % filename.replace(name + '/', '')
            files[extention] = ('data/%s' % filename, size, date)
            out = out + "<div class='box'>|    <a href='%s'>%s</a> <span class='size'>%s</span></div>" % (download, extention.upper(), sizeof_fmt(size))
        gpkg_filename = "%s/%s.gpkg" % (name, name)
        gpkg_size = os.path.getsize(gpkg_filename)
        gpkg_download = '%s' % gpkg_filename.replace(name + '/', '')
        out = out + "<div class='box'>|    <a href='%s'>%s</a> <span class='size'>%s</span></div>" % (gpkg_download, 'GPKG', sizeof_fmt(gpkg_size))

        out = out + "|    %s</li>" % date
    out= out + "</ul>"
    
    click.echo(actions % out)

if __name__ == '__main__':
    stats()