def register_user(user_name, user_id, weibo_token, is_bot=None): if is_bot: user = Bot() user.name, user.weibo_id, user.access_token = (user_name, user_id, weibo_token) user.type = 2 db.session.add(user) db.session.commit() else: user = User.query.filter_by(weibo_id=user_id).first() if user: user.access_token, user.name = weibo_token, user_name db.session.commit() else: user = User() user.name, user.weibo_id, user.access_token = (user_name, user_id, weibo_token) db.session.add(user) db.session.commit() return user.generate_token()
#coding: utf-8 from server.base import db from server.config import weibot from server.models import Bot print '=> creating tables' db.create_all() print '=> creating bots' for i in weibot.keys(): b = Bot() b.type = i b.assign(weibot[i]) db.session.add(b) db.session.commit()