from settings import settings
from v2 import *

db_client = CK(**settings.CK_CONFIG)
sketch = Sketch(db_client)
handler_event = HandlerEvent(db_client)
handler_user = HandlerUser(db_client)
transmitter = Transmitter(db_client, sketch)


def run():
    transmitter.add_source(handler_event, 1000, 60)
    transmitter.add_source(handler_user, 100, 60)
    for topic, msg in consumer():
        # print(msg)
        type_ = msg['#type']
        db = settings.APPID_TO_CKDB.get(msg['#app_id'])
        if 'user' in type_:
            # continue
            obj = getattr(handler_user, type_)
        elif 'track' in type_:
            # continue
            obj = getattr(handler_event, type_)
        else:
            continue

        del msg['#type']
        obj(db, msg)

        transmitter.run()


if __name__ == '__main__':
    run()