下面的代码可能会帮你解决问题。它不是特别令人兴奋,而且故意简单。这不是很多程序员解决这个问题的方式,但是如果没有更多的信息,它似乎可以满足您的需求。
我还假设您是python新手;如果我错了,可以忽略这篇文章。允许在命令行上传递数据库凭据、输出目录和日期(开始和结束)。
使用子进程代替os.system。Subprocess提供了从python调用外部可执行文件的首选机制。这段代码使用其中最简单的一个:call(),因为它类似于os.system()
使用optparse处理命令行参数。虽然代码肯定会更长、更详细,但将来对arg处理进行添加和修改会更容易。它也很清楚发生了什么(而且代码的读取频率总是远远高于它的写入频率)。
命令行设置只在脚本在__main__块中执行时运行。由于脚本的“逻辑”在main()方法中,您还可以导入它并从另一个源提供options对象(和arg list)。
如果不需要将每个日期输出到单独的文件中,则可以让数据库引擎计算SUM()并按日期对它们进行分组。您可以在一个db调用中得到所有的和,这样会更快,并且可以生成更简单的代码。#!/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)