Skip to content

Instantly share code, notes, and snippets.

@ogawa
Created December 22, 2010 10:10
Show Gist options
  • Save ogawa/751338 to your computer and use it in GitHub Desktop.
Save ogawa/751338 to your computer and use it in GitHub Desktop.
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# --- From: http://docs.python.org/library/csv.html
# --- BEGIN
import csv, codecs, cStringIO
class UTF8Recoder:
"""
Iterator that reads an encoded stream and reencodes the input to UTF-8
"""
def __init__(self, f, encoding):
self.reader = codecs.getreader(encoding)(f)
def __iter__(self):
return self
def next(self):
return self.reader.next().encode("utf-8")
class UnicodeReader:
"""
A CSV reader which will iterate over lines in the CSV file "f",
which is encoded in the given encoding.
"""
def __init__(self, f, dialect=csv.excel, encoding="utf-8", **kwds):
f = UTF8Recoder(f, encoding)
self.reader = csv.reader(f, dialect=dialect, **kwds)
def next(self):
row = self.reader.next()
return [unicode(s, "utf-8") for s in row]
def __iter__(self):
return self
class UnicodeWriter:
"""
A CSV writer which will write rows to CSV file "f",
which is encoded in the given encoding.
"""
def __init__(self, f, dialect=csv.excel, encoding="utf-8", **kwds):
# Redirect output to a queue
self.queue = cStringIO.StringIO()
self.writer = csv.writer(self.queue, dialect=dialect, **kwds)
self.stream = f
self.encoder = codecs.getincrementalencoder(encoding)()
def writerow(self, row):
self.writer.writerow([s.encode("utf-8") for s in row])
# Fetch UTF-8 output from the queue ...
data = self.queue.getvalue()
data = data.decode("utf-8")
# ... and reencode it into the target encoding
data = self.encoder.encode(data)
# write to the target stream
self.stream.write(data)
# empty queue
self.queue.truncate(0)
def writerows(self, rows):
for row in rows:
self.writerow(row)
# --- END
import sys, getopt
def usage():
print 'Usage: %s [--help] [--outputdir=/output/dir] inputfile' % sys.argv[0]
print
print ' --help print this message.'
print ' --outputdir specify an output directory.'
def main():
try:
optlist, args = getopt.getopt(sys.argv[1:], 'hd:', longopts=['help', 'outputdir='])
except getopt.GetoptError:
usage()
sys.exit(2)
OUTPUT_DIR = '/tmp/'
for opt, arg in optlist:
if opt in ('-h', '--help'):
usage()
sys.exit()
if opt in ('-d', '--outputdir'):
if arg.endswith('/'):
OUTPUT_DIR = arg
else:
OUTPUT_DIR = arg + '/'
if len(args) < 1:
print "filename not specified."
usage()
sys.exit(2)
reader = UnicodeReader(file(args[0], 'rU'), encoding='cp932')
# Ultrasoft MoneyLink emits:
# [0] 番号
# [1] 日付
# [2] 資産
# [3] 摘要
# [4] 照合
# [5] 金額
# [6] 費目
# [7] 内訳
# [8] 備考
data = {}
for row in reader:
account = row[2]
if account == u'資産':
continue
if not account in data:
data[account] = [
['Date','Payee','Category','Amount','Memo','Account']
]
d = row[1].split('/')
# date-format: MM/dd/yy
# date = '%02d/%02d/%02d' % (int(d[1]), int(d[2]), int(d[0][-2:]))
# date-format: yyyy/MM/dd
date = '%04d/%02d/%02d' % (int(d[0]), int(d[1]), int(d[2]))
payee = row[3]
if not payee:
payee = row[7]
category = row[6]
if category and row[7]:
category = category + ':' + row[7]
amount = row[5].replace(',', '')
memo = row[8]
data[account].append([
date, payee, category, amount, memo, account
])
for account in data.keys():
filename = OUTPUT_DIR + account + '.csv'
writer = UnicodeWriter(file(filename, 'w'), encoding='utf-8')
writer.writerows(data[account])
print filename+' created.'
if __name__ == "__main__":
main()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment