-
Notifications
You must be signed in to change notification settings - Fork 103
/
title_number.py
164 lines (156 loc) · 8.04 KB
/
title_number.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
from io import TextIOWrapper
import sys
import os
import time
import re
from unittest import result
headline = ['#','##','###','####','#####','######']
title_sign_list = []
"""用于判断标题产生环境"""
titles_added_number = []
"""保存嵌入了编号的标题，用于产生新编号"""
is_continue = 'n'
suf = '.'
"""给某一行添加编号"""
def add_number_for_line(line_which_is_title: str,title_sign: str):
global suf
global is_continue
global title_sign_list
global titles_added_number
title_sign_list.append(title_sign)
if len(title_sign_list) == 1:#如果line_which_is_title是第一个标题
titles_added_number.append(line_which_is_title.replace(title_sign + ' ',title_sign + ' 1. '))
return titles_added_number[0]
else:
#当标题级别为一级(需要标号为1.,2.,3.,...)
for title in titles_added_number[::-1]:
number = title.lstrip().split(' ')[1]
sign = title.lstrip().split(' ')[0]
if len(number) == 2 and number.count('.') == 1:#如果发现一级标题(序号为1.,2.,3.,...)
if len(title_sign) == len(sign):#如果line_which_is_title是一级标题（与发现的一级标题title级别相同）
titles_added_number.append(
line_which_is_title.replace(title_sign + ' ',title_sign + ' ' + str(int(number[:-1]) + 1) + '. '))
return titles_added_number[-1]
elif len(title_sign) < len(sign):#如果line_which_is_title是一级标题（比发现的第一个一级标题级别更高）
if is_continue != 'y':
print('Markdown文件中的：\n' + title.strip() + '或\n' + line_which_is_title + "\n不规范\n建议将Markdown文件中的标题分级、规范地写好后再继续")
is_continue = input('是否忽略此类警告并继续？（y/n）')
if is_continue.strip().lower() == 'y':
titles_added_number.append(line_which_is_title.replace(title_sign + ' ',title_sign + ' ' + str(int(number[:-1]) + 1) + '. '))
return titles_added_number[-1]
elif is_continue.strip().lower() == 'n':
os._exit(0)
else:
print('接收到y/n以外的输入，默认退出')
os._exit(0)
else:
break
#当标题级别不是一级(序号不是1.,2.,3.,...)
number = titles_added_number[-1].lstrip().split(' ')[1]
if len(number) == 2 and number.count('.') == 1:#如果line_which_is_title的上一级标题为一级标题(序号为1.,2.,3.,...)
titles_added_number.append(
line_which_is_title.replace(
title_sign + ' ',title_sign + ' ' + number + '1' + suf + ' '))
return titles_added_number[-1]
elif len(title_sign_list[-1]) > len(title_sign_list[-2]):#如果line_which_is_title的上一个标题比它更高
number = re.search('(.*\.\d+)[^\d]?$', number).group(1)
titles_added_number.append(
line_which_is_title.replace(
title_sign + ' ',title_sign + ' ' + number + '.1' + suf + ' '))
return titles_added_number[-1]
elif len(title_sign_list[-1]) == len(title_sign_list[-2]):#如果line_which_is_title与上一个标题等级别
number_suf = re.search('(.*\.)(\d+)[^\d]?$', number).group(2)
number_pre = re.search('(.*\.)(\d+)[^\d]?$', number).group(1)
titles_added_number.append(
line_which_is_title.replace(
title_sign + ' ',title_sign + ' ' + number_pre + str(int(number_suf) + 1) + suf + ' '))
return titles_added_number[-1]
elif len(title_sign_list[-1]) < len(title_sign_list[-2]):#如果line_which_is_title的上一个标题比它更低
for title in titles_added_number[::-1]:
number = title.lstrip().split(' ')[1]
sign = title.lstrip().split(' ')[0]
if len(number) == 2 and number.count('.') == 1:#如果先发现一级标题
if is_continue != 'y':
print('Markdown文件中的：\n' + title.strip() + '\n或\n' + line_which_is_title + "\n不规范\n建议将Markdown文件中的标题分级、规范地写好后再继续")
is_continue = input('是否忽略此类警告并继续？（y/n）')
if is_continue.strip() == 'y':
titles_added_number.append(line_which_is_title.replace(title_sign + ' ',title_sign + ' ' + str(int(number[:-1]) + 1) + '. '))
return titles_added_number[-1]
elif is_continue.strip() == 'n':
os._exit(0)
else:
print('接收到y/n以外的输入，默认退出')
os._exit(0)
if len(sign) == len(title_sign):#如果找到等级别标题
number_suf = re.search('(.*\.)(\d+)[^\d]?$', number).group(2)
number_pre = re.search('(.*\.)(\d+)[^\d]?$', number).group(1)
titles_added_number.append(
line_which_is_title.replace(
title_sign + ' ',title_sign + ' ' + number_pre + str(int(number_suf) + 1) + suf + ' '))
return titles_added_number[-1]
"""
判断某行是否需要添加编号（是否为标题行）
是一个标题时返回标题开头的井号，不是一个标题时或者返回空字符串
——不完备
"""
def get_title_sign(line: str):
result = re.compile('(^#+) [^\s]').search(line)
if result and result.group(1) in headline:
return result.group(1)
else:
return ""
"""给传入文件添加编号，返回行数据"""
def get_lines_with_number(f: TextIOWrapper):
lines = f.readlines()
f.close()
for i in range(len(lines)):
title_sign = get_title_sign(lines[i])
if title_sign in headline:
lines[i] = add_number_for_line(lines[i], title_sign)
return lines
"""生成添加了标题编号的文件"""
def create_markdown_file_with_number(f: TextIOWrapper, file_name: str):
lines_with_number = get_lines_with_number(f)
# 根据原文件名生成标题添加了序号的文件的文件名
markdown_file_with_number = os.getcwd() + '\\' + file_name[::-1].split('.',1)[1][::-1] + '_withNum.md'
if not os.path.exists(markdown_file_with_number):
with open(markdown_file_with_number, 'w+',encoding='utf-8') as f:
for line in lines_with_number:
f.write(line)
print('文件已生成')
else:
if input('文件名重复，是否覆盖 ' + markdown_file_with_number + ' (y/n)').lower() == 'y':
with open(markdown_file_with_number, 'w+',encoding='utf-8') as f:
for line in lines_with_number:
f.write(line)
print('文件已生成')
else:
print('程序退出')
file_name = ''
#如果未传入文件
if len(sys.argv) < 2:
path = os.getcwd()
file_and_dir = os.listdir(path)
md_file = []
for item in file_and_dir:
if item.split('.')[-1].lower() in ['md','mdown','markdown'] and os.path.isfile(item):
md_file.append(item)
if len(md_file) != 0:
print('当前目录下的Markdown文件：')
for file in md_file:
print(file)
file_name = input('请输入文件名(含后缀)或回车选择第一个文件\n')
if not file_name:
file_name = md_file[0]
else:
print('该目录下无Markdown文件，即将退出...')
time.sleep(2)
os._exit(0)
else:
file_name = sys.argv[1]
if os.path.exists(file_name) and os.path.isfile(file_name):
with open(file_name,'r',encoding='utf-8') as f:
create_markdown_file_with_number(f,file_name)
else:
msg = "未找到文件"
print(msg)