python django models 正则方式取类名及属性名

python django models 正则方式取类名及属性名

一、背景

在python的django，models文件中，使用 正则方式取类名及属性名。

二、解决方案

• 读取文件
• 按行分析
• 取类名，并记录成功的行号
• 取属性名，放入行号的列中

三、代码

from django.shortcuts import render
import re

def models_to_plum(request):
    file = 'models.py'
    out_dict = {}
    exclude_attr_list = ['verbose_name', 'constraints',]
    class_list = []
    int_行号=0
    with open(file=file,mode='r',encoding='utf-8') as f :
        str_lines = f.readlines()
        for str_line in str_lines[:]:
            pattern = re.compile('(?<=^class\s)[\d\w\D]+(?=\()')  # 选取类名
            m_class = pattern.search(str_line)
            if m_class:
                m_class_name = m_class.group(0)
                int_行号 += 1 # 找到类名后，才加1，否则不加
                out_dict[int_行号]=['class ',m_class_name,'{\n','}\n']
            else:
                patter_attr = re.compile('^\s[\d\D]+(?=\s=)') # 获取属性名，但前面有空格要处理
                m_attr = patter_attr.search(str_line)
                if m_attr:
                    m_attr_name = m_attr.group(0)
                    print(m_attr_name)
                    out_dict[int_行号].insert(-1,m_attr_name.strip())  if m_attr_name.strip() not in exclude_attr_list else None # 删除两边的空格，因正则表达式没写好，对属性处理不好，留了空格
    result_str = out_dict
    return render(request,'models.html',locals())