def get(self, bank_id):
bank = banks.get_bank(bank_id)
if not bank:
abort(404, message="Not found. Supported banks = %s" % ",".join(banks.get_bank_ids()))
data = bank.quote()
quote_date = datetime.datetime.fromtimestamp(data["date"]) if "date" in data else None
ancestor_key = ndb.Key("Bank", bank.name())
# sanity check if quotes are already stored
if quote_date:
qry = XchgRecord.query(XchgRecord.quote_date == quote_date, ancestor=ancestor_key)
if qry.get(keys_only=True):
return "", 204
for c, r in data["data"].iteritems():
rec = XchgRecord(parent=ancestor_key)
rec.base_currency = c
rec.to_currency = "TWD"
rec.cash_buy = r[0]
rec.cash_sell = r[1]
rec.spot_buy = r[2]
rec.spot_sell = r[3]
if quote_date:
rec.quote_date = quote_date
# TODO: put quote records at once
rec.put()
return "", 201
def get(self, bank_id):
args = self.parser.parse_args(strict=True)
bank = banks.get_bank(bank_id)
if not bank:
abort(404, message="Not found. Supported banks = %s" % ",".join(banks.get_bank_ids()))
try:
start_date = datetime.datetime.strptime(args["start_date"], "%Y%m%d")
# until the end of the day
end_date = datetime.datetime.strptime(args["end_date"], "%Y%m%d") + datetime.timedelta(days=1)
except:
abort(400, message="Unable to parse start/end date")
ancestor_key = ndb.Key("Bank", bank.name())
data = {"quotes": []}
if (end_date - start_date) > datetime.timedelta(days=1):
# FIXME we need to generate a different table for this instead of query ndb day by day
for sd in daterange(start_date, end_date):
ed = sd + datetime.timedelta(days=1)
x = (
XchgRecord.get_latest_quotes(ancestor_key)
.filter(XchgRecord.base_currency == args["currency"])
.filter(XchgRecord.quote_date <= ed)
.filter(XchgRecord.quote_date >= sd)
.get()
)
if x:
x.quote_date -= datetime.timedelta(hours=8) # FIXME: ndb records are stored as GMT +8
data["quotes"].append(
[
x.quote_date.strftime("%Y-%m-%dT%H:%M:%S.%fZ"),
x.cash_buy,
x.cash_sell,
x.spot_buy,
x.spot_sell,
]
)
else:
for x in (
XchgRecord.get_latest_quotes(ancestor_key)
.filter(XchgRecord.base_currency == args["currency"])
.filter(XchgRecord.quote_date <= end_date)
.filter(XchgRecord.quote_date >= start_date)
.fetch()
):
x.quote_date -= datetime.timedelta(hours=8) # FIXME: ndb records are stored as GMT +8
data["quotes"].append(
[x.quote_date.strftime("%Y-%m-%dT%H:%M:%S.%fZ"), x.cash_buy, x.cash_sell, x.spot_buy, x.spot_sell]
)
return jsonify(data)