kaslo/synapse
1
0
Fork 0
Code Wiki Activity

STASH

Browse Source
This commit is contained in:
Olivier Wilkinson (reivilibre)
2022-03-01 13:15:20 +00:00
parent a0e699b301
commit 85774e17be
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
scripts-dev/workers_setup.py
View File

@@ -18,6 +18,9 @@ from pathlib import Path
from typing import List, Tuple, Iterable
from jinja2 import Environment, FileSystemLoader
from signedjson.key import generate_signing_key, write_signing_keys
from synapse.util.stringutils import random_string
DESIRED_WORKERS = (
("main", 1),
@@ -48,7 +51,7 @@ class Worker:
ip: str
def worker_num_to_ip(num: int):
def worker_num_to_ip(num: int) -> str:
return f"127.0.57.{num}"
@@ -70,12 +73,18 @@ def make_workers(workers: Iterable[Tuple[str, int]]) -> List[Worker]:
return result
def generate(worker_counts: Tuple[Tuple[str, int], ...], target_path: Path):
def generate(worker_counts: Tuple[Tuple[str, int], ...], target_path: Path) -> None:
if target_path.exists():
print("Target path already exists. Won't overwrite.")
return
target_path.mkdir()
# Generate a signing key
key_id = "a_" + random_string(4)
key = (generate_signing_key(key_id),)
with open(target_path.joinpath("signing.key"), "w") as fout:
write_signing_keys(fout, key)
env = Environment(loader=FileSystemLoader(dirname(__file__) + "/workers_setup"))
hs_template = env.get_template("homeserver.yaml.j2")
worker_template = env.get_template("worker.yaml.j2")
@@ -125,10 +134,10 @@ def generate(worker_counts: Tuple[Tuple[str, int], ...], target_path: Path):
fout.write(hs_config)
def main(target_path: Path):
def main(target_path: Path) -> None:
generate(DESIRED_WORKERS, target_path)
if __name__ == '__main__':
target_path = Path(sys.argv[1])
main(target_path)
main(target_path)