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
})
