ProBotE/discordactions.py at master · Alestrio/ProBotE · 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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
#
# Copyright (c) 2020 by Alexis LEBEL. Released under MIT License
#
import discord
from discord.ext import commands
import logging
import credential
import pronoteactions as pa
import gdrive


class DiscordBot(commands.Bot):


    def __init__(self, *args, **kwargs):
        super().__init__(command_prefix='?',*args, **kwargs)
        logging.basicConfig(level=logging.ERROR)
        logger = logging.getLogger('discord')
        logger.setLevel(logging.ERROR)
        handler = logging.FileHandler(filename='discord.log', encoding='utf-8', mode='w')
        handler.setFormatter(logging.Formatter('%(asctime)s:%(levelname)s:%(name)s: %(message)s'))
        logger.addHandler(handler)

        self.pronote = pa.PronoteActions()
        self.drive = gdrive.GDrive()

        return None

    async def on_ready(self):
        print('Succesfully logged in for Discord as {0.user}'.format(bot))

    async def on_message(self, message):
        probote_channel = self.get_channel(credential.probote_channel)
        if message.content.startswith('pro devoirs'):
            await self.updateChannel(self.pronote.getHomeworks(), self.pronote.getLessons())
        if message.content.startswith('pro help'):
            await self.introduceBot()
        if message.content.startswith('pro dossiers'):
            await self.sendFolderHierarchy()
        if message.content.startswith('pro upload'):
            splittedMsg = message.content.split(' ')
            if len(splittedMsg) > 4:
                await self.sendCommandErrorMsg()
            else:
                att = message.attachments[0]
                await att.save('tempfile')
                title = att.filename
                if len(splittedMsg) == 3:
                    folderId = self.drive.parseFolderArgument(splittedMsg[2], -1)
                    if folderId != -1:
                        self.drive.uploadFile(folderID, title)
                        await self.sendUploadOkMsg()
                    else:
                        await self.sendCommandErrorMsg()
                elif len(splittedMsg) == 2:
                    self.drive.uploadFile(file, self.drive.folderHierarchy['general_folder']['id'])
                    await self.sendUploadOkMsg()
                else:
                    folderId = self.drive.parseFolderArgument(int(splittedMsg[2]), int(splittedMsg[3]))
                    print(folderId)
                    if folderId != -1:
                        self.drive.uploadFile(folderId, title)
                        await self.sendUploadOkMsg()
                    else:
                        await self.sendCommandErrorMsg()
        if message.content.startswith('pro fichiers'):
            splittedMsg = message.content.split(' ')
            if len(splittedMsg) > 4:
                await self.sendCommandErrorMsg()
            else:
                if len(splittedMsg) == 3:
                    folderId = self.drive.parseFolderArgument(int(splittedMsg[2]), -1)
                    if folderId != -1:
                        await probote_channel.send(self.drive.getFileTitles(folderId))
                    else:
                        await self.sendCommandErrorMsg()
                elif len(splittedMsg) == 2:
                    await probote_channel.send(self.drive.getFileTitles(self.drive.folderHierarchy['general_folder']['id']))

                else:
                    folderId = self.drive.parseFolderArgument(int(splittedMsg[2]), int(splittedMsg[3]))
                    print(folderId)
                    if folderId != -1:
                        await probote_channel.send(self.drive.getFileTitles(folderId))
                    else:
                        await self.sendCommandErrorMsg()
        if message.content.startswith('pro mkdir'):
            splittedMsg = message.content.split(' ')
            if len(splittedMsg) == 5:
                parent = self.drive.parseFolderArgument(int(splittedMsg[2]), int(splittedMsg[3]))
                if parent != -1:
                    self.drive.createSubfolder(splittedMsg[4], parent)
                    await probote_channel.send('Dossier créé !')
                else:
                    await self.sendMkdirErrorMsg()
            elif len(splittedMsg) == 4:
                parent = self.drive.parseFolderArgument(int(splittedMsg[2]), -1)
                if parent != -1:
                    self.drive.createSubfolder(splittedMsg[3], parent)
                    await probote_channel.send('Dossier créé !')
                else:
                    await self.sendMkdirErrorMsg()
            elif len(splittedMsg) == 3:
                parent = self.drive.folderHierarchy['general_folder']['id']
                if parent != -1:
                    self.drive.createSubfolder(splittedMsg[2], parent)
                    await probote_channel.send('Dossier créé !')
                else:
                    await self.sendMkdirErrorMsg()
            else:
                await self.sendMkdirErrorMsg()
        return None


    async def updateChannel(self, homeworks, lessons):
        probote_channel = self.get_channel(credential.probote_channel)
        await probote_channel.send("DEVOIRS :")
        for hw in homeworks:
            formattedMessage = hw[0]
            for file in hw[1]:
                formattedMessage +=  file.url + '\n'
            await probote_channel.send(formattedMessage)
        await probote_channel.send("CONTENU DES COURS :")
        for le in lessons:
             formattedMessage = le[0]
             if le[1] != None:
                 for file in le[1]:
                     formattedMessage +=  file.url + '\n'
             print(formattedMessage)
             await probote_channel.send(formattedMessage)

        return None

    async def introduceBot(self):
        probote_channel = self.get_channel(credential.probote_channel)
        await probote_channel.send('Bonjour ! \n' +
        'Je suis un automate qui fait la liaison entre Discord, Pronote, et Google Drive ! \n' +
        'Voici mes commandes : \n' +
        '-pro devoirs : permet de récolter les devoirs sur une période de 15 jours et le contenu des cours du jour, avec les liens des fichiers \n' +
        '-pro help : permet d\'afficher ce message \n' +
        '-pro dossiers : permet d\'afficher la liste des dossiers du drive sous forme d\'arbre \n' +
        '-pro upload : permet de téléverser un fichier vers google drive (usage : pro upload [dossier matière] [sous dossier]) \n' +
        '-pro fichiers : permet de connaître la liste des fichiers d\'un dossier (usage : pro fichiers [dossier matière] [sous dossier]) \n' +
        '-pro mkdir : permet de créer un sous dossier (usage : pro mkdir [dossier matière] [sous-dossier] [nom du dossier à créer]) \n' +
        'Bon courage ! \n' +
        'PS : Mon code source est disponible ici : https://github.com/Alestrio/ProBotE \n' +
        'Version actuelle : ' + credential.version_number)
        return None

    async def sendCommandErrorMsg(self):
        probote_channel = self.get_channel(credential.probote_channel)
        await probote_channel.send('Oups, il semblerait qu\'il y ait un problème avec votre commande ! :/ \n'+
                                    'Voici la structure des dossiers, peut être que vous pourrez corriger votre commande grâce à elle !')
        await self.sendFolderHierarchy()
        return None

    async def sendUploadOkMsg(self):
        probote_channel = self.get_channel(credential.probote_channel)
        await probote_channel.send('Fichier envoyé !')
        return None

    async def sendFolderHierarchy(self):
        probote_channel = self.get_channel(credential.probote_channel)
        self.drive.updateFolderHierarchy()
        fh = self.drive.folderHierarchy
        message = 'Voici l\'arbre des dossiers du Drive : \n'
        i = 0
        for folder in fh['general_folder']['subfolders']:
            message += str(i) + ' - ' + folder['displayName'] + '\n'
            i += 1
            j = 0
            for subf in folder['subfolders']:
                message += '    ' + str(j) + ' - ' + "".join(subf["displayName"]) + '\n'
                j += 1
        await probote_channel.send(message)
        return None

    async def sendMkdirErrorMsg(self):
        probote_channel = self.get_channel(credential.probote_channel)
        message = 'Attention, il y a une erreur dans votre commande (usage : pro mkdir [dossier matière] [sous-dossier] [nom du dossier à créer])'
        await probote_channel.send(message)
        return None

bot = DiscordBot()
bot.run(credential.token)