From ca22621d15aa99c9ee7918109579350f583e943b Mon Sep 17 00:00:00 2001
From: jerrylizilong <lizion138@163.com>
Date: Fri, 4 Jan 2019 18:31:27 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A4=84=E7=90=86=20\=20=E7=9A=84=E4=BF=9D?=
 =?UTF-8?q?=E5=AD=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/db/test_batch_manage.py | 5 +++++
 app/db/test_case_manage.py  | 2 +-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/app/db/test_batch_manage.py b/app/db/test_batch_manage.py
index 3abbc90..5e4f4bf 100644
--- a/app/db/test_batch_manage.py
+++ b/app/db/test_batch_manage.py
@@ -14,6 +14,7 @@ class test_batch_manage(object):
         steps.replace('"','""')
         import re
         steps = re.sub('"', '""', steps)
+        steps = str(steps).replace('\\', '\\\\')
         sql = string.Template('insert into test_batch (test_suite_id, test_case_id, name,steps,browser_type) values ("$test_suite_id","$test_case_id","$name","$steps","$browser_type");')
         sql = sql.substitute(test_suite_id = test_suite_id, test_case_id = test_case_id,steps=steps, name = name,browser_type=browser_type)
         useDB.useDB().insert(sql)
@@ -23,6 +24,7 @@ class test_batch_manage(object):
         steps.replace('"', '""')
         import re
         steps = re.sub('"', '""', steps)
+        steps = str(steps).replace('\\', '\\\\')
         sql = string.Template(
             'insert into test_batch (test_suite_id, test_case_id, name,steps,ip) values ("$test_suite_id","$test_case_id","$name","$steps","$ip");')
         sql = sql.substitute(test_suite_id=test_suite_id, test_case_id=test_case_id, steps=steps, name=name,ip=ip)
@@ -90,6 +92,8 @@ class test_batch_manage(object):
     def update_test_batch(self,id,fieldlist,valuelist):
         update_value = ''
         for i in range(len(fieldlist)):
+            if fieldlist[i]=='steps':
+                valuelist[i]=valuelist[i].replace('\\','\\\\')
             update_value = update_value+' %s = "%s"' %(str(fieldlist[i]),str(valuelist[i]))
             if i < len(fieldlist)-1 :
                 update_value = update_value+','
@@ -110,6 +114,7 @@ class test_batch_manage(object):
                     steps.replace('"', '""')
                     import re
                     steps = re.sub('"', '""', steps)
+                    steps = str(steps).replace('\\', '\\\\')
                     useDB.useDB().insert('update test_batch set status=0, steps = "%s" where id = %s ;' % (steps,case[0]) )
                     log.log().logger.info('update test_batch set status=0, steps = "%s" where id = %s ;' % (steps,case[0]) )
                     # log.log().logger.info(steps[0])
diff --git a/app/db/test_case_manage.py b/app/db/test_case_manage.py
index 9ab16c5..0f74a0c 100644
--- a/app/db/test_case_manage.py
+++ b/app/db/test_case_manage.py
@@ -18,7 +18,7 @@ class test_case_manage:
         if len(result):
             result=result[0]
             sql = string.Template('insert into test_case (module,name,steps,description,isPublicFunction) values ("$module","$name","$steps","$description",$isPublic);')
-            sql = sql.substitute(name = result['name'], module = result['module'], steps = result['steps'],description=result['description'], isPublic=result['isPublic'])
+            sql = sql.substitute(name = result['name'], module = result['module'], steps = result['steps'].replace('\\', '\\\\'),description=result['description'], isPublic=result['isPublic'])
             useDB.useDB().insert(sql)
             result = 1
         else: