Add support for diff and deploy for GTS

This commit is contained in:
moanos [he/him] 2023-05-04 10:47:06 +02:00
parent 27168a2a6e
commit 2180f28c78

View File

@ -11,12 +11,7 @@ class Instance:
self.reject_reports = False
self.id = None
"""Remote blocks and local blocks are parsed differently"""
try:
instance_dict["id"]
self.parse_remote_block(instance_dict)
except KeyError:
self.parse_local_block(instance_dict)
self.parse_block(instance_dict)
def __str__(self):
return f"{self.domain}: {self.severity}"
@ -36,10 +31,10 @@ class Instance:
exportable[key] = getattr(self, key)
return exportable
def parse_remote_block(self, instance_dict):
def parse_block(self, instance_dict):
# this specifies possible properties and default values if not found on the remote source. If a default is None
# the value is required and the parse will fail
properties_and_defaults = [("domain", None), ("id", None), ("severity", "suspend"), ("public_comment", ""),
properties_and_defaults = [("domain", None), ("severity", "suspend"), ("public_comment", ""),
("private_comment", ""), ("obfuscate", False), ("reject_media", True),
("reject_reports", True)]
for key, default in properties_and_defaults:
@ -49,29 +44,7 @@ class Instance:
if default is not None:
setattr(self, key, default)
else:
raise KeyError(f"The key {key} was not in the remote servers response.")
def parse_local_block(self, instance_dict):
try:
self.name = instance_dict["name"]
except KeyError:
pass
self.domain = instance_dict["domain"]
self.severity = instance_dict["severity"]
self.public_comment = instance_dict["public_comment"]
self.private_comment = instance_dict["private_comment"]
try:
self.obfuscate = instance_dict["obfuscate"]
except KeyError:
pass
try:
self.reject_media = instance_dict["reject_media"]
except KeyError:
pass
try:
self.reject_reports = instance_dict["reject_reports"]
except KeyError:
pass
raise KeyError(f"The key {key} was not in the instance_dict response.")
def apply(self, server, token, block_id=None):
"""Applies instance block on the remote server"""