From 7045b1779a37c7d8cd1d16da1e53bd3f85f86a0e Mon Sep 17 00:00:00 2001
From: tabsdavis <sdavis@tableau.com>
Date: Wed, 21 Jun 2023 11:49:45 -0700
Subject: [PATCH] Added the tdvtconnection property when writing/updating the
 ds tds file.

---
 tdvt/tdvt/setup_env.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tdvt/tdvt/setup_env.py b/tdvt/tdvt/setup_env.py
index 78137d72..abfb2cbe 100644
--- a/tdvt/tdvt/setup_env.py
+++ b/tdvt/tdvt/setup_env.py
@@ -222,7 +222,7 @@ def mangle_tds(file_path, connection_password_name):
     try:
         r1 = re.compile('(^\s*<named-connection .*? name=\').*?(\'>)')
         r2 = re.compile('(^\s*<.*relation connection=\').*?(\' .*>)')
-        r3 = re.compile('(^\s*<connection .*?)(\s*/>)')
+        r3 = re.compile('(^\s*<connection .*?)''(^\s*<tdvtconnection .*?)(\s*/>)')
 
         f = open(file_path, 'r')
         new_tds = ''