下面的代码可能会对您有所帮助。它不是特别令人兴奋，并且故意简单。这不是许多程序员解决此问题的方式，但如果没有更多信息，它似乎可以满足您的要求。
我还假设您是 python 新手；如果我错了，请随意忽略这篇文章。
- 允许在命令行上传递数据库凭据、输出目录和日期（开始和结束）。
- 使用子进程代替 os.system。 Subprocess 提供了从 python 调用外部可执行文件的首选机制。这段代码使用了其中最简单的； call() 因为它类似于 os.system()
- 使用 optparse 处理命令行参数。虽然代码肯定更长更冗长，但您将来对 arg 处理进行添加和修改会更容易。发生了什么也很清楚（而且代码的阅读频率总是远远高于编写的频率）。
- 命令行设置仅在脚本执行时运行，因为它位于
__main__ 块中。由于脚本的“逻辑”在 main() 方法中，您还可以导入它并从其他来源提供选项对象（和 arg 列表）。
如果您无需在单独的文件中输出每个日期，您可以让数据库引擎计算 SUM() 并按日期对它们进行分组。您将在一个数据库调用中获得所有总和，这会更快并且可以生成更简单的代码。
#!/usr/bin/python
# -*- coding: utf-8 -*-
import datetime
import os
import subprocess
from optparse import OptionParser
SQL = """SELECT d.Date, SUM(d.CostUsd) FROM Stats d WHERE d.Date = '%s' GROUP BY d.Date"""
def get_stats(options, dateobj):
"""Return statistics for the date of `dateobj`"""
_datestr = dateobj.strftime('%Y-%m-%d')
sql = SQL % _datestr
filepath = os.path.join(options.outdir, 'DateLoop-%s.txt' % _datestr)
return subprocess.call('mysql -h %s -u %s -p -sN -e "%s" db > %s' % (options.dbhost, options.dbuser, sql, filepath), shell=True)
def main(options, args):
""""""
_date = options.startdate
while _date <= options.enddate:
rs = get_stats(options, _date)
_date += datetime.timedelta(days=1)
if __name__ == '__main__':
parser = OptionParser(version="%prog 1.0")
parser.add_option('-s', '--startdate', type='string', dest='startdate',
help='the start date (format: yyyymmdd)')
parser.add_option('-e', '--enddate', type='string', dest='enddate',
help='the end date (format: yyyymmdd)')
parser.add_option('--output', type='string', dest='outdir', default='/home/output/',
help='target directory for output files')
parser.add_option('--dbhost', type='string', dest='dbhost', default='myhost',
help='SQL server address')
parser.add_option('--dbuser', type='string', dest='dbuser', default='dbuser',
help='SQL server user')
options, args = parser.parse_args()
## Process the date args
if not options.startdate:
options.startdate = datetime.datetime.today()
else:
try:
options.startdate = datetime.datetime.strptime('%Y%m%d', options.startdate)
except ValueError:
parser.error("Invalid value for startdate (%s)" % options.startdate)
if not options.enddate:
options.enddate = options.startdate + datetime.timedelta(days=7)
else:
try:
options.enddate = datetime.datetime.strptime('%Y%m%d', options.enddate)
except ValueError:
parser.error("Invalid value for enddate (%s)" % options.enddate)
main(options, args)