快捷導(dǎo)航

詳解python校驗(yàn)SQL腳本命名規(guī)則

更新時(shí)間：2019年03月22日 11:09:18 作者：紅燜鯉魚

這篇文章主要介紹了python校驗(yàn)SQL腳本命名規(guī)則，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

需求背景

檢查腳本文件中SQL語句是否按規(guī)范編寫，規(guī)則如下：

創(chuàng)建表時(shí)，表名稱需以"t_"開頭且均為小寫
增加和創(chuàng)建列時(shí)，列名稱均為小寫字母和_組成
創(chuàng)建函數(shù)，函數(shù)名稱需以"f_"開頭
創(chuàng)建存儲(chǔ)過程，存儲(chǔ)過程名稱需以"p_"開頭
創(chuàng)建索引，索引名稱需以"idx_"開頭
創(chuàng)建視圖，視圖名稱需以"v_"開頭

需求分析

首先要從腳本文件中提取出來表名稱、列名稱、函數(shù)名稱、存儲(chǔ)過程名稱、索引名稱、視圖名稱
這里需要使用python 相應(yīng)的re和os模塊
分別創(chuàng)建相應(yīng)的規(guī)則，相應(yīng)的名稱依次分別進(jìn)行校驗(yàn)，返回信息和具體提示

代碼實(shí)現(xiàn)

以校驗(yàn)表名稱為例文本內(nèi)容如下；

-- mysql創(chuàng)建view、trigger、function、procedure、event時(shí)都會(huì)定義一個(gè)Definer
-- 更新中！?。。。。。。。。。。?！
CREATE TABLE `t_auth_group_permissions` (
create table t_business_apply_reback(
CREATE TABLE `t_exception_record` (
CREATE TABLE `t_lend_channel_insurance_company` (

alter table hb_product.admit_params add column originalid varchar(80);
alter table hb_product.admit_params_detail add column originalid varchar(80);

第一步——提取表名稱

def get_result(rec, str):
  """
  :param rec: 匹配規(guī)則
  :param str: 匹配的對(duì)象
  :return: 查詢的結(jié)果
  """
  result = re.findall(rec, str)
  return result

def get_table_names(str):
  """
  提取表名稱返回?cái)?shù)組
  :param str:
  :return:
  """
  table_name_rc1 = re.compile(r'CREATE TABLE [\S]*?(\w+)[\S].*\(', flags=re.I)
  result_table_name = get_result(table_name_rc1, str) + get_result(table_name_rc2, str)
  return result_table_name

flags=re.I 匹配時(shí)忽略大小寫，因?yàn)槲覀兊谋淼膭?chuàng)建語句CREATE TABLE或create table t_business_apply_reback
\w 匹配字母數(shù)字下劃線
\S 匹配任意空字符
[] 用來表示一組字符
() 匹配括號(hào)內(nèi)的表達(dá)式
* 匹配0個(gè)或多個(gè)
. 匹配任意字符除了換行符
？匹配0個(gè)或多個(gè)，非貪婪模式

運(yùn)行效果如下：

第二步——表名稱匹配

def is_match_t(strs):

  """
  表名稱規(guī)則校驗(yàn)：t_開頭小寫字母和下劃線
  :param strs:
  :return: 1通過，0不通過，msg返回信息
  """
  code, msg = 1, '驗(yàn)證通過'
  for i in strs:
    if re.match(r'^t_[a-z_]+$', i) is None:
      return 0, '創(chuàng)建表名稱:' + i + '不符合命名規(guī)范，請(qǐng)仔細(xì)檢查！'
  return code, msg