fixes for samba

python-server/filesystem.py

@@ -2,22 +2,28 @@ import base64
 import os.path
 import audiosegment_wrapper as AudioSegment
 
+audiofiles = []
+FTP_DIR = "/home/ths/"
+
 def save_audio(filename, base64_string):
+    global audiofiles
+    audiofiles += [filename[:-4]]
     decoded = None
     orig_filename = filename[:-4]+"_orig"
-    with open("latestBase64","wb") as f:
-        f.write(base64_string)
     try:
         decoded = base64.b64decode(base64_string)
     except TypeError:
         return b"ERROR_INVALID_ENCODING_64"
-    #if os.path.isfile(filename):
-    #    return b"ERROR_FILE_EXISTS"
-    with open(orig_filename,"wb") as f:
-        f.write(decoded)
+    if os.path.isfile(filename):
+        return b"ERROR_FILE_EXISTS"
+    write_file(orig_filename, decoded)
+
     seg = AudioSegment.from_file(orig_filename)
     seg = seg.resample(sample_rate_Hz=32000, sample_width=2, channels=1)
+
     seg.export(filename,format="wav")
+    seg.export(FTP_DIR + filename,format="wav")
+    
     return b"SUCCESS"
 
 def save_audio_chain(file_str_tupels):
@@ -41,20 +47,31 @@ def save_audio_chain(file_str_tupels):
     else:
         completeAudio = completeAudio.resample(sample_rate_Hz=32000, sample_width=2, channels=1)
         completeAudio.export(file_str_tupels[0][0],format="wav")
+        completeAudio.export(FTP_DIR + file_str_tupels[0][0],format="wav")
         return b"SUCCESS"
 
 def save_transcript(filename, transcript):
+    global audiofiles
+    try:
+      audiofiles.remove(filename[:4])
+    except:
+      print("Audiofile not in working list")
     if os.path.isfile(filename):
         pass
     with open(filename + "_transcript","w") as f:
         f.write(transcript)
+    with open(FTP_DIR +  filename + "_transcript","w") as f:
+        f.write(transcript)
 
 def get_transcript(filename):
+    global audiofiles
     try:
       with open("data/" + filename + ".wav_transcript","r") as f:
         return f.read()
     except FileNotFoundError:
-        return "File not found on Server!"
+        if filename in audiofiles:
+            return "ERROR_FILE_STILL_IN_WORK"
+        return "ERROR_FILE_NOT_AVAILIABLE"
 
 def filelist():
     return ""
@@ -64,3 +81,10 @@ def fileinfo(filename):
 
 def copy_to_output(filename):
     return ""
+
+def write_file(filename, data):
+    FTP_DIR = "/home/ths/"
+    with open(FTP_DIR + filename,"wb") as f:
+        f.write(data)
+    with open(filename,"wb") as f:
+        f.write(data)