Python的subprocess调用ls命令时能否输出带颜色的结果？

问题原因 #

GNU ls 默认会自动检测输出目标：如果输出是交互式终端（TTY），就自动开启颜色输出；如果输出是管道、文件或者被其他程序捕获，就默认关闭颜色避免输出多余的ANSI转义码。你通过Python subprocess 调用ls时，默认stdout没有关联到TTY，所以ls自动禁用了颜色。

解决方法1：强制ls输出颜色（最简单） #

直接给ls命令添加 --color=always 参数即可强制开启颜色输出，修改后的代码参考：

import argparse
import subprocess
import sys # 可选，用来判断当前输出是否为TTY，适配管道场景

parser = argparse.ArgumentParser(description='A fancy version of ls')
group = parser.add_mutually_exclusive_group(required=False)
group.add_argument('-n', '--name', action='store_true', help='Sort entries by name')
group.add_argument('-c', '--created', action='store_true', help='Sort entries by creation date')

args = parser.parse_args()

ls = ['ls']
# 智能判断：只有脚本输出到终端时才开颜色，避免管道输出带多余转义码
if sys.stdout.isatty():
    ls.append('--color=always') # 如果你用macOS自带的BSD ls，这里换成'-G'

flags = '-AbcFghkLs'  # 排序参数前的默认输出参数
if args.name:
    flags += 'l'
    ls.append(flags)
elif args.created:
    ls.append(flags)
else:
    ls.append('-AbcFghkLst')
    
subprocess.run(ls)

解决方法2：模拟TTY执行（不需要修改ls参数） #

如果不想改动ls的调用参数，可以通过模拟TTY的方式执行命令，让ls以为自己是直接输出到终端，自动开启颜色。可以用Python标准库的pty模块实现：

import os
import pty

def run_ls_with_tty(cmd):
    pid, fd = pty.fork()
    if pid == 0:
        os.execvp(cmd[0], cmd)
    else:
        output = os.read(fd, 1024*1024)
        os.waitpid(pid, 0)
        print(output.decode(), end='')

# 替换原来的subprocess.run调用即可
run_ls_with_tty(ls)

这种方式不需要调整ls参数，但输出可能会附带额外的终端控制符，如果你需要解析ls的输出内容不推荐使用。

其他实现lf的方案 #

如果你不想依赖系统的ls命令，也可以纯Python实现lf的功能：

• 用pathlib遍历目录获取文件列表
• 用os.stat读取文件的创建时间、修改时间、权限、大小等元数据
• 自行根据文件类型、权限匹配对应的ANSI颜色码输出
• 自主实现排序逻辑
  这种方式灵活性更高，不需要适配不同系统的ls参数差异，但是开发量会相对大一些。

内容的提问来源于stack exchange，提问作者mintermine