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