to_ck/main.py
2021-04-21 18:02:03 +08:00

30 lines
616 B
Python

from ck.ck import CK
from common import *
from settings import settings
import traceback
ck_client = CK()
def run():
for topic, msg in consumer():
# print(msg)
try:
db = settings.APPID_TO_CKDB.get(msg['app_id'])
if 'user' in msg['type']:
table = 'user'
elif 'track' in msg['type']:
table = 'event'
else:
continue
del msg['type']
ck_client.send(db, table, msg)
except Exception as e:
print(traceback.print_exc())
if __name__ == '__main__':
run()