使默认输出文件依赖于输入文件

Question

我正在尝试制作一个将一种文件类型转换为另一种文件类型的 python 脚本，我希望可以选择指定一个输出文件，但默认情况下只是更改默认文件名的扩展名。

例如：我希望convert('foo.gb') 输出foo.faa，但convert('foo.gb', 'bar.faa') 输出bar.faa

我的实现方式是：

#!/usr/bin/env python

def convert(inputFile, outputFile = None):
    [code that converts data]

    if not outputFile:
        import re
        name = re.match('(.+)\.\w+', inputFile)
        outputFile = './{0}.faa'.format(name.group(1))

    with open(outputFile, 'w+') as outFaa:
        outputFaa.write([stuff to write])

因此，它检查是否指定了输出，如果没有指定，则使用正则表达式更改输入文件名以具有正确的扩展名。这段代码有效，但似乎有些草率，或者至少不是很可读。如果文件名在扩展名之前以外的任何地方都有.，它也会中断，它有时可能会有。

有没有更好的方法来做到这一点？

Answer 1

请使用友好人士在您的 Python 安装中提供的电池。看看os.path，尤其是splitext。

那么你可以这样写：

def convert(filename, changeto=None):
    basename, ext = os.path.splitext(in)
    if changeto is None:
        return basename + ".faa"
    return changeto

在您的应用程序中，您需要注意路径结构。我只想对不包含目录分隔符的文件名使用上面的函数。

Answer 2

import os

if output_fname is None:
    basename, ext = os.path.splitext(input_fname)
    output_fname = basename + ".faa"

请参阅https://docs.python.org/3.4/library/os.path.html 的文档

编辑：还不错；我真正要做的唯一不同的事情是让输出文件名处理函数中的第一件事。这只是让下一个人更容易弄清楚参数的来源以及如何正确使用它们：

import os
def convert(in_fname, out_fname = None):
    # start by sanitizing parameters
    if out_fname is None:
        basename, ext = os.path.splitext(in_fname)
        out_fname = basename + ".faa"

    # read data
    with open(in_fname) as inf:
        data = inf.read()   # or inf.readlines()

    # do conversion
    result = do_conversion(data)

    # write result
    with open(out_fname, "w") as outf:
        outf.write(result)  # or .write("\n".join(result))