Implement restricted sources

This commit is contained in:
Mark Veidemanis 2022-09-18 13:01:19 +01:00
parent dab5e81715
commit 3ed382ec13
Signed by: m
GPG Key ID: 5ACFCEED46C0904F
1 changed files with 8 additions and 1 deletions

9
db.py
View File

@ -32,6 +32,12 @@ TYPES_MAIN = [
"topic",
"update",
]
MAIN_SRC_MAP = {
"dis": "main",
"irc": "restricted",
"4ch": "main",
}
TYPES_META = ["who"]
TYPES_INT = ["conn", "highlight", "znc", "query", "self"]
KEYPREFIX = "queue."
@ -44,7 +50,8 @@ async def store_kafka_batch(data):
batch = producer.create_batch()
for msg in data:
if msg["type"] in TYPES_MAIN:
index = "main"
#index = "main"
index = MAIN_SRC_MAP[msg["src"]]
# schema = mc_s.schema_main
elif msg["type"] in TYPES_META:
index = "meta"