Mekhanik evgenii/fix 525 3

helpers/__init__.py

@@ -12,6 +12,7 @@
  "prepare",
  "util",
  "networker",
+ "custom_error_page",
 ]
 
 # vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4

helpers/custom_error_page.py

@@ -0,0 +1,35 @@
+"""
+Custom error page generator.
+
+Tempesta administrator can specify files with page bodies for error responses
+generated by Tempesta FW.
+"""
+from helpers import tf_cfg
+
+__author__ = "Tempesta Technologies, Inc."
+__copyright__ = "Copyright (C) 2023 Tempesta Technologies, Inc."
+__license__ = "GPL2"
+
+
+class CustomErrorPageGenerator(object):
+ def __init__(self, data, f_path=None, default=False):
+ workdir = tf_cfg.cfg.get("General", "workdir")
+ self.f_path = f_path if f_path else workdir + "/tempesta.html"
+ self.data = data
+ if default:
+ self.generate()
+
+ @staticmethod
+ def __write(path, data):
+ with open(path, "wt") as fdesc:
+ fdesc.write(data)
+
+ def generate(self):
+ self.__write(self.f_path, data)
+
+ def __str__(self):
+ assert self.data, "Stringify null custom error page object"
+ return self.data
+
+ def get_file_path(self):
+ return self.f_path