Implement restricted sources

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"