Implement restricted sources
This commit is contained in:
parent
dab5e81715
commit
3ed382ec13
9
db.py
9
db.py
|
@ -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"
|
||||
|
|
Loading…
Reference in New Issue