def _init(self, stmtrs):
# BANKTRANLIST
tranlist = stmtrs.find('BANKTRANLIST')
if tranlist is not None:
self.transactions = TransactionList(self.account, tranlist)
DBSession.add_all(self.transactions)
# LEDGERBAL - mandatory
ledgerbal = stmtrs.find('LEDGERBAL')
self.ledgerbal = ledgerbal.instantiate(acctfrom=self.account)
DBSession.add(self.ledgerbal)
# AVAILBAL
availbal = stmtrs.find('AVAILBAL')
if availbal is not None:
self.availbal = availbal.instantiate(acctfrom=self.account)
DBSession.add(self.availbal)
else:
self.availbal = None
ballist = stmtrs.find('BALLIST')
if ballist:
self.other_balances = [bal.instantiate() for bal in ballist]
DBSession.add_all(self.other_balances)
# Unsupported subaggregates
for tag in ('MKTGINFO', ):
child = stmtrs.find(tag)
if child:
stmtrs.remove
def instantiate(self, **extra_attrs):
"""
Create an instance of a SQLAlchemy model class corresponding to
my OFX tag, with attributes given by my contained OFX elements.
If an instance that matches the given primary key signature has
already been given, return that instead of creating a new one.
"""
self.extra_attributes = extra_attrs
# SECID needs to instantiate as SECINFO
if self.tag == 'SECID':
SubClass = models.SECINFO
else:
SubClass = getattr(models, self.tag)
self._preflatten()
self.attributes = self._flatten()
self._postflatten()
self.attributes.update(self.extra_attributes)
self.extra_attributes = {}
try:
fingerprint = SubClass._fingerprint(**self.attributes)
instance = DBSession.query(SubClass).filter_by(**fingerprint).one()
except NoResultFound:
instance = SubClass(**self.attributes)
DBSession.add(instance)
return instance
def __init__(self, stmtrs):
""" Initialize with *STMTRS Element """
self.currency = stmtrs.find('CURDEF').text
acctfrom = stmtrs.find(self._acctTag)
self.account = acctfrom.instantiate()
DBSession.add(self.account)
self.transactions = []
self.other_balances =[]
self._init(stmtrs)
def _init(self, invstmtrs):
dtasof = invstmtrs.find('DTASOF').text
self.datetime = ofxtools.types.DateTime().convert(dtasof)
# INVTRANLIST
tranlist = invstmtrs.find('INVTRANLIST')
if tranlist is not None:
self.transactions = TransactionList(self.account, tranlist)
DBSession.add_all(self.transactions)
# INVPOSLIST
poslist = invstmtrs.find('INVPOSLIST')
if poslist is not None:
# FIs can list multiple INVPOS lots per security, so we
# can't just naively instantiate what they give us or we'll
# violate uniqueness constraints on the PKs.
# Instead we need to manually tally up the units and instantiate
# only a single INVPOS lot per security
positions = {}
for pos in poslist:
secid = pos.find('.//SECID')
uniqueid = secid.find('UNIQUEID')
uniqueidtype = secid.find('UNIQUEIDTYPE')
units = pos.find('./INVPOS/UNITS')
seckey = (uniqueid.text, uniqueidtype.text)
position = positions.get(seckey, None)
if position is None:
positions[seckey] = (pos, Decimal(units.text))
else:
positions[seckey] = (position[0],
position[1] + Decimal(units.text)
)
self.positions = [pos.instantiate(
units=units, acctfrom=self.account,
dtasof=self.datetime) \
for pos, units in positions.values()]
DBSession.add_all(self.positions)
else:
self.positions = []
# INVBAL
invbal = invstmtrs.find('INVBAL')
if invbal is not None:
# First strip off BALLIST & process it
ballist = invbal.find('BALLIST')
if ballist is not None:
invbal.remove(ballist)
self.other_balances = [
bal.instantiate(acctfrom=self.account, dtasof=self.datetime,
) for bal in ballist
]
DBSession.add_all(self.other_balances)
# Now we can flatten the rest of INVBAL
self.balances = invbal.instantiate(
acctfrom=self.account, dtasof=self.datetime,
)
DBSession.add(self.balances)
else:
self.balances = []
# Unsupported subaggregates
for tag in ('INVOOLIST', 'INV401K', 'INV401KBAL', 'MKTGINFO'):
child = invstmtrs.find(tag)
if child is not None:
invstmtrs.remove