generator: Allow to overwrite the locale

Signed-off-by: Michael Tremer <michael.tremer@ipfire.org>

diff --git a/src/suricata-report-generator.in b/src/suricata-report-generator.in
index 18c6012135c1f122e27e7cd45f398f2845485422..25a93990b697a4e5320c80239dc4bd7fe5183340 100644 (file)
--- a/src/suricata-report-generator.in
+++ b/src/suricata-report-generator.in
@@ -25,6 +25,7 @@ import collections
 import datetime
 import email.message
 import email.utils
+import locale
 import logging
 import reportlab
 import reportlab.lib.styles
@@ -488,6 +489,9 @@ def main():
        parser.add_argument("--database", help="Database",
                default="@suricatalogdir@/reporter.db")
 
+       # Locale
+       parser.add_argument("--locale", help=_("Locale"))
+
        # Require some output parameters
        group = parser.add_mutually_exclusive_group(required=True)
        group.add_argument("--output", "-o", help=_("Output Path"))
@@ -528,6 +532,10 @@ def main():
 
        setup_logging(loglevel=loglevel)
 
+       # Change locale
+       if args.locale:
+               locale.setlocale(locale.LC_ALL, locale.normalize(args.locale))
+
        # Create the report
        generator = ReportGenerator(args.database)