Пример #1
0
    def post(self):
        parser = reqparse.RequestParser()
        parser.add_argument('id', required=True)
        parser.add_argument('pw', required=True)
        parser.add_argument('level', required=True)
        args = parser.parse_args()

        account = Account.get_account(args['id'])
        if account:
            return {'msg': "id is already existed"}, HTTPStatus.CONFLICT

        if args['level'] != "U":
            return {'msg': "unknown account level"}, HTTPStatus.BAD_REQUEST

        account = Account()
        account.id = args['id']
        account.pw = sha256.hash(args['pw'])
        account.level = args['level']
        account.active = "N"

        try:
            db.session.add(account)
            db.session.commit()

            return account, HTTPStatus.CREATED
        except Exception as e:
            return {
                'msg': "fail to create account {}".format(e)
            }, HTTPStatus.INTERNAL_SERVER_ERROR
Пример #2
0
from run import db
from models.account import Account

import sqlalchemy
import os, uuid, base62
from passlib.hash import pbkdf2_sha256 as sha256

# add Account
account = Account()
account.id = "user"
account.pw = sha256.hash("user")
account.level = "U"
account.active = "Y"

db.session.add(account)
db.session.commit()
Пример #3
0
from run import db

from models.token import RevokedToken
from models.account import Account

import sqlalchemy
import os, uuid, base62
from passlib.hash import pbkdf2_sha256 as sha256

engine = sqlalchemy.create_engine(run.DB_ENGINE_URI)

def create_table(model):
    try:
        model.__table__.drop(engine)
        model.__table__.create(engine)
    except:
        model.__table__.create(engine)

create_table(RevokedToken)
create_table(Account)

# add Admin
admin = Account()
admin.id = "admin"
admin.pw = sha256.hash("admin")
admin.level = "A"
admin.active = "Y"

db.session.add(admin)
db.session.commit()