add sentence_mapper and get existing sentences

Author: thanit456

.gitignore

@@ -6,8 +6,9 @@ __pycache__/
 # C extensions
 *.so
 
-# clips
+# local data 
 clips/
+sentence_mapper.json
 
 # Distribution / packaging
 .Python

src/bot.py

@@ -9,10 +9,19 @@
 import os 
 from random import choice
 import asyncio
+import json 
 
 last = None
 
-bot = commands.Bot(command_prefix=['clip!', 'c!'])
+try:
+    with open('../sentence_mapper.json', 'r') as f:
+        sentence_mapper = json.load(f)
+except:
+    print('Please create sentence_mapper.json')
+    exit()
+
+
+bot = commands.Bot(command_prefix=['clip!', 'c!', '?'])
 
 @bot.event
 async def on_ready():
@@ -26,14 +35,17 @@ async def play(ctx, *args):
         if ctx.author.voice is None: 
             print("Connect to voice, idiot.")
         return 
-    file = '../clips/' + args[0] + '.mp3'
+    if args[0] not in sentence_mapper.keys():
+        print("No this sentence in sentence_mapper")
+        await ctx.reply("Not this sentence in list")
+        return 
+    file = '../clips/' + sentence_mapper[args[0]]
     if not path.exists(file):
-        print(os.getcwd())
         print("No such file (" + args[0] + ".mp3)")
         await ctx.reply(args[0] + " does not exists.")
         return 
     global last 
-    last = args[0]
+    last = sentence_mapper[args[0]]
     user_channel = ctx.author.voice.channel
 
     voice_client = None 
@@ -50,6 +62,11 @@ async def play(ctx, *args):
     except AttributeError:
         pass 
 
+@bot.command(aliases=["ls"])
+async def get_existing_sentences(ctx):
+    global sentence_mapper
+    await ctx.reply("```python\n Existing sentences\n{}```".format('\n'.join([sentence.strip() for sentence in sentence_mapper.keys()])))
+
 @bot.command(aliases=["r"])
 async def random(ctx):
     r = choice(listdir("../clips"))

url2mp3.py

@@ -1,11 +1,20 @@
 import requests
 import re 
+import os 
 from tqdm import tqdm  
+import json 
+
+os.makedirs('clips', exist_ok=True)
 
 raw_sounds = requests.get('https://prophet-button.netlify.app/')
-sounds = re.findall(r'src=\"(.*\.mp3)\"', raw_sounds.text)
+extracted_data = re.findall(r'src=\"(.*\.mp3)\".*\">([^\"<]*)<', raw_sounds.text)
+
+sentence_mapper = dict() 
+for (filename, sentence) in tqdm(extracted_data):
+    sentence_mapper[sentence.strip()] = filename
+    r = requests.get(f'https://prophet-button.netlify.app/sound/{filename}')
+    with open(f'clips/{filename}', 'wb') as f:
+        f.write(r.content)
 
-for sound in tqdm(sounds): 
-    r = requests.get(f'https://prophet-button.netlify.app/sound/{sound}')
-    with open(f'clips/{sound}', 'wb') as f:
-        f.write(r.content)
+with open('sentence_mapper.json', 'w') as f:
+    json.dump(sentence_mapper, f)