def trade_account_edit():
investor_name = request.form.get('investor')
i = Investor.query_one({'name': investor_name})
if not i:
return jsonify(status=404,
reason='投资人name={}未找到'.format(investor_name))
user = current_user._id
ta = {
'user': user,
'investor': i._id,
'exchange': request.form.get('exchange'),
'bank': request.form.get('bank'),
'login_name': request.form.get('login_name'),
'login_password': request.form.get('login_password'),
'money_password': request.form.get('money_password'),
}
try:
TradeAccount(ta).upsert()
except Exception as e:
TradeAccount.delete_one({
'investor': ta['investor'],
'exchange': ta['exchange']
})
try:
TradeAccount(ta).upsert()
except Exception as e:
log.exception(str(e))
return jsonify(status=500, reason=str(e))
return jsonify(status=200, reason='')
def trade_account_edit():
investor_name = request.form.get('investor')
i = Investor.query_one({'name': investor_name})
if not i:
return jsonify(
status=404,
reason='投资人name={}未找到'.format(investor_name))
user = current_user._id
ta = {
'user': user,
'investor': i._id,
'exchange': request.form.get('exchange'),
'bank': request.form.get('bank'),
'login_name': request.form.get('login_name'),
'login_password': request.form.get('login_password'),
'money_password': request.form.get('money_password'),
}
try:
TradeAccount(ta).upsert()
except Exception as e:
TradeAccount.delete_one({'investor': ta['investor'],
'exchange': ta['exchange']})
try:
TradeAccount(ta).upsert()
except Exception as e:
log.exception(str(e))
return jsonify(status=500, reason=str(e))
return jsonify(status=200, reason='')
def investor_delete():
_id = request.form.get('id')
i = Investor.query_one({'_id': _id})
if i:
try:
i.remove()
except Exception as e:
log.exception(str(e))
return jsonify(status=500, reason=str(e))
else:
return jsonify(status=200, reason='')
else:
return jsonify(status=404, reason='投资人id={}未找到, 无法删除'.format(_id))
def investor_delete():
_id = request.form.get('id')
i = Investor.query_one({'_id': _id})
if i:
try:
i.remove()
except Exception as e:
log.exception(str(e))
return jsonify(status=500, reason=str(e))
else:
return jsonify(status=200, reason='')
else:
return jsonify(status=404, reason='投资人id={}未找到, 无法删除'.format(_id))
def investor_upsert():
def extract_image(src):
if src:
return base64.b64decode(src[src.find('base64,') + 7:])
else:
return b''
user = current_user._id
d = json.loads(request.data.decode('utf-8'))
bank_accounts = d['bank_accounts']
for ba in bank_accounts:
ba['number'] = ba['number'].replace(' ', '')
i = {
'user': user,
'name': d.get('name', ''),
'mobile': d.get('mobile', ''),
'id_type': '身份证',
'id_number': d.get('id_number', ''),
'id_front': extract_image(d.get('id_front', '')),
'id_back': extract_image(d.get('id_back', '')),
'province': d.get('province', ''),
'city': d.get('city', ''),
'address': d.get('address', ''),
'bank_accounts': bank_accounts,
}
oldi = Investor.query_one({'user': user}, sort=[('order', -1)])
thei = Investor.query_one({'id_number': i['id_number']})
if not thei:
i['order'] = oldi.order + 1 if oldi else 1
else:
i['order'] = thei.order
try:
Investor(i).upsert()
except Exception as e:
log.exception(str(e))
return jsonify(status=500, reason=str(e))
return jsonify(status=200, reason='')
def investor_upsert():
def extract_image(src):
if src:
return base64.b64decode(src[src.find('base64,') + 7:])
else:
return b''
user = current_user._id
d = json.loads(request.data.decode('utf-8'))
bank_accounts = d['bank_accounts']
for ba in bank_accounts:
ba['number'] = ba['number'].replace(' ', '')
i = {
'user': user,
'name': d.get('name', ''),
'mobile': d.get('mobile', ''),
'id_type': '身份证',
'id_number': d.get('id_number', ''),
'id_front': extract_image(d.get('id_front', '')),
'id_back': extract_image(d.get('id_back', '')),
'province': d.get('province', ''),
'city': d.get('city', ''),
'address': d.get('address', ''),
'bank_accounts': bank_accounts,
}
oldi = Investor.query_one({'user': user}, sort=[('order', -1)])
thei = Investor.query_one({'id_number': i['id_number']})
if not thei:
i['order'] = oldi.order + 1 if oldi else 1
else:
i['order'] = thei.order
try:
Investor(i).upsert()
except Exception as e:
log.exception(str(e))
return jsonify(status=500, reason=str(e))
return jsonify(status=200, reason='')
def personal_info():
nav = 'accounts'
tab = 'personal_info'
user = current_user._id
investors = Investor.user_investors(user)
investor_ids = [i._id for i in investors]
trade_accounts = TradeAccount.user_accounts(user)
account_table = {}
for ta in trade_accounts:
if ta.exchange not in account_table:
account_table[ta.exchange] = [None] * len(investors)
order = investor_ids.index(ta.investor)
account_table[ta.exchange][order] = ta
colspan = len(investors) + 1
return render_template('user/personal_info.html', **locals())
def personal_info():
nav = 'accounts'
tab = 'personal_info'
user = current_user._id
investors = Investor.user_investors(user)
investor_ids = [i._id for i in investors]
trade_accounts = TradeAccount.user_accounts(user)
account_table = {}
for ta in trade_accounts:
if ta.exchange not in account_table:
account_table[ta.exchange] = [None] * len(investors)
order = investor_ids.index(ta.investor)
account_table[ta.exchange][order] = ta
colspan = len(investors) + 1
return render_template('user/personal_info.html', **locals())
def investor_edit():
def pack_image(b):
if b:
return base64.b64encode(b).decode('ascii')
else:
return ''
_id = request.args.get('id')
i = Investor.query_one({'_id': _id})
if not i:
return jsonify(status=404, reason='投资人id={}未找到'.format(_id))
else:
i = i.to_dict()
i['id_front_b64'] = pack_image(i['id_front'])
i['id_back_b64'] = pack_image(i['id_back'])
del i['id_front']
del i['id_back']
return jsonify(status=200, data=i, reason='')
def investor_edit():
def pack_image(b):
if b:
return base64.b64encode(b).decode('ascii')
else:
return ''
_id = request.args.get('id')
i = Investor.query_one({'_id': _id})
if not i:
return jsonify(status=404, reason='投资人id={}未找到'.format(_id))
else:
i = i.to_dict()
i['id_front_b64'] = pack_image(i['id_front'])
i['id_back_b64'] = pack_image(i['id_back'])
del i['id_front']
del i['id_back']
return jsonify(status=200, data=i, reason='')
def investor_info():
nav = 'accounts'
tab = 'investor_info'
user = current_user._id
investors = Investor.user_investors(user)
return render_template('user/investor_info.html', **locals())
def trade_account():
nav = 'accounts'
tab = 'trade_account'
investor = request.args.get('investor', '')
exchange = request.args.get('exchange', '')
add_investor = request.args.get('add_investor', '')
add_exchange = request.args.get('add_exchange', '')
bank = request.args.get('bank', '')
investor = Investor.query_one({'_id': investor})
exchange = Exchange.query_one({'abbr': exchange})
user = current_user._id
trade_accounts = TradeAccount.user_accounts(user)
if investor:
trade_accounts = [ta for ta in trade_accounts
if ta.investor == investor._id]
if exchange:
trade_accounts = [ta for ta in trade_accounts
if ta.exchange == exchange.abbr]
if bank:
trade_accounts = [ta for ta in trade_accounts
if ta.bank == bank]
investors = Investor.user_investors(user)
exchanges = sorted(list(Exchange.query()), key=lambda x: x.abbr)
exchange_list = [{'text': e.abbr, 'value': e.abbr} for e in exchanges]
investor_list = [{'text': i.name, 'value': i.name} for i in investors]
investor_banks = {i.name: sorted([ba.bank for ba in i.bank_accounts])
for i in investors}
exchange_banks = {conf['abbr']: sorted(conf['opening']['bank'])
for conf in CONFS}
bank_exchanges = {}
for ex, banks in exchange_banks.items():
for b in banks:
if b not in bank_exchanges:
bank_exchanges[b] = []
bank_exchanges[b].append(ex)
investor_exchanges = {
invesotr: sorted(set(sum(
[bank_exchanges[b] for b in banks], [])))
for invesotr, banks in investor_banks.items()
}
avail_investors = set(ta.investor for ta in trade_accounts)
avail_exchanges = set(ta.exchange for ta in trade_accounts)
avail_banks = set(ta.bank for ta in trade_accounts)
investors = [i for i in investors if i._id in avail_investors]
exchanges = [e for e in exchanges if e.abbr in avail_exchanges]
thebanks = sorted(list(avail_banks))
account_positions = {ta._id: sorted([p.to_dict() for p in ta.position],
key=lambda p: p['symbol'])
for ta in trade_accounts}
account_moneys = {ta._id: ta.money.to_dict() if ta.money else {}
for ta in trade_accounts}
account_order_status = {ta._id: sorted([o.to_dict()
for o in ta.order_status],
key=lambda o: o['order'])
for ta in trade_accounts}
account_orders = {ta._id: sorted([o.to_dict() for o in ta.orders],
key=lambda o: o['symbol'])
for ta in trade_accounts}
return render_template('user/trade_account.html', **locals())
def investor_info():
nav = 'accounts'
tab = 'investor_info'
user = current_user._id
investors = Investor.user_investors(user)
return render_template('user/investor_info.html', **locals())
def trade_account():
nav = 'accounts'
tab = 'trade_account'
investor = request.args.get('investor', '')
exchange = request.args.get('exchange', '')
add_investor = request.args.get('add_investor', '')
add_exchange = request.args.get('add_exchange', '')
bank = request.args.get('bank', '')
investor = Investor.query_one({'_id': investor})
exchange = Exchange.query_one({'abbr': exchange})
user = current_user._id
trade_accounts = TradeAccount.user_accounts(user)
if investor:
trade_accounts = [
ta for ta in trade_accounts if ta.investor == investor._id
]
if exchange:
trade_accounts = [
ta for ta in trade_accounts if ta.exchange == exchange.abbr
]
if bank:
trade_accounts = [ta for ta in trade_accounts if ta.bank == bank]
investors = Investor.user_investors(user)
exchanges = sorted(list(Exchange.query()), key=lambda x: x.abbr)
exchange_list = [{'text': e.abbr, 'value': e.abbr} for e in exchanges]
investor_list = [{'text': i.name, 'value': i.name} for i in investors]
investor_banks = {
i.name: sorted([ba.bank for ba in i.bank_accounts])
for i in investors
}
exchange_banks = {
conf['abbr']: sorted(conf['opening']['bank'])
for conf in CONFS
}
bank_exchanges = {}
for ex, banks in exchange_banks.items():
for b in banks:
if b not in bank_exchanges:
bank_exchanges[b] = []
bank_exchanges[b].append(ex)
investor_exchanges = {
invesotr: sorted(set(sum([bank_exchanges[b] for b in banks], [])))
for invesotr, banks in investor_banks.items()
}
avail_investors = set(ta.investor for ta in trade_accounts)
avail_exchanges = set(ta.exchange for ta in trade_accounts)
avail_banks = set(ta.bank for ta in trade_accounts)
investors = [i for i in investors if i._id in avail_investors]
exchanges = [e for e in exchanges if e.abbr in avail_exchanges]
thebanks = sorted(list(avail_banks))
account_positions = {
ta._id: sorted([p.to_dict() for p in ta.position],
key=lambda p: p['symbol'])
for ta in trade_accounts
}
account_moneys = {
ta._id: ta.money.to_dict() if ta.money else {}
for ta in trade_accounts
}
account_order_status = {
ta._id: sorted([o.to_dict() for o in ta.order_status],
key=lambda o: o['order'])
for ta in trade_accounts
}
account_orders = {
ta._id: sorted([o.to_dict() for o in ta.orders],
key=lambda o: o['symbol'])
for ta in trade_accounts
}
return render_template('user/trade_account.html', **locals())
