Esempio n. 1
0
def add_user(mobile, username, password, is_admin=None):
    """ 添加用户 """
    try:
        u = User({'mobile': mobile,
                  'username': username,
                  'password': password})
        if is_admin is not None:
            u._is_admin = is_admin
        print(u._data)
        u.upsert()
        log.info('创建/更新用户{}成功!'.format(u._id))
        return True
    except Exception as e:
        log.exception(str(e))
        return False
Esempio n. 2
0
def add_account(user, exchange, login_name, login_password,
                money_password='', bank_password=''):
    """ 添加账号 """
    try:
        u = User.query_one({'_id': user})
        if not u:
            log.info('找不到用户{}, 无法创建账号'.format(user))
            return False
        if exchange not in exchanges:
            log.info('找不到交易所{}, 无法创建账号'.format(exchange))
            return False
        a = Account({
            'user': user,
            'exchange': exchange,
            'login_name': login_name,
            'login_password': login_password,
            'money_password': money_password,
            'bank_password': bank_password,
        })
        a.upsert()
        log.info('创建/更新账号{}成功'.format(a._id))
        return True
    except Exception as e:
        log.exception(str(e))
        return False
Esempio n. 3
0
File: app.py Progetto: sopnic/ybk
def user_login():
    mobile = request.form.get('mobile')
    password = request.form.get('password')
    user = User.query_one({'_id': mobile})
    if user and user.password == password:
        login_user(user)

    return redirect(request.args.get('next', '/'))
Esempio n. 4
0
def update_user(user):
    """ 更新用户的账号信息 """
    accounts = list(Account.query({'user': user}))
    u = User.query_one({'_id': user})
    for key, value in accounts_summary(accounts).items():
        setattr(u, key, value)
    u.num_accounts = len(accounts)
    u.num_exchanges = len(set(a.exchange for a in accounts))
    u.upsert()
Esempio n. 5
0
def update_users(users=None):
    for user in users or [u._id for u in User.query()]:
        update_user(user)
Esempio n. 6
0
File: app.py Progetto: sopnic/ybk
def load_user(user_id):
    return User.query_one({'_id': user_id})
Esempio n. 7
0
File: app.py Progetto: sopnic/ybk
def delete_user():
    user = request.form.get('user')
    User.delete_one({'_id': user})
    return jsonify(status=200)
Esempio n. 8
0
File: app.py Progetto: sopnic/ybk
def admin_account():
    users = list(User.query())
    accounts = list(Account.query())
    locals()['exchanges'] = exchanges
    return render_template('admin/account.html', **locals())
Esempio n. 9
0
File: app.py Progetto: sopnic/ybk
def admin_user():
    users = list(User.query())
    return render_template('admin/user.html', **locals())
Esempio n. 10
0
File: app.py Progetto: sopnic/ybk
def admin():
    num_users = User.count()
    num_accounts = Account.count()
    return render_template('admin/index.html', **locals())