pinecrypt/pinecrypt-gateway-firewall
2
1
Fork 0
Code Issues 3 Pull Requests Projects Activity

Fix REPLICAS handling

Browse Source
This commit is contained in:
Lauri Võsandi 2021-06-18 14:51:15 +03:00
parent 3f9ef3efa4
commit 5cb6c48996
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
firewall.py
View File

@ -11,6 +11,13 @@ DISABLE_MASQUERADE = os.getenv("DISABLE_MASQUERADE")
REPLICAS = [j for j in os.getenv("REPLICAS", "").split(",") if j]
MONGO_URI = os.getenv("MONGO_URI")
if REPLICAS:
if MONGO_URI:
raise ValueError("Simultanously specifying MONGO_URI and REPLICAS doesn't make sense")
MONGO_URI = "mongodb://%s/default?replicaSet=rs0" % (",".join(["%s:27017" % j for j in REPLICAS]))
elif not MONGO_URI:
raise ValueError("MONGO_URI not specified")
def generate_firewall_rules(disabled=False):
default_policy = "REJECT" if DEBUG else "DROP"