Add parameters for tls and knot

main.py

@@ -11,7 +11,6 @@ import sys
 
 from re2oapi import Re2oAPIClient
 
-import knot
 
 
 path = os.path.dirname(os.path.abspath(__file__))
@@ -21,6 +20,11 @@ config.read(path+'/config.ini')
 api_hostname = config.get('Re2o', 'hostname')
 api_password = config.get('Re2o', 'password')
 api_username = config.get('Re2o', 'username')
+use_tls = config.getboolean('Re2o', 'use_tls')
+use_knot = config.getboolean('Re2o', 'use_knot')
+
+if use_knot:
+    import knot
 
 template_soa = (
     "$ORIGIN {zone}.\n"
@@ -217,7 +221,7 @@ def write_dns_file(zone, verbose=False):
         for x in zone['dname_records']
     )
 
-    if zone['name'][1:] == "crans.org":
+    if use_knot and zone['name'][1:] == "crans.org":
         ds_records = ""
         for extension in filter(lambda zone: zone.endswith('.crans.org'), zone_names):
             for ds in knot.get_ds(extension, verbose):
@@ -275,7 +279,6 @@ def write_dns_reverse_file(api_client):
      # because some iptype may share the same prefix
      # in which case we must append to the file zone already created
      zone_v6 = []
-
      for zone in api_client.list("dns/reverse-zones"):
         # We start by defining the soa, ns, mx which are comon to v4/v6
         now = datetime.datetime.now(datetime.timezone.utc)
@@ -413,7 +416,7 @@ def write_dns_reverse_file(api_client):
                     f.write(zone_file_content)
                 zone_v6.append(zone6_name)
 
-api_client = Re2oAPIClient(api_hostname, api_username, api_password, use_tls=False)
+api_client = Re2oAPIClient(api_hostname, api_username, api_password, use_tls=use_tls)
 
 client_hostname = socket.gethostname().split('.', 1)[0]
 
@@ -422,7 +425,8 @@ if __name__ == '__main__':
     parser.add_argument('-f', '--force', '--forced', help="Forcer la régénaration des fichiers de zone.", action='store_true')
     parser.add_argument('-k', '--keep', help="Ne pas changer le statut du service.", action='store_true')
     parser.add_argument('-p', '--processes', help="Regénérer en utilisant n processus en parallèle (par défaut ne pas parallèliser).", metavar='n', nargs=1, type=int, default=[0])
-    parser.add_argument('-n', '--no-reload', help="Ne pas recharger les zones dans knot.", action='store_true')
+    if use_knot:
+        parser.add_argument('-n', '--no-reload', help="Ne pas recharger les zones dans knot.", action='store_true')
     parser.add_argument('-v', '--verbose', help="Afficher des informations de debug.", action='store_true')
     args = parser.parse_args()
 
@@ -452,7 +456,7 @@ if __name__ == '__main__':
             with open(path + '/serial.json', 'w') as serial_json:
                 json.dump(serial + 1, serial_json)
 
-    if not args.no_reload:
+    if use_knot and not args.no_reload:
         error = os.system('/usr/sbin/knotc zone-reload >/dev/null 2>&1')
         if error:
             # reload again and display the error message