| 166 | return sql |
| 167 | |
| 168 | def processSQL(self, sql: Optional[str], dbID: Optional[str] = None): |
| 169 | if str(sql).__contains__('\n'): |
| 170 | sql = str(sql).replace('\n', ' ') |
| 171 | self.orm.save(sql) |
| 172 | res = "" |
| 173 | |
| 174 | # 匹配SQL语句 |
| 175 | pattern = r'\b(SELECT|INSERT|UPDATE|DELETE|CREATE|DROP|ALTER)\b.*?(?=;|$)' |
| 176 | sql_list = re.findall(pattern, sql, re.IGNORECASE | re.DOTALL) |
| 177 | # 输出匹配结果 |
| 178 | for item in sql_list: |
| 179 | res = res + item |
| 180 | res = res + " " |
| 181 | return sql |
| 182 | |
| 183 | def operate_sql(self, sql: Optional[str], dbID: Optional[str] = None): |
| 184 | tag, res = self.sqlHelper.operate(sql=sql, dbID=dbID) |