Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

oracle规避对象名大小写敏感 #1438

Merged
merged 14 commits into from
Apr 26, 2022
Merged

oracle规避对象名大小写敏感 #1438

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
14 commits
Select commit Hold shift + click to select a range
df217cd
update sql/engines/oracle.py
peng19832 Nov 4, 2021
bfbaa1d
Merge branch 'master' of https://github.com/hhyo/Archery
peng19832 Nov 4, 2021
fed6532
update sql/engines/oracle.py
peng19832 Nov 4, 2021
59f3b0e
Update oracle.py
peng19832 Nov 10, 2021
a8182b1
Update sql_review.py
peng19832 Nov 30, 2021
73d6928
Merge branch 'master' into master
peng19832 Dec 28, 2021
d9f33bc
Merge branch 'hhyo:master' into master
peng19832 Jan 7, 2022
6e22ad1
规避双引号
peng19832 Mar 31, 2022
8345e43
Merge branch 'hhyo:master' into master
peng19832 Mar 31, 2022
65a1e1f
Merge branch 'master' of https://github.com/peng19832/Archery
peng19832 Mar 31, 2022
671aaa0
规避sql中对象名使用双引号问题
peng19832 Apr 19, 2022
4bf674d
get_sql_first_object_name方法忽略对象名大小写
peng19832 Apr 19, 2022
ba3c718
Merge branch 'hhyo:master' into master
peng19832 Apr 19, 2022
6c80b08
Merge branch 'master' into master
peng19832 Apr 26, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
122 changes: 89 additions & 33 deletions sql/engines/oracle.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -340,9 +340,25 @@ def object_name_check(self, db_name=None, object_name=''):
if '.' in object_name:
schema_name = object_name.split('.')[0]
object_name = object_name.split('.')[1]
sql = f"""SELECT object_name FROM all_objects WHERE OWNER = upper('{schema_name}') and OBJECT_NAME = upper('{object_name}')"""
if '"' in schema_name:
schema_name = schema_name.replace( '"','' )
if '"' in object_name:
object_name = object_name.replace( '"','' )
else:
object_name = object_name.upper()
else:
schema_name = schema_name.upper()
if '"' in object_name:
object_name = object_name.replace( '"','' )
else:
object_name = object_name.upper()
nick2wang marked this conversation as resolved.
Show resolved Hide resolved
else:
sql = f"""SELECT object_name FROM all_objects WHERE OWNER = upper('{db_name}') and OBJECT_NAME = upper('{object_name}')"""
schema_name = db_name
if '"' in object_name:
object_name = object_name.replace( '"','' )
else:
object_name = object_name.upper()
sql = f""" SELECT object_name FROM all_objects WHERE OWNER = '{schema_name}' and OBJECT_NAME = '{object_name}' """
result = self.query(db_name=db_name, sql=sql, close_conn=False)
if result.affected_rows > 0:
return True
Expand All @@ -353,26 +369,26 @@ def object_name_check(self, db_name=None, object_name=''):
def get_sql_first_object_name(sql=''):
"""获取sql文本中的object_name"""
object_name = ''
if re.match(r"^create\s+table\s", sql):
object_name = re.match(r"^create\s+table\s(.+?)(\s|\()", sql, re.M).group(1)
elif re.match(r"^create\s+index\s", sql):
object_name = re.match(r"^create\s+index\s(.+?)\s", sql, re.M).group(1)
elif re.match(r"^create\s+unique\s+index\s", sql):
object_name = re.match(r"^create\s+unique\s+index\s(.+?)\s", sql, re.M).group(1)
elif re.match(r"^create\s+sequence\s", sql):
object_name = re.match(r"^create\s+sequence\s(.+?)(\s|$)", sql, re.M).group(1)
elif re.match(r"^alter\s+table\s", sql):
object_name = re.match(r"^alter\s+table\s(.+?)\s", sql, re.M).group(1)
elif re.match(r"^create\s+function\s", sql):
object_name = re.match(r"^create\s+function\s(.+?)(\s|\()", sql, re.M).group(1)
elif re.match(r"^create\s+view\s", sql):
object_name = re.match(r"^create\s+view\s(.+?)\s", sql, re.M).group(1)
elif re.match(r"^create\s+procedure\s", sql):
object_name = re.match(r"^create\s+procedure\s(.+?)\s", sql, re.M).group(1)
elif re.match(r"^create\s+package\s+body", sql):
object_name = re.match(r"^create\s+package\s+body\s(.+?)\s", sql, re.M).group(1)
elif re.match(r"^create\s+package\s", sql):
object_name = re.match(r"^create\s+package\s(.+?)\s", sql, re.M).group(1)
if re.match(r"^create\s+table\s", sql, re.M|re.IGNORECASE):
object_name = re.match(r"^create\s+table\s(.+?)(\s|\()", sql, re.M|re.IGNORECASE).group(1)
elif re.match(r"^create\s+index\s", sql, re.M|re.IGNORECASE):
object_name = re.match(r"^create\s+index\s(.+?)\s", sql, re.M|re.IGNORECASE).group(1)
elif re.match(r"^create\s+unique\s+index\s", sql, re.M|re.IGNORECASE):
object_name = re.match(r"^create\s+unique\s+index\s(.+?)\s", sql, re.M|re.IGNORECASE).group(1)
elif re.match(r"^create\s+sequence\s", sql, re.M|re.IGNORECASE):
object_name = re.match(r"^create\s+sequence\s(.+?)(\s|$)", sql, re.M|re.IGNORECASE).group(1)
elif re.match(r"^alter\s+table\s", sql, re.M|re.IGNORECASE):
object_name = re.match(r"^alter\s+table\s(.+?)\s", sql, re.M|re.IGNORECASE).group(1)
elif re.match(r"^create\s+function\s", sql, re.M|re.IGNORECASE):
object_name = re.match(r"^create\s+function\s(.+?)(\s|\()", sql, re.M|re.IGNORECASE).group(1)
elif re.match(r"^create\s+view\s", sql, re.M|re.IGNORECASE):
object_name = re.match(r"^create\s+view\s(.+?)\s", sql, re.M|re.IGNORECASE).group(1)
elif re.match(r"^create\s+procedure\s", sql, re.M|re.IGNORECASE):
object_name = re.match(r"^create\s+procedure\s(.+?)\s", sql, re.M|re.IGNORECASE).group(1)
elif re.match(r"^create\s+package\s+body", sql, re.M|re.IGNORECASE):
object_name = re.match(r"^create\s+package\s+body\s(.+?)\s", sql, re.M|re.IGNORECASE).group(1)
elif re.match(r"^create\s+package\s", sql, re.M|re.IGNORECASE):
object_name = re.match(r"^create\s+package\s(.+?)\s", sql, re.M|re.IGNORECASE).group(1)
else:
return object_name.strip()
return object_name.strip()
Expand Down Expand Up @@ -451,7 +467,7 @@ def explain_check(self, db_name=None, sql='', close_conn=False):
conn = self.get_connection()
cursor = conn.cursor()
if db_name:
cursor.execute(f"ALTER SESSION SET CURRENT_SCHEMA = {db_name}")
cursor.execute(f" ALTER SESSION SET CURRENT_SCHEMA = \"{db_name}\" ")
nick2wang marked this conversation as resolved.
Show resolved Hide resolved
if re.match(r"^explain", sql, re.I):
sql = sql
else:
Expand Down Expand Up @@ -515,7 +531,7 @@ def query(self, db_name=None, sql='', limit_num=0, close_conn=True, **kwargs):
conn = self.get_connection()
cursor = conn.cursor()
if db_name:
cursor.execute(f"ALTER SESSION SET CURRENT_SCHEMA = {db_name}")
cursor.execute(f" ALTER SESSION SET CURRENT_SCHEMA = \"{db_name}\" ")
sql = sql.rstrip(';')
# 支持oralce查询SQL执行计划语句
if re.match(r"^explain", sql, re.I):
Expand Down Expand Up @@ -575,6 +591,7 @@ def execute_check(self, db_name=None, sql='', close_conn=True):
sqlitemList = get_full_sqlitem_list(sql, db_name)
for sqlitem in sqlitemList:
sql_lower = sqlitem.statement.lower().rstrip(';')
sql_nolower = sqlitem.statement.rstrip(';')
# 禁用语句
if re.match(r"^select|^with|^explain", sql_lower):
check_result.is_critical = True
Expand Down Expand Up @@ -642,12 +659,25 @@ def execute_check(self, db_name=None, sql='', close_conn=True):
else:
# 对create table\create index\create unique index语法做对象存在性检测
if re.match(r"^create\s+table|^create\s+index|^create\s+unique\s+index", sql_lower):
object_name = self.get_sql_first_object_name(sql=sql_lower)
object_name = self.get_sql_first_object_name(sql=sql_nolower)
nick2wang marked this conversation as resolved.
Show resolved Hide resolved
# 保存create对象对后续SQL做存在性判断
if '.' in object_name:
object_name = object_name
schema_name = object_name.split('.')[0]
object_name = object_name.split('.')[1]
if '"' in schema_name:
schema_name = schema_name
if '"' not in object_name:
object_name = object_name.upper()
else:
schema_name = schema_name.upper()
if '"' not in object_name:
object_name = object_name.upper()
else:
object_name = f"""{db_name}.{object_name}"""
schema_name = ( '"' + db_name + '"' )
if '"' not in object_name:
object_name = object_name.upper()

object_name = f"""{schema_name}.{object_name}"""
if self.object_name_check(db_name=db_name,
object_name=object_name) or object_name in object_name_list:
check_result.is_critical = True
Expand Down Expand Up @@ -706,11 +736,24 @@ def execute_check(self, db_name=None, sql='', close_conn=True):
else:
# 对alter table做对象存在性检查
if re.match(r"^alter\s+table\s", sql_lower):
object_name = self.get_sql_first_object_name(sql=sql_lower)
object_name = self.get_sql_first_object_name(sql=sql_nolower)
if '.' in object_name:
object_name = object_name
schema_name = object_name.split('.')[0]
object_name = object_name.split('.')[1]
if '"' in schema_name:
schema_name = schema_name
if '"' not in object_name:
object_name = object_name.upper()
else:
schema_name = schema_name.upper()
if '"' not in object_name:
object_name = object_name.upper()
else:
object_name = f"""{db_name}.{object_name}"""
schema_name = ( '"' + db_name + '"' )
if '"' not in object_name:
object_name = object_name.upper()

object_name = f"""{schema_name}.{object_name}"""
if not self.object_name_check(db_name=db_name,
object_name=object_name) and object_name not in object_name_list:
check_result.is_critical = True
Expand All @@ -731,11 +774,24 @@ def execute_check(self, db_name=None, sql='', close_conn=True):
execute_time=0, )
# 对create做对象存在性检查
elif re.match(r"^create", sql_lower):
object_name = self.get_sql_first_object_name(sql=sql_lower)
object_name = self.get_sql_first_object_name(sql=sql_nolower)
if '.' in object_name:
object_name = object_name
schema_name = object_name.split('.')[0]
object_name = object_name.split('.')[1]
if '"' in schema_name:
schema_name = schema_name
if '"' not in object_name:
object_name = object_name.upper()
else:
schema_name = schema_name.upper()
if '"' not in object_name:
object_name = object_name.upper()
else:
object_name = f"""{db_name}.{object_name}"""
schema_name = ( '"' + db_name + '"' )
if '"' not in object_name:
object_name = object_name.upper()

object_name = f"""{schema_name}.{object_name}"""
if self.object_name_check(db_name=db_name,
object_name=object_name) or object_name in object_name_list:
check_result.is_critical = True
Expand Down
2 changes: 1 addition & 1 deletion sql/utils/sql_utils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -293,7 +293,7 @@ def get_exec_sqlitem_list(reviewResult, db_name):
:return:
"""
list = []
list.append(SqlItem(statement=f"ALTER SESSION SET CURRENT_SCHEMA = {db_name}"))
list.append(SqlItem(statement=f" ALTER SESSION SET CURRENT_SCHEMA = \"{db_name}\" "))

for item in reviewResult:
list.append(SqlItem(statement=item['sql'],
Expand Down