Python文本整理功能处理小说目录结构

整理文本小说的目录

有时下载的网络文本小说,目录比较混乱,导入到阅读软件时,从目录浏览时很不方便。于是用Python整理下,让目录格式尽量统一,方便阅读。
源代码如下:

# -*- encoding: utf-8 -*-
import os

def is_content(a):
    tmpini=len(a)
    con_a='第'
    con_b='一二三四五六七八九十零百千1234567890'
    con_c='章节卷'
    con_end=False
    if a[0]==con_a:
        i=1
    else:
        i=0
    while i<=tmpini and con_end==False:
        if con_b.find(a[i])>=0:
            i+=1
        else:
            return False
        if con_c.find(a[i])>=0:
            con_end=True
            return True 

def no_empty(a):
    tmpstr=a.strip()
    if len(tmpstr)>0:
        return True
    else:
        return False

def title_isdi(a):
    if a[0]!='第':
        return '第'+a
    else:
        return a

f=open('tmp1.txt','r', encoding='utf-8')
f1=open('tmp2.txt','w')
lines=f.readlines()
f.close()
tmpstr1="a"
for i in lines:
    if no_empty(i):
        tmpstr=i.replace(' ','')
        if is_content(tmpstr):
            tmpstr=title_isdi(tmpstr)
            if tmpstr!=tmpstr1:
                if tmpstr1[0]!='-':
                    f1.write('------------'+'\n')
                f1.write(tmpstr)
                tmpstr1=tmpstr
        else:
            tmpstr1=tmpstr
            f1.write(i)
f1.close()
print('完成')

你可能感兴趣的:(Python,python)