|
1
|
|
|
# Copyright 2017 Starbot Discord Project |
|
2
|
|
|
# |
|
3
|
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); |
|
4
|
|
|
# you may not use this file except in compliance with the License. |
|
5
|
|
|
# You may obtain a copy of the License at |
|
6
|
|
|
# |
|
7
|
|
|
# http://www.apache.org/licenses/LICENSE-2.0 |
|
8
|
|
|
# |
|
9
|
|
|
# Unless required by applicable law or agreed to in writing, software |
|
10
|
|
|
# distributed under the License is distributed on an "AS IS" BASIS, |
|
11
|
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
12
|
|
|
# See the License for the specific language governing permissions and |
|
13
|
|
|
# limitations under the License. |
|
14
|
|
|
|
|
15
|
|
|
# Manage message log size. |
|
16
|
|
|
|
|
17
|
|
|
from api import database |
|
18
|
|
|
from api.database.table import Table, TableTypes |
|
19
|
|
|
|
|
20
|
|
|
|
|
21
|
|
|
def message_count_get(server_id): |
|
22
|
|
|
'''Get the current message count.''' |
|
23
|
|
|
database.init() |
|
24
|
|
|
table_message_count = Table('user_messages_{}'.format(server_id), TableTypes.pGlobal) |
|
25
|
|
|
return table_message_count.getLatestID() |
|
26
|
|
|
|
|
27
|
|
|
# Log messages to database. |
|
28
|
|
|
|
|
29
|
|
|
def message_log(msg, server_id): |
|
30
|
|
|
'''Log a message into the database.''' |
|
31
|
|
|
database.init() |
|
32
|
|
|
table_log = Table('user_messages_{}'.format(server_id), TableTypes.pGlobal) |
|
33
|
|
|
Table.insert(table_log, dict(userid=msg.author.id, username=msg.author.name, |
|
34
|
|
|
message=msg.content, serverid=msg.guild.id, |
|
35
|
|
|
servername=msg.guild.name)) |
|
36
|
|
|
|
StarbotDiscord /
Starbot
