Exemple #1
0
def replace_imported_bots(json_text):
    Game.query.delete()
    Bot.query.delete()
    Team.query.delete()
    data = json.loads(json_text)
    for team in data['teams']:
        new_team = Team(team['name'])
        db.session.add(new_team)
        for bot in team['bots']:
            new_bot = Bot(new_team, bot['name'], bot['s3_key'])
            db.session.add(new_bot)
    db.session.commit()
Exemple #2
0
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()