diff options
Diffstat (limited to 'ofxparse/ofxprinter.py')
| -rw-r--r-- | ofxparse/ofxprinter.py | 194 |
1 files changed, 194 insertions, 0 deletions
diff --git a/ofxparse/ofxprinter.py b/ofxparse/ofxprinter.py new file mode 100644 index 0000000..4c0a13d --- /dev/null +++ b/ofxparse/ofxprinter.py @@ -0,0 +1,194 @@ +class OfxPrinter(): + ofx = None + out_filename = None + out_handle = None + term = "\r\n" + + def __init__(self, ofx, filename, term="\r\n"): + self.ofx = ofx + self.out_filename = filename + self.term = term + + def writeLine(self, data, tabs=0, term=None): + if term is None: + term = self.term + + tabbing = (tabs * "\t") if (tabs > 0) else '' + + return self.out_handle.write( + "{0}{1}{2}".format( + tabbing, + data, + term + ) + ) + + def writeHeaders(self): + for k, v in self.ofx.headers.iteritems(): + if v is None: + self.writeLine("{0}:NONE".format(k)) + else: + self.writeLine("{0}:{1}".format(k, v)) + self.writeLine("") + + def writeSignOn(self, tabs=0): + # signon already has newlines and tabs in it + # TODO: reimplement signon printing with tabs + self.writeLine(self.ofx.signon.__str__(), term="") + + def printDate(self, dt, msec_digs=3): + strdt = dt.strftime('%Y%m%d%H%M%S') + strdt_msec = dt.strftime('%f') + if len(strdt_msec) < msec_digs: + strdt_msec += ('0' * (msec_digs - len(strdt_msec))) + elif len(strdt_msec) > msec_digs: + strdt_msec = strdt_msec[:msec_digs] + return strdt + '.' + strdt_msec + + def writeTrn(self, trn, tabs=5): + self.writeLine("<STMTTRN>", tabs=tabs) + tabs += 1 + + self.writeLine("<TRNTYPE>{}".format(trn.type.upper()), tabs=tabs) + self.writeLine("<DTPOSTED>{}".format( + self.printDate(trn.date) + ), tabs=tabs) + self.writeLine("<TRNAMT>{0:.2f}".format(float(trn.amount)), tabs=tabs) + + self.writeLine("<FITID>{}".format(trn.id), tabs=tabs) + + if len(str(trn.checknum)) > 0: + self.writeLine("<CHECKNUM>{}".format( + trn.checknum + ), tabs=tabs) + + self.writeLine("<NAME>{}".format(trn.payee), tabs=tabs) + + if len(trn.memo.strip()) > 0: + self.writeLine("<MEMO>{}".format(trn.memo), tabs=tabs) + + tabs -= 1 + self.writeLine("</STMTTRN>", tabs=tabs) + + def writeLedgerBal(self, statement, tabs=4): + bal = getattr(statement, 'balance') + baldt = getattr(statement, 'balance_date') + + if bal and baldt: + self.writeLine("<LEDGERBAL>", tabs=tabs) + self.writeLine("<BALAMT>{0:.2f}".format(float(bal)), tabs=tabs+1) + self.writeLine("<DTASOF>{0}".format( + self.printDate(baldt) + ), tabs=tabs+1) + self.writeLine("</LEDGERBAL>", tabs=tabs) + + def writeAvailBal(self, statement, tabs=4): + bal = getattr(statement, 'available_balance') + baldt = getattr(statement, 'available_balance_date') + + if bal and baldt: + self.writeLine("<AVAILBAL>", tabs=tabs) + self.writeLine("<BALAMT>{0:.2f}".format(float(bal)), tabs=tabs+1) + self.writeLine("<DTASOF>{0}".format( + self.printDate(baldt) + ), tabs=tabs+1) + self.writeLine("</AVAILBAL>", tabs=tabs) + + def writeStmTrs(self, tabs=3): + for acct in self.ofx.accounts: + self.writeLine("<STMTRS>", tabs=tabs) + tabs += 1 + + if acct.curdef: + self.writeLine("<CURDEF>{0}".format( + acct.curdef + ), tabs=tabs) + + if acct.routing_number or acct.account_id or acct.account_type: + self.writeLine("<BANKACCTFROM>", tabs=tabs) + if acct.routing_number: + self.writeLine("<BANKID>{0}".format( + acct.routing_number + ), tabs=tabs+1) + if acct.account_id: + self.writeLine("<ACCTID>{0}".format( + acct.account_id + ), tabs=tabs+1) + if acct.account_type: + self.writeLine("<ACCTTYPE>{0}".format( + acct.account_type + ), tabs=tabs+1) + self.writeLine("</BANKACCTFROM>", tabs=tabs) + + self.writeLine("<BANKTRANLIST>", tabs=tabs) + tabs += 1 + self.writeLine("<DTSTART>{0}".format( + self.printDate(acct.statement.start_date) + ), tabs=tabs) + self.writeLine("<DTEND>{0}".format( + self.printDate(acct.statement.end_date) + ), tabs=tabs) + + for trn in acct.statement.transactions: + self.writeTrn(trn, tabs=tabs) + + tabs -= 1 + + self.writeLine("</BANKTRANLIST>", tabs=tabs) + + self.writeLedgerBal(acct.statement, tabs=tabs) + self.writeAvailBal(acct.statement, tabs=tabs) + + tabs -= 1 + + self.writeLine("</STMTRS>", tabs=tabs) + + def writeBankMsgsRsv1(self, tabs=1): + self.writeLine("<BANKMSGSRSV1>", tabs=tabs) + tabs += 1 + self.writeLine("<STMTTRNRS>", tabs=tabs) + tabs += 1 + if self.ofx.trnuid is not None: + self.writeLine("<TRNUID>{0}".format( + self.ofx.trnuid + ), tabs=tabs) + if self.ofx.status: + self.writeLine("<STATUS>", tabs=tabs) + self.writeLine("<CODE>{0}".format( + self.ofx.status['code'] + ), tabs=tabs+1) + self.writeLine("<SEVERITY>{0}".format( + self.ofx.status['severity'] + ), tabs=tabs+1) + self.writeLine("</STATUS>", tabs=tabs) + self.writeStmTrs(tabs=tabs) + tabs -= 1 + self.writeLine("</STMTTRNRS>", tabs=tabs) + tabs -= 1 + self.writeLine("</BANKMSGSRSV1>", tabs=tabs) + + def writeOfx(self, tabs=0): + self.writeLine("<OFX>", tabs=tabs) + tabs += 1 + self.writeSignOn(tabs=tabs) + self.writeBankMsgsRsv1(tabs=tabs) + tabs -= 1 + # No newline at end of file + self.writeLine("</OFX>", tabs=tabs, term="") + + def write(self, filename=None, tabs=0): + if self.out_handle: + raise Exception("Already writing file") + + if filename is None: + filename = self.out_filename + + self.out_handle = open(filename, 'wb') + + self.writeHeaders() + + self.writeOfx(tabs=tabs) + + self.out_handle.flush() + self.out_handle.close() + self.out_handle = None |