Merge pull request #332 from thomas-mangin/T2230

template: T2230: use render to generate templates

1 files changed, 5 insertions, 16 deletions

diff --git a/src/conf_mode/vpn-pptp.py b/src/conf_mode/vpn-pptp.py
index 45b2c4b40..15b80f984 100755
--- a/src/conf_mode/vpn-pptp.py
+++ b/src/conf_mode/vpn-pptp.py
@@ -17,15 +17,15 @@
 import os
 import re
 
-from jinja2 import FileSystemLoader, Environment
 from socket import socket, AF_INET, SOCK_STREAM
 from sys import exit
 from time import sleep
 
 from vyos.config import Config
-from vyos.defaults import directories as vyos_data_dir
 from vyos import ConfigError
 from vyos.util import run
+from vyos.template import render
+
 
 pidfile = r'/var/run/accel_pptp.pid'
 pptp_cnf_dir = r'/etc/accel-ppp/pptp'
@@ -206,11 +206,6 @@ def generate(c):
     if c == None:
         return None
 
-    # Prepare Jinja2 template loader from files
-    tmpl_path = os.path.join(vyos_data_dir['data'], 'templates', 'pptp')
-    fs_loader = FileSystemLoader(tmpl_path)
-    env = Environment(loader=fs_loader, trim_blocks=True)
-
     # accel-cmd reload doesn't work so any change results in a restart of the daemon
     try:
         if os.cpu_count() == 1:
@@ -223,19 +218,13 @@ def generate(c):
         else:
             c['thread_cnt'] = int(os.cpu_count()/2)
 
-    tmpl = env.get_template('pptp.config.tmpl')
-    config_text = tmpl.render(c)
-    with open(pptp_conf, 'w') as f:
-        f.write(config_text)
+    render(pptp_conf, 'pptp/pptp.config.tmpl', c, trim_blocks=True)
 
     if c['authentication']['local-users']:
-        tmpl = env.get_template('chap-secrets.tmpl')
-        chap_secrets_txt = tmpl.render(c)
         old_umask = os.umask(0o077)
-        with open(chap_secrets, 'w') as f:
-            f.write(chap_secrets_txt)
+        render(chap_secrets, 'pptp/chap-secrets.tmpl', c, trim_blocks=True)
         os.umask(old_umask)
-
+    # return c ??
     return c