Python 将 0 写入文件

Question

import os


fileHandle = open('booksNames.txt', 'r+')

def getData():
    data = os.system('dir /b /a /s *.pdf *.epub *.mobi')
    fileHandle.writelines(str(data))

fileHandle.close()

我正在尝试将 os.system 函数返回的数据写入文件。但唯一写入文件的是 0。以下是我尝试过的其他一些变体。

import os

fileHandle = open('booksNames.txt', 'r+')

getData = lambda:os.system('dir /b /a /s *.pdf *.epub *.mobi')
data = getData()

fileHandle.writelines(str(data))
fileHandle.close()

在输出窗口中，它提供了完美的输出，但在写入文本文件时它会写入零。我也尝试过使用 return 但没有用。请帮忙。

Answer 1

对于 Windows（我假设您使用的是 Windows，因为您使用的是“dir”命令，而不是 Unix/Linux 的“ls”）：

只需让命令完成工作。

os.system('dir /b /a /s *.pdf *.epub *.mobi >> booksNames.txt')

使用“>>”将追加到任何现有文件。只需使用 '>' 写入一个新文件。

我喜欢使用子进程的其他解决方案，但由于这是特定于操作系统的，我认为这更简单。

Answer 2

使用subprocess 模块。方法有很多，但最简单的是：

>>> import subprocess
>>> with open('out.txt','w') as f:
...     subprocess.call(['dir','/b','/a','/s','*.pdf','*.epub','*.mobi'],stdout=f,stderr=f,shell=True)
...
0

零是退出代码，但内容将在out.txt中。