Update: Language

- switching to english and rename file

mcc-flow/__main__.py

-from downloadFilesOfNeuroFlowDevice import download_data
+from file_downloader import download_data
 from merge_records import process_data_files_in_dir
 from gui import show_message, select_folder_in_explorer
 
@@ -13,7 +13,7 @@ def main():
     )
     show_message('Give credentials in terminal!', TITLE)
     path = download_data(path)
-    out_file = input('Type output file format ( edf / fif ) ')
+    out_file = input('Select output file format ( edf / fif ) ')
     process_data_files_in_dir(path, out_file)
     print(f'Processed files are available in {path}')
 

mcc-flow/downloadFilesOfNeuroFlowDevice.py → mcc-flow/file_downloader.py

@@ -5,43 +5,43 @@ from pathlib import Path
 import requests
 
 
-class tcolors:
+class Colors:
     HEADER = '\033[93m'
     OK = '\033[92m'
     FAIL = '\033[91m'
-    ENDC = '\033[0m'
+    END = '\033[0m'
 
 
 def download_data(path='.'):
     path = Path(path)
-    print("\n" + tcolors.HEADER + "================================" + tcolors.ENDC)
-    print(tcolors.HEADER + "NeuroFlow Lab Kutatói Alkalmazás" + tcolors.ENDC)
-    print(tcolors.HEADER + "================================" + tcolors.ENDC + "\n")
+    print(f'\n{Colors.HEADER}================================{Colors.END}')
+    print(f'{Colors.HEADER}NeuroFlow Lab File Downloader{Colors.END}')
+    print(f'{Colors.HEADER}================================{Colors.END}\n')
 
-    email = input("Kutató email címe: ")
-    password = getpass("Kutató jelszava: ")
-    deviceID = input("Eszköz egyedi azonosítója: ")
-    path = path.joinpath(deviceID)
+    email = input("email address: ")
+    password = getpass("password: ")
+    device_id = input("NeuroFlow device ID: ")
+    path = path.joinpath(device_id)
 
     headers = {'Content-Type': 'application/json'}
-    payload = {'email': email, 'password': password, 'deviceID': deviceID}
+    payload = {'email': email, 'password': password, 'deviceID': device_id}
     r = requests.post('https://europe-west3-neuroflowtest.cloudfunctions.net/testListFilesByScript', headers=headers,
                       data=json.dumps(payload))
 
     if r.status_code == 200:
-        listOfFiles = r.json()['files']
-        print(f'\nLe fogok tölteni {len(listOfFiles)} darab fájl-t a {path} könyvtárba...')
+        list_of_files = r.json()['files']
+        print(f'\n{len(list_of_files)} files will be downloaded to {path}')
 
         if not path.exists():
             path.mkdir(parents=True, exist_ok=True)
-            print(tcolors.OK + "Létrehoztam a '" + str(path) + "' könytárat." + tcolors.ENDC)
+            print(f'{Colors.OK}{path} was created{Colors.END}')
 
         print("")
-        for i, file in enumerate(listOfFiles):
-            print("Indítom a '" + file + "' nevű fájl letöltését...")
+        for i, file in enumerate(list_of_files):
+            print(f'Downloading file {file}')
 
             headers = {'Content-Type': 'application/json'}
-            payload = {'email': email, 'password': password, 'deviceID': deviceID, 'fileName': file}
+            payload = {'email': email, 'password': password, 'deviceID': device_id, 'fileName': file}
             r = requests.post('https://europe-west3-neuroflowtest.cloudfunctions.net/testDownloadByScript',
                               headers=headers, data=json.dumps(payload))
 
@@ -50,18 +50,19 @@ def download_data(path='.'):
                 with open(save_path, "wb") as binary_file:
                     binary_file.write(r.content)
 
-                print(tcolors.OK + "LETÖLTVE" + tcolors.ENDC, f'{100. * i / len(listOfFiles):.2f} %')
+                print(f'{Colors.OK}DONE{Colors.END} {100. * (i + 1) / len(list_of_files):.2f} %')
             else:
-                print(tcolors.FAIL + "HIBA" + tcolors.ENDC)
+                print(f'{Colors.FAIL}ERROR{Colors.END}')
 
-        print(
-            "\nA '" + deviceID + "' eszközhöz tartozó fájlok letöltésre kerültek a '" + deviceID + "' nevű könyvtárba.\n")
+        print(f'Files, corresponding to device {device_id} are stored in {device_id} folder.\n')
+        return path
 
     elif r.status_code in (401, 444, 445):
-        print("\n" + tcolors.FAIL + "Hibás email cím vagy jelszó!" + tcolors.ENDC + "\n")
+        print(f'\n{Colors.FAIL}Wrong email or password!{Colors.END}\n')
     elif r.status_code == 446:
-        print("\n" + tcolors.FAIL + "Hibás eszköz azonosító!" + tcolors.ENDC + "\n")
-    return path
+        print(f'\n{Colors.FAIL}Wrong device ID!{Colors.END}\n')
+
+    exit()
 
 
 if __name__ == '__main__':