コード例 #1
0
def serverInfo(guildId, channelId):
    # Auth and session
    token = session.get('oauth2_token')
    (user, usertoken) = get_user_cache(token, cache)
    check_auth(user, usertoken)

    channelId = channelId if (channelId != None) else guildId

    mysqldb = MySQL(current_app.config['DB_HOST'], current_app.config['DB_USER'], current_app.config['DB_PASS'])
    mysqldb.connect()
    messages = mysqldb.getMessages(channelId)
    nbMessages = len(messages)

    # API calls
    headers = {'authorization': usertoken}
    channel = requests.get(API_BASE_URL + '/channels/' + channelId, headers=headers).json()
    guild    = requests.get(API_BASE_URL + '/guilds/' + guildId, headers=headers).json()
    channels = requests.get(API_BASE_URL + '/guilds/' + guildId + '/channels', headers=headers).json()
    cache['guilds'][guildId] = guild

    guild = cache['guilds'][channel['guild_id']]
    data = {
        'guildId': guildId,
        'channelId': channelId,
        'channels': channels
    }

    channelinfos = {
        'nbmessages' : nbMessages,
        'nbchannels' : 0,
        'firstmessage' : mysqldb.getFirstMessage(channelId),
        'lastmessage' : mysqldb.getLastMessage(channelId)
    }

    # Compute things

    # Render
    return render_template("layout.html", contentTemplate="server-info.html", user=user, server=guild, data=data, channelinfos=channelinfos)
コード例 #2
0
def compute(guildId):
    # Auth and session
    token = session.get('oauth2_token')
    (user, usertoken) = get_user_cache(token, cache)

    # Logic
    way = request.args.get('time')
    channelId = request.args.get('channelId')
    mysqldb = MySQL(current_app.config['DB_HOST'], current_app.config['DB_USER'], current_app.config['DB_PASS'])
    mysqldb.connect()
    params = {}

    nbMessagesStr = request.args.get('number')
    if nbMessagesStr is not None:
        nbMessagesLeft = int(request.args.get('number'))
    else:
        nbMessagesLeft = 100
    if nbMessagesLeft < 0:
        nbMessagesLeft = 100
    if nbMessagesLeft > 1000:
        nbMessagesLeft = 1000

    # API calls
    headers = {'authorization': usertoken}

    first = True

    while nbMessagesLeft > 0:
        if not first:
            time.sleep(1)
        first = False
        nbToTake = nbMessagesLeft
        if nbMessagesLeft > 100:
            nbToTake = 100

        lastMessageId = mysqldb.getLastMessage(channelId)
        if lastMessageId is None:
            channel = requests.get(API_BASE_URL + '/channels/' + channelId, headers=headers).json()
            params = {'around': int(channel['last_message_id']), 'limit': nbToTake}
        elif way == 'after':
            lastMessageId = mysqldb.getLastMessage(channelId)
            params = {'after': int(lastMessageId['id']), 'limit': nbToTake}
        elif way == 'before':
            firstMessageId = mysqldb.getFirstMessage(channelId)
            params = {'before': int(firstMessageId['id']), 'limit': nbToTake}
        else:
            abort(412, "Incorrect 'time' parameter.")
            return

        messages = requests.get(API_BASE_URL + '/channels/' + channelId + '/messages', headers=headers, params=params).json()

        def take8(l): return [l[0], l[1][:8]]

        toInsert = [(str(message['id']),
                     str(guildId),
                     str(channelId),
                     str(message['author']['id']),
                     str(message['content']),
                     str(' '.join(take8(message['timestamp'].split('T')))),
                     str(message['author']['username']),
                     message['author']['discriminator'],
                     str(message['author']['avatar'])) for message in messages]
        # DB Fill
        returnValue = mysqldb.insertMessages(toInsert)

        nbMessagesLeft -= nbToTake

    # Render
    json = {
        'result': 'ok',
        'message': 'Successfully inserted messages'
    }

    return jsonify(json)