在 LeetCode 上刷题的时候,总需在要本地创建一个文件以存储一些代码。
可每次都需要手动输入相关信息就让人很烦,因此写了如下脚本,可以稍稍简化流程。
"""
生成一个存储 LeetCode 刷题信息所需的文档,文档内包含必要的信息
后续规划:
结合爬虫库,实现只需要 url 即可创建一个文件夹
"""
import os
def createFile(dic):
"""创建文件"""
contents = f"""\"""
Title:
{dic['title_en']}
{dic['title_zh']}
Address:
{dic['address']}
\"""
"""
title = dic['title_en']
newTitle = f"{title.replace('. ', '_').replace(' ', '-')}.py"
os.chdir(r"E:\LeetCode")
with open( newTitle, 'w', encoding='utf-8' ) as f:
f.write(contents)
def createDict():
"""创建一个存储英文标题、中文标题和地址的字典
'title_en': 英文标题
'title_zh': 中文标题
'address': 对应网页的 URL
"""
dic = {}
title_en = input("Engish Title:")
dic['title_en'] = title_en
title_zh = input("Chinese Title:")
dic['title_zh'] = title_zh
address = input("Address:")
dic['address'] = address
return dic
if __name__ == '__main__':
dic = createDict()
createFile(dic)
print("The End")