Initial commit

server/init_data.py

@@ -0,0 +1,82 @@
+from datetime import datetime
+from furl import furl
+
+from database.model.auth import User, Credentials
+from config import config
+from database.model.auth import Role
+from database.model.company import Company
+from elastic.apply_mappings import apply_mappings_to_host
+
+log = config.logger(__file__)
+
+
+class MissingElasticConfiguration(Exception):
+    """
+    Exception when cluster configuration is not found in config files
+    """
+
+    pass
+
+
+def init_es_data():
+    hosts_key = "hosts.elastic.events.hosts"
+    hosts_config = config.get(hosts_key, None)
+    if not hosts_config:
+        raise MissingElasticConfiguration(hosts_key)
+
+    for conf in hosts_config:
+        host = furl(scheme="http", host=conf["host"], port=conf["port"]).url
+        log.info(f"Applying mappings to host: {host}")
+        res = apply_mappings_to_host(host)
+        log.info(res)
+
+
+def _ensure_company():
+    company_id = config.get("apiserver.default_company")
+    company = Company.objects(id=company_id).only("id").first()
+    if company:
+        return company_id
+
+    company_name = "trains"
+    log.info(f"Creating company: {company_name}")
+    company = Company(id=company_id, name=company_name)
+    company.save()
+    return company_id
+
+
+def _ensure_user(user_data, company_id):
+    user = User.objects(
+        credentials__match=Credentials(key=user_data["key"], secret=user_data["secret"])
+    ).first()
+    if user:
+        return user.id
+
+    log.info(f"Creating user: {user_data['name']}")
+    user = User(
+        id=f"__{user_data['name']}__",
+        name=user_data["name"],
+        company=company_id,
+        role=user_data["role"],
+        email=user_data["email"],
+        created=datetime.utcnow(),
+        credentials=[Credentials(key=user_data["key"], secret=user_data["secret"])],
+    )
+
+    user.save()
+
+    return user.id
+
+
+def init_mongo_data():
+    company_id = _ensure_company()
+    users = [
+        {"name": "apiserver", "role": Role.system, "email": "apiserver@example.com"},
+        {"name": "webserver", "role": Role.system, "email": "webserver@example.com"},
+        {"name": "tests", "role": Role.user, "email": "tests@example.com"},
+    ]
+
+    for user in users:
+        credentials = config.get(f"secure.credentials.{user['name']}")
+        user["key"] = credentials.user_key
+        user["secret"] = credentials.user_secret
+        _ensure_user(user, company_id)