def init_data(engine=user_engine): conn = engine.connect() password = '******' passwd_md5 = hashlib.md5(password.encode('utf-8')).hexdigest() conn.execute(User.insert().values(uid='admin', password=passwd_md5, group='admin', name='管理员')) print('- 初始化表:User') conn.execute(Group.insert().values(gid='admin', role='admin', name='管理员')) conn.execute(Group.insert().values(gid='core', role='user', name='核心组')) conn.execute(Group.insert().values(gid='front', role='user', name='前置组')) conn.execute(Group.insert().values(gid='manger', role='user', name='管理组')) conn.execute(Group.insert().values(gid='chanle', role='user', name='渠道组')) conn.execute(Group.insert().values(gid='public', role='user', name='公共')) print('- 初始化表:Group') conn.close()
async def add(app, gid, name): async with app['db'].acquire() as conn: cursor = await conn.execute(Group.select().where(Group.c.gid == gid)) row = await cursor.fetchone() if row: return web.json_response({ 'result': False, 'message': '组%s已存在' % gid }) else: cursor = await conn.execute(Group.insert().values(gid=gid, name=name, role='user')) if cursor.rowcount == 1: return web.json_response({ 'result': True, 'message': '新增组%s成功' % gid }) else: return web.json_response({ 'result': False, 'message': '新增组%s失败' % gid })