faker.py

Go to the documentation of this file.

import os, logging, time
from datetime import datetime

log = logging.getLogger(__name__)


class Faker(list):
    """
    create fake source instances and insert them 
    """
    def __init__(self, srcs, cfg ):
        self.cfg = cfg
        log.info("faking srcs %s from %s  " % ( srcs, srcs[0].db ) )
        for src in srcs:
            self.append(src)
        #
        self.sleep = cfg.pop('sleep') 
        self.interval = cfg.pop('interval') 
        self.timestart = cfg.pop('faker_timestart')

        self.maxiter = cfg.pop('maxiter')
        self.profile = cfg.pop('profile', None)  

        if len(cfg)>0:
            log.warn("%s unpopped cfg %r " % (self.__class__.__name__, cfg ) )


    def _insertfake(self, index ):
        """
        :param index:  None or zero based fake index, typically the iteration loop counter 

        An index of `None` causes and SA source query to be performed to determine the 
        `id` of the last source entry for each source. 
        """
        for src in self:
            
            inst = src()
            if index == None:        
                last = src.last()   
                id = 1000 if last == None else last.id
                id += 1
            else:
                id = index  
            dt = self.timestart + (index*self.interval)

            #
            # set defaults here, users can override if needed 
            # NOPE :. not so simple as inst could be a join with multiple id/date_time to set 
            # so for now leave it to specialized code
            #
            # setattr( inst , 'id' , id ) 
            # setattr( inst , 'date_time' , dt ) 
            #

            self.fake( inst , id , dt )

            log.info("IF %-3d %s %s %r " % (id, inst.date_time.strftime("%Y-%m-%d %H:%M:%S"), src.__name__, filter(lambda _:type(_) != datetime, inst.asdict.values() )))
            db = src.db
            db.add(inst)   
            db.commit()

    def __repr__(self):
        return "%s %s " % ( self.__class__.__name__, self.sleep )

    def __call__(self): 

        i = 0 
        while i<self.maxiter or self.maxiter==0:
            self._insertfake(i)
            time.sleep(self.sleep.seconds)
            i += 1



if __name__ == '__main__':
    pass