如何在不修改原命令的前提下,让argparse正确解析含\n的命令行参数并生成带换行的文件
如何在不修改原命令的前提下,让argparse正确解析含\n的命令行参数并生成带换行的文件
我明白你的问题啦——你想用python test.py -t "aa\nbb\ncc\n"这个命令,让程序把带换行的内容写入文件,但现在argparse把\n当成了普通的字符,导致文件里显示的是aa\nbb\ncc\n而不是真正的换行。关键问题在于:shell在传递参数时,不会把双引号里的\n解析成换行符,它只是把这两个字符原封不动传给了Python程序。
我们可以在Python代码里对拿到的参数做转义字符解析,把字符串里的\n转换成实际的换行符。Python里有个很方便的方法:用str.encode().decode('unicode_escape'),它能自动处理常见的转义序列(比如\n、\t这些)。
把你的代码修改成下面这样就可以解决问题:
#! /usr/bin/env python3 # -*- coding: utf-8 -*- import argparse def wwrite(): # parser parser = argparse.ArgumentParser(description='Atest.') parser.add_argument('-t', '--tail', help='Add some lines to file') args = parser.parse_args() # check if args.tail is not None: # 新增转义解析步骤,将字符串中的\n转为实际换行符 zz_tail = args.tail.encode().decode('unicode_escape') else: zz_tail = '' # write with open("aaa", 'w', encoding='utf-8') as fw: fw.write(zz_tail) with open("bbb", 'w', encoding='utf-8') as fw: fw.write("aa\nbb\ncc") # 主文件 if __name__ == "__main__": wwrite()
现在再运行你原来的命令python test.py -t "aa\nbb\ncc\n",打开aaa文件就会和bbb一样,显示成:
aa bb cc
这个方法的好处是完全不用修改你的命令,只需要在代码里加一步转义解析。如果之后需要处理其他转义字符(比如\t制表符),这个方法也能自动识别处理,非常实用。
备注:内容来源于stack exchange,提问作者cwind
相关产品推荐
相关产品推荐

