-
Notifications
You must be signed in to change notification settings - Fork 0
/
susumutakuan.py
111 lines (85 loc) · 2.86 KB
/
susumutakuan.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
import discord
import asyncio
import os
import signal
import sys
import imp
#Local Imports
import developer
import database
import server
#Set up Client State
CLIENT_TOKEN=os.environ['TOKEN']
#Import config
f = open('takuan.config')
global config
config = imp.load_source('config', '', f)
f.close()
#Create database session
session = database.DBSession()
#Register bot class and functions
server.create_internal_server(session)
developer.register_functions(session)
developer.register_developer_access(session, config.developers)
#Create Discord client
client = discord.Client()
#Handle shutdown gracefully
def sigterm_handler(signum, frame):
print('Logging out...', flush=True)
raise KeyboardInterrupt
print('Shutting down...')
sys.exit(0)
#Register SIGTERM Handler
signal.signal(signal.SIGTERM, sigterm_handler)
@client.event
async def on_ready():
print('Logged in as')
print(client.user.name)
print(client.user.id)
print('------')
for client_server in client.servers:
server.register_server(session, client_server)
#Register events
@client.event
async def on_message(message):
#Attempt to ensure user exists
the_user = database.get_user_by_id(session, message.author.id)
if ( the_user == None ):
server.add_user(session, author)
#Look at DMs for special commands
if message.channel.type == discord.ChannelType.private:
if message.content.startswith('!update'):
await developer.update_git(client, message, session)
elif message.content.startswith('!restart'):
await developer.restart_bot(client, message, session)
elif message.content.startswith('!debug_output'):
await developer.debug_output(client, message, session)
elif message.content.startswith('!debug_error'):
await developer.debug_error(client, message, session)
if message.content.startswith('!test'):
counter = 0
tmp = await client.send_message(message.channel, 'Calculating messages...')
async for log in client.logs_from(message.channel, limit=100):
if log.author == message.author:
counter += 1
await client.edit_message(tmp, 'You have {} messages.'.format(counter))
@client.event
async def on_server_join(client_server):
server.register_server(session, client_server)
@client.event
async def on_member_join(member):
server.add_user(session, member)
@client.event
async def on_member_update(before, after):
server.update_user_roles(session, before, after)
@client.event
async def on_server_role_create(role):
server.add_server_role(session, role)
@client.event
async def on_server_role_delete(role):
server.del_server_role(session, role)
@client.event
async def on_server_role_update(before, after):
server.update_server_role(session, before, after)
#Start event loop
client.run(CLIENT_TOKEN)