python文件第三关

任务描述

本关任务：读取wjcl/src/step1/宋词文本文件，为每位作者在wjcl/src/step5/cr下建立文件夹，再在作者文件夹下为每首词建立一个文件 注意：标题行的词牌名和作者之间是全角空格 （"　")可在此复制

李清照文件夹下的文件

测试说明

平台会对你编写的代码进行测试：

测试输入： 李清照 声声慢 预期输出： `声声慢 李清照

寻寻觅觅，冷冷清清，凄凄惨惨戚戚。乍暖还寒时侯，最难将息。三杯两 盏淡酒，怎敌他、晚来风急。雁过也，最伤心，却是旧时相识。 满地黄花堆积，惟悴损、如今有谁堪摘。守着窗儿，独自怎生得黑？梧桐 更兼细雨，到黄昏、点点滴滴。这次第，怎一个愁字了得。`

测试输入： 范仲淹 苏幕遮 预期输出： `苏幕遮　范仲淹

碧云天，黄叶地，秋色连波，波上寒烟翠。山映斜阳天接水，芳草无情， 更在斜阳外。 黯乡魂，追旅思。夜夜除非，好梦留人睡。明月楼高休独倚，酒入愁肠， 化作相思泪。`

---

开始你的任务吧，祝你成功！

 

import os
import shutil
if  os.path.exists("wjcl/src/step5/cr"):
    shutil.rmtree("wjcl/src/step5/cr")
os.mkdir("wjcl/src/step5/cr")
f1=open("wjcl/src/step1/宋词.txt",'r')
#代码开始
cr=set()
for line in f1:
    if "　" in line.strip():
        a=line.find("　")
        xm=line.strip()[a+1:]
        cr.add(xm)
for i in cr:
    lj="wjcl/src/step5/cr/"+i
    if not os.path.exists(lj):
        os.mkdir(lj)
f1.seek(0)
for line in f1:
    if "　" in line.strip():
        a=line.find("　")
        sxm=line.strip()[a+1:]
        cpm=line.strip()[:a]
        f2=open("wjcl/src/step5/cr/"+sxm+"/"+cpm+".txt","a+")
    f2.write(line)

#代码结束