discordでbotを作ろう~チートシート~

import discord

client = discord.Client()

@client.event
async def on_ready():
    # 起動時の処理

@client.event
async def on_message(message):
    # テキストチャンネルにメッセージが送信されたときの処理

client.run(client_id)

from gtts import gTTS

def make_by_gtts(text, filepath):
    tts = gTTS(text=text, lang='ja')
    tts.save(filepath + '.mp3')

import subprocess
from pydub import AudioSegment

def make_by_jtalk(text, filepath='voice_message'):
    open_jtalk = ['open_jtalk']
    mech = ['-x', '/usr/local/Cellar/open-jtalk/1.10_1/dic']
    htsvoice = ['-m', '/usr/local/Cellar/open-jtalk/1.10_1/voice/mei/mei_normal.htsvoice']
    speed = ['-r', '1.0']
    outwav = ['-ow', filepath + '.wav']
    cmd = open_jtalk + mech + htsvoice + speed + outwav
    c = subprocess.Popen(cmd, stdin=subprocess.PIPE)
    c.stdin.write(text.encode())
    c.stdin.close()
    c.wait()
    audio_segment = AudioSegment.from_wav(filepath + '.wav')
    os.remove(filepath + '.wav')
    audio_segment.export(filepath + 'mp3', format='mp3')
    aplay = ['afplay', filepath + '.mp3']
    wr = subprocess.Popen(aplay)

def after_play(e):
    print(e)

import discord

client = discord.Client()
client_id = '***'

voice = None
volume = None

@client.event
async def on_ready():
    # 起動時の処理
    print('Bot is wake up.')

@client.event
async def on_message(message):
    # テキストチャンネルにメッセージが送信されたときの処理
    global voice, volume
    if client.user != message.author:
        if message.content.startswith('ログイン'):
            channel = message.author.voice.channel
            voice = await channel.connect()
            source = discord.PCMVolumeTransformer(voice.source)
            volume = source.volume
            message.channel.send('ボイスチャンネルにログインしました')
        elif message.content.startswitch('ログアウト'):
            await voice.disconnect()
            message.channel.send('ボイスチャンネルからログアウトしました')
        elif message.content.startswitch('接続状態'):
            if voice.is_connected():
                message.channel.send('ボイスチャンネルに接続中です')
        elif message.content.startswitch('再生'):
            filepath = 'music.mp3'
            audio_source = discord.FFmpegPCMAudio(filepath)
            voice.play(audio_source, after=lambda e: after_play(e))
            message.channel.send('{}を再生します'.format(filepath))
        elif message.content.startswitch('一時停止'):
            voice.pause()
            message.channel.send('再生を一時停止しました')
        elif message.content.startswitch('再開'):
            voice.resume()
            message.channel.send('再生を再開しました')
        elif message.content.startswitch('停止'):
            voice.stop()
            message.channel.send('再生を停止しました')
        elif message.content.startswitch('音量アップ'):
            volume += 0.1
            message.channel.send('音量を上げました')
        elif message.content.startswitch('音量ダウン'):
            volume -= 0.1
            message.channel.send('音量を下げました')
        elif message.content.startswitch('しゃべって'):
            text = 'こんにちは'
            filepath = 'voice_message'
            if 'gtts' in message.content:
                make_by_gtts(text, filepath)
                audio_source = discord.FFmpegPCMAudio(filepath + '.mp3')
                voice.play(audio_source, after=lambda e: after_play(e))
            elif 'jtalk' in message.content:
                make_by_jtalk(text, filepath)
                audio_source = discord.FFmpegPCMAudio(filepath + '.mp3')
                voice.play(audio_source, after=lambda e: after_play(e))
            message.channel.send(text)

client.run(client_id)