Backend.py

import os
import datetime, pytz
import HumanTime
import osm.changeset as oc
import pprint

class Backend(object):
    def __init__(self, config, subcfg):
        self.debug = False
        self.generation = -1 # DB might start out with 'None'
        self.config = config
        self.subcfg = subcfg

    def build_filename(self, globalconfig, subcfg, filename_key='filename'):
        if filename_key:
            filename = subcfg[filename_key]
        else:
            filename = ''
        if 'path' in subcfg:
            filename = os.path.join(subcfg['path'], filename)
        return os.path.join(globalconfig.getpath('path', 'tracker'), filename)

    def print_state(self, state, update_reason):
        if update_reason!='new_generation.osmtracker':
            return
        time = state.generation_timestamp.strftime('%Y:%m:%d %H:%M:%S')
        print 'Generation {} @ {}'.format(state.generation, time)
        if self.generation != state.generation:
            self.generation = state.generation

            self.print_chgsets(state.area_chgsets,
                               state.area_chgsets_info)

    def _pluS(self, num):
        '''Return plural s'''
        if num==1:
            return ''
        return 's'

    def _i2s(self, num):
        '''Int to string using human rounding'''
        sign = ''
        if num < 0:
            sign = '-'
            num = -num
        ss = str(num)
        txt = None
        if len(ss) <=8:
            for pp in range(0, len(ss), 3):
                if txt:
                    txt = ss[-3:]+','+txt
                else:
                    txt = ss[-3:]
                ss = ss[:-3]
            return sign+txt
        else:
            return sign+str(num) # TODO: Better rounding for large numbers

    def merge_int_dict(self, a, b):
        '''Recursively merge dictionaries where values are either numbers or other dicts'''
        for k,v in b.iteritems():
            if type(v) is int or type(v) is float:
                a[k] = a.get(k, 0)+v
            else:
                a[k] = self.merge_int_dict(a.get(k, dict()), v)
        return a