Browse Source

Correct sha1 & add wipe cache function

Unknown 8 years ago
parent
commit
25ea66ac1e
1 changed files with 14 additions and 7 deletions
  1. 14 7
      tts/voxygen/voxygen.py

+ 14 - 7
tts/voxygen/voxygen.py

@@ -1,8 +1,10 @@
-import sha
+import hashlib
 import os
+import shutil
 
 import pygame
 import requests
+import logging
 
 VOXYGEN_LANGUAGES = {
     "fr" : {"electra":"Electra","emma":"Emma","becool":"Becool","agnes":"Agnes","loic":"Loic","fabienne":"Fabienne","helene":"Helene","marion":"Marion","matteo":"Matteo","melodine":"Melodine","mendoo":"Mendoo","michel":"Michel","moussa":"Moussa","philippe":"Philippe","sorciere":"Sorciere"},
@@ -14,14 +16,15 @@ VOXYGEN_LANGUAGES = {
     "it" : {"sonia":"Sonia"}
 }
 
+CACHE_PATH = "/tmp/jarvis/tts/voxygen/"
+
 def say(words=None, voice=None, language=None, cache=None):
-    path = "/tmp/jarvis/tts/voxygen/"
-    if not os.path.exists(path):
-        os.makedirs(path)
+    if not os.path.exists(CACHE_PATH):
+        os.makedirs(CACHE_PATH)
 
-    sha1 = sha.new(words).hexdigest()
+    sha1 = hashlib.sha1(words).hexdigest()
 
-    tempfile = path+voice+"."+sha1+".tts"
+    tempfile = CACHE_PATH+voice+"."+sha1+".tts"
 
     get_audio(voice,words,tempfile,cache)
 
@@ -66,4 +69,8 @@ def get_voice(voice=None, language=None):
             return VOXYGEN_LANGUAGES[language][voice]
 
     logging.debug("Cannot find language maching language: %s voice: %s",language,voice)
-    return ""
+    return ""
+
+def wipe_cache():
+    shutil.rmtree(CACHE_PATH)
+