Update queryanalyse.py

修改bug 数据库为空或者表名为空时候处理
baopbird2005 · Aug 13, 2018 · 4cdcedb · 4cdcedb
1 parent 069a0ab
commit 4cdcedb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/queryanalyse.py b/queryanalyse.py
@@ -133,7 +133,7 @@ def rowrecord(self):
                     # print bline.replace("\n", "")
                     out_sql = out_sql + bline
 
-                if isEnd and ((self.search_td != '' and db_name == self.search_td) or (self.search_tb != '' and tb_name == self.search_tb)):
+                if isEnd and (self.search_tb == '' or self.search_td == '' or (self.search_td != '' and db_name == self.search_td) or (self.search_tb != '' and tb_name == self.search_tb)):
                     out_file.write(out_sql)
                     self.total_num = self.total_num + 1
                     # print out_sql
@@ -170,4 +170,4 @@ def main():
     print '\n******************************************************************************'
 
 if __name__ == "__main__":
-    main()
+    main()