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