CamelBot/Core.py at master · zoopr/CamelBot · GitHub

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
112
113
114
115
116
117
118
119
120
121
122
123
124
import discord
import asyncio
import os
import sys
from random import randint

import Constants
import Commands
import Servers
import Utilities.Archive
import Utilities.Reminder
import RandomImage
from MediaPoster import MediaPoster
from Utilities import Utilities
from Logger import Logger
#from cleverbot import Cleverbot
# import clever
# from chatterbot import ChatBot


client = discord.Client()
media_poster = MediaPoster()
utilities = Utilities.Utilities()
logger = Logger()


@client.event
async def on_ready():
    print('Logged in as:')
    print(client.user.name)
    print(client.user.id)
    print('------')

    '''server_names = Servers.server_name_list(client)
    print('\nConnected servers (' + str(len(server_names)) + '): ')
    for server in server_names:
        print(str(server).encode('utf-8'))
    print('\n------')'''


@client.event
async def on_message(message):
    #Archiving contents of specific servers
    await Utilities.Archive.channel_archive(message, client)

    '''usrmsg = Utilities.Reminder.reminders_check_timer_event(client)
    if usrmsg[0] == True:
        print('Sending reminder message to: ' + str(usermsg[1]))
        await client.send_message(usermsg[1], usermsg[2])'''

    #ignoring self
    if message.author.id == client.user.id or message.author.id in Constants.BANNED_USERS:
        return

    #If server has random image posting enabled, a random image will be posted
    try:
        await RandomImage.post_random_image(message, client)
    except:
        pass

    msg_content = message.content.lower()
    if not msg_content.startswith('.'):
        return

    if msg_content == '.invite':
        await invite(message, client)

    if msg_content == '.help' or msg_content == '.info':
        await Commands.post_commands(message, client)

    #General utility functionality
    if await utilities.execute_command(message, client):
        return

    #Simple one liner message posting commands not handled by MediaPoster
    if await utilities.one_liner_commands(message, client):
        return

    #Searches in MediaURLs directory for a .txt file with the corresponding name
    if await media_poster.find(msg_content):
        if msg_content[1:].strip(' ') in Constants.NSFW_COMMANDS or message.channel.is_private or message.server.id in Constants.NSFW_WHITELISTED_SERVER_IDS:
            if await media_poster.nsfw_allowed(message):
                await client.send_message(message.channel, await media_poster.get_url(msg_content))
                return
            else:
                await client.send_message(message.channel, 'Requires channel topic modification. See .help for info.')
                return

        if msg_content[1:].strip(' ') == 'jew':
            if await media_poster.jew_allowed(message):
                await client.send_message(message.channel, await media_poster.get_url(msg_content))
                return
            else:
                await client.send_message(message.channel, 'Command has been banned on this server.')
                return

        else:
            await client.send_message(message.channel, await media_poster.get_url(msg_content))
            return


async def invite(message, client):
    if message.channel.is_private:
        await client.send_message(message.channel, 'Hi! Click the following link and select the server you wish to add me to from the drop-down box! :camel:')
        await client.send_message(message.channel, 'https://discordapp.com/oauth2/authorize?client_id=190756702226874368&scope=bot&permissions=0')
        await client.send_message(message.channel, 'NOTE: This will only work if you\'re a server owner/manager.')
    else:
        await client.send_message(message.channel, 'Sending ' + message.author.name + ' an invite URL.')
        await client.send_message(message.author, 'Hi! Click the following link and select the server you wish to add me to from the drop-down box! :camel:')
        await client.send_message(message.author, 'https://discordapp.com/oauth2/authorize?client_id=190756702226874368&scope=bot&permissions=0')
        await client.send_message(message.author, 'NOTE: This will only work if you\'re a server owner/manager.')
    return


# async def cleverbot_chat(message, client):
#     for mention in message.mentions:
#         if mention.id == client.user.id and message.author.id != client.user.id:
#             cb_resp = cb.ask(message.content.strip(' ').strip(Constants.USERNAME))
#             print('Cleverbot response: {}'.format(cb_resp))
#             await client.send_message(message.channel, cb_resp)

while(1):
    client.run(Constants.TOKEN)
    # client.run(Constants.TESTING_TOKEN)