Update gofilecli.py

.github/workflows/release.yml

@@ -41,20 +41,18 @@ jobs:
       - name: Install Dependencies
         run: |
           pip install -r requirements.txt
+          pip install nuitka
 
       - name: Build Executable with Nuitka
-        uses: Nuitka/Nuitka-Action@main
+        run: |
-        with:
+          python -m nuitka --onefile --assume-yes-for-downloads --output-dir=dist gofilecli.py --include-data-files=./assets/sounds/Blow_edited.wav=assets/sounds/Blow_edited.wav
-          nuitka-version: main
-          script-name: gofilecli.py
-          onefile: true
 
       - name: Upload Artifact[win-x64]
         uses: actions/upload-artifact@v3.1.3
         with:
           name: GoFileCLI_win-x64
           path: |
-            build/gofilecli.exe
+            dist/gofilecli.exe
   
   build-linux-x64:
     runs-on: ubuntu-latest
@@ -74,24 +72,23 @@ jobs:
       
       - name: Install Dependencies
         run: |
+          sudo apt-get install -y libasound2-dev
           pip install -r requirements.txt
+          pip install nuitka
 
       - name: Build Executable with Nuitka
-        uses: Nuitka/Nuitka-Action@main
+        run: |
-        with:
+          python -m nuitka --onefile --assume-yes-for-downloads --output-dir=dist gofilecli.py --include-data-files=./assets/sounds/Blow_edited.wav=assets/sounds/Blow_edited.wav
-          nuitka-version: main
-          script-name: gofilecli.py
-          onefile: true
         
       - name: Rename Executable
-        run: mv build/gofilecli.bin build/gofilecli
+        run: mv dist/gofilecli.bin dist/gofilecli
 
       - name: Upload Artifact[linux-x64]
         uses: actions/upload-artifact@v3.1.3
         with:
           name: GoFileCLI_linux-x64
           path: |
-            build/gofilecli
+            dist/gofilecli
     
   build-macos:
     runs-on: macos-latest
@@ -116,8 +113,8 @@ jobs:
 
       - name: Build Executable with Nuitka
         run: |
-          python -m nuitka --onefile --assume-yes-for-downloads --output-dir=dist --macos-target-arch=arm64 gofilecli.py
+          python -m nuitka --onefile --assume-yes-for-downloads --output-dir=dist --macos-target-arch=arm64 gofilecli.py --include-data-files=./assets/sounds/Blow_edited.wav=assets/sounds/Blow_edited.wav
-          python -m nuitka --onefile --assume-yes-for-downloads --output-dir=dist_x86_64 --macos-target-arch=x86_64 gofilecli.py
+          python -m nuitka --onefile --assume-yes-for-downloads --output-dir=dist_x86_64 --macos-target-arch=x86_64 gofilecli.py --include-data-files=./assets/sounds/Blow_edited.wav=assets/sounds/Blow_edited.wav
       
       - name: Rename Executable
         run: |

.gitignore

@@ -133,3 +133,7 @@ encrypt.py
 /temp
 .copy
 .github/workflows/build_latest_ex.yml
+assets/sounds/Blow.aiff
+assets/sounds/Blow.mp3
+assets/sounds/Blow.wav
+/VP8SWj

README.md

@@ -1,50 +1,60 @@
 # GoFileCLI :
-......
+Upload or download a file to GoFile.
 
-# Installation :
+# Install :
+
+- Linux/MacOS :
+```bash
+curl -fsSL https://raw.githubusercontent.com/NohamR/GoFileCLI/main/scripts/install.sh | sudo bash
+```
+
+- Windows :
+Grab latest release from [releases](https://github.com/NohamR/GoFileCLI/releases/latest)
+
+- Build on your own system :
+```bash
+git clone https://github.com/NohamR/GoFileCLI.git
+apt install ccache patchelf
+pip install nuitka
+pip install -r requirements.txt
+python -m nuitka --onefile --assume-yes-for-downloads --output-dir=dist --static-libpython=no gofilecli.py 
+# dist/gofilecli.bin
+```
 
 ## Set env variables :
 
 Get API token from https://gofile.io/myProfile.
 Copy folderId from a folder you own.
 
-Windows :
+- Set up a .env based on [.env.example](.env.example)
-```bash
-setx GOPLOAD_TOKEN "XXXXXXX"
-setx GOPLOAD_PRIVATE_PARENT_ID "UUID"
-setx GOPLOAD_ACCOUNT_ID "UUID"
-```
 
-(Reluch Command Prompt to take effect)
 
-Linux/MacOS :
+- Linux/MacOS :
 ```bash
 export GOPLOAD_TOKEN="XXXXXXX"
 export GOPLOAD_PRIVATE_PARENT_ID="UUID"
 export GOPLOAD_ACCOUNT_ID="UUID"
 ```
 
-# Usage :
+- Windows :
-Upload a file :
 ```bash
-gofilecli -i 'file.txt'
+setx GOPLOAD_TOKEN "XXXXXXX"
+setx GOPLOAD_PRIVATE_PARENT_ID "UUID"
+setx GOPLOAD_ACCOUNT_ID "UUID"
 ```
+(Reluch Command Prompt to take effect)
 
-# Compile on your own system :
+# Usage :
 ```bash
-git clone https://github.com/NohamR/GoFileCLI.git
+gofilecli -i 'file.txt' # to upload a file
-apt install ccache patchelf
+gofilecli -f folder/ # to upload a folder
-pip install nuitka
+gofilecli -s # to get stats of your account
-pip install -r requirements.txt
+gofilecli -d https://gofile.io/d/XXXXX # to download a folder
-python -m nuitka --standalone --assume-yes-for-downloads --output-dir=dist --static-libpython=no gofilecli.py 
-# dist/gofilecli.dist/gofilecli.bin
 ```
 
 # To do :
-- acc stats
+- KeyboardInterrupt + Lost connexion
 - error-rateLimit
 - env via CLI
 - finish README.md
-- download
+- chiffrer & dechiffrer uploads
-- build + release
-- chiffrer et dechiffrer les uploads

gofilecli.py

@@ -6,10 +6,69 @@ import argparse
 import os
 import logging
 import sys
-from dotenv import load_dotenv, set_key
 import platform
 import subprocess
 import sys
+import simpleaudio as sa
+from tqdm import tqdm
+import json
+from dotenv import load_dotenv
+
+# Suppress ALSA warnings
+os.environ['SDL_AUDIODRIVER'] = 'dummy'
+os.environ['AUDIODEV'] = 'null'
+
+
+def reqst(url, method ,headers=None, data=None, files=None, params=None, json=None, logger=None):
+    try:
+        if method == "get":
+            response = requests.get(url, headers=headers, data=data, params=params, json=json)
+        elif method == "post":
+            response = requests.post(url, headers=headers, data=data, files=files, params=params, json=json)
+        elif method == "put":
+            response = requests.put(url, headers=headers, data=data, files=files, params=params, json=json)
+        elif method == "delete":
+            response = requests.delete(url, headers=headers, data=data, files=files, params=params, json=json)
+        logger.debug(f"Request to {url} with method {method} returned status code {response.status_code}")
+        json_response = response.json()  # If response content is not JSON, this will raise a ValueError
+        return json_response
+    except requests.exceptions.HTTPError as http_err:
+        logger.debug(f"Response: {response.text}")
+        logger.error(f"HTTP error occurred: {http_err}")  # Handles HTTP errors (e.g., 404, 500)
+        sys.exit()
+    except requests.exceptions.ConnectionError as conn_err:
+        logger.debug(f"Response: {response.text}")
+        logger.error(f"Connection error occurred: {conn_err}")  # Handles network-related errors
+        sys.exit()
+    except requests.exceptions.Timeout as timeout_err:
+        logger.debug(f"Response: {response.text}")
+        logger.error(f"Timeout error occurred: {timeout_err}")  # Handles request timeouts
+        sys.exit()
+    except requests.exceptions.RequestException as req_err:
+        logger.debug(f"Response: {response.text}")
+        logger.error(f"An error occurred: {req_err}")  # Catches any other requests-related errors
+        sys.exit()
+    except ValueError as json_err:
+        logger.debug(f"Response: {response.text}")
+        logger.error(f"JSON decode error: {json_err}")  # Handles issues with JSON decoding
+        sys.exit()
+
+
+def load_file(file_name: str) -> str:
+    """
+    Get the correct path to a file, regardless of development or compiled mode.
+    """
+    return os.path.join(os.path.dirname(__file__), file_name)
+
+
+def play_sound(logger):
+    try:
+        sound_path = load_file("assets/sounds/Blow_edited.wav")
+        wave_obj = sa.WaveObject.from_wave_file(sound_path)
+        play_obj = wave_obj.play()
+        play_obj.wait_done()
+    except Exception as e:
+        logger.debug(f"An error occurred while playing sound: {e}")
 
 
 def set_env_var_unix(name, value, shell="bash"):
@@ -37,12 +96,15 @@ def format_file_size(file=None, num_bytes=None):
         file_size_bytes = num_bytes
     else:
         file_size_bytes = os.path.getsize(file)
+    size_in_kb = file_size_bytes / 1024
     size_in_mb = file_size_bytes / (1024 * 1024)
     size_in_gb = file_size_bytes / (1024 * 1024 * 1024)
     if size_in_gb >= 1:
         return size_in_gb, "GB"
-    else:
+    elif size_in_mb >= 1:
         return size_in_mb, "MB"
+    else:
+        return size_in_kb, "KB"
 
 
 def file_size(file=None, num_bytes=None):
@@ -53,9 +115,10 @@ def file_size(file=None, num_bytes=None):
 def calculate_upload_speed(file, start_time):
     elapsed_time_seconds = time.time() - start_time
     file_size, size_unit = format_file_size(file)
+
     if size_unit == "GB":
-        average_speed = file_size / elapsed_time_seconds  # GB/s
+        average_speed = (file_size * 1024) / elapsed_time_seconds  # Convert GB to MB and calculate MB/s
-        speed_unit = "GB/s"
+        speed_unit = "MB/s"
     else:
         average_speed = file_size / elapsed_time_seconds  # MB/s
         speed_unit = "MB/s"
@@ -80,9 +143,16 @@ def get_file_paths(folderPath):
     return filePaths
 
 
+def check_folderPath(path):
+    if not os.path.exists(path):
+        os.makedirs(path)
+    return path
+
+
 def getservers(logger):
     servers = []
-    response = requests.get("https://api.gofile.io/servers").json()
+    # response = requests.get("https://api.gofile.io/servers").json()
+    response = reqst("https://api.gofile.io/servers", logger=logger, method="get")
     if response["status"] == "ok":
         for server in response["data"]["servers"]:
             servers.append(server["name"])
@@ -93,22 +163,65 @@ def getservers(logger):
         return None
     
 
+def ping_server(url, logger, num_requests=4, delay=0.1):
+    response_times = []
+    for _ in range(num_requests):
+        try:
+            start_time = time.time()
+            response = requests.head(url)
+            response_time = time.time() - start_time
+            if response.status_code == 200:
+                response_times.append(response_time)
+        except requests.exceptions.RequestException as e:
+            logger.error(f"Error: {e}")
+        time.sleep(delay)
+    if response_times:
+        avg_response = sum(response_times) / len(response_times)
+        return avg_response
+    else:
+        return float('inf')
+    
+
+def test_servers(servers, logger):
+    best_server = None
+    best_time = float('inf')
+    for server in servers:
+        logger.debug(f"Pinging {server}...")
+        url = f"https://{server}.gofile.io"
+        avg_time = ping_server(url, logger)
+        logger.debug(f"Average response time for {server}: {avg_time:.2f} ms")
+        if avg_time < best_time:
+            best_time = avg_time
+            best_server = server
+    if best_server:
+        logger.debug(f"\nThe best server is {best_server} with an average response time of {best_time:.2f} ms.")
+    else:
+        logger.error("All pings failed.")
+    return best_server
+
+
 def get_stats(logger):
     headers = {'authorization': f'Bearer {TOKEN}',}
-    data = requests.get(f'https://api.gofile.io/accounts/{ACCOUNT_ID}', headers=headers).json()
+    # data = requests.get(f'https://api.gofile.io/accounts/{ACCOUNT_ID}', headers=headers).json()
-    stats = data["data"]["statsCurrent"]
+    data = reqst(f'https://api.gofile.io/accounts/{ACCOUNT_ID}', headers=headers, logger=logger, method="get")
-    logger.info("Account stats:")
+    if data["status"] == "ok":
-    logger.info(f"Total files: {stats['fileCount']}")
+        stats = data["data"]["statsCurrent"]
-    logger.info(f"Total folders: {stats['folderCount']}")
+        logger.info("Account stats:")
-    size = file_size(num_bytes=stats['storage'])
+        logger.info(f"Total files: {stats['fileCount']}")
-    logger.info(f"Total size: {size}")
+        logger.info(f"Total folders: {stats['folderCount']}")
-    traffic = file_size(num_bytes=stats['trafficWebDownloaded'])
+        size = file_size(num_bytes=stats['storage'])
-    logger.info(f"Total traffic: {traffic}")
+        logger.info(f"Total size: {size}")
+        traffic = file_size(num_bytes=stats['trafficWebDownloaded'])
+        logger.info(f"Total traffic: {traffic}")
+    else:
+        logger.error(f"{data}")
+        return None
 
 
 def get_rootfolder(logger):
     headers = {'authorization': f'Bearer {TOKEN}',}
-    data = requests.get(f'https://api.gofile.io/accounts/{ACCOUNT_ID}', headers=headers).json()
+    # data = requests.get(f'https://api.gofile.io/accounts/{ACCOUNT_ID}', headers=headers).json()
+    data = reqst(f'https://api.gofile.io/accounts/{ACCOUNT_ID}', headers=headers, logger=logger, method="get")
     root_folder = data.get("data", {}).get("rootFolder", {})
     if root_folder:
         return root_folder
@@ -120,7 +233,8 @@ def get_rootfolder(logger):
 def get_code(folderId, logger):
     headers = {'authorization': f'Bearer {TOKEN}',}
     params = (('wt', '4fd6sg89d7s6'),('cache', 'false'),)
-    data = requests.get(f'https://api.gofile.io/contents/{folderId}', headers=headers, params=params).json()
+    # data = requests.get(f'https://api.gofile.io/contents/{folderId}', headers=headers, params=params).json()
+    data = reqst(f'https://api.gofile.io/contents/{folderId}', headers=headers, params=params, logger=logger, method="get")
     code = data.get("data", {}).get("code", {})
     if code:
         return code
@@ -130,10 +244,10 @@ def get_code(folderId, logger):
 
 
 def get_children(id, logger):
-    # if len(id) == 36:
     headers = {'authorization': f'Bearer {TOKEN}',}
     params = (('wt', '4fd6sg89d7s6'),('cache', 'false'),)
-    data = requests.get(f'https://api.gofile.io/contents/{id}', headers=headers, params=params).json()
+    # data = requests.get(f'https://api.gofile.io/contents/{id}', headers=headers, params=params).json()
+    data = reqst(f'https://api.gofile.io/contents/{id}', headers=headers, params=params, logger=logger, method="get")
     children = data.get("data", {}).get("children", {})
     if children:
         return children
@@ -153,11 +267,12 @@ def createfolder(parentFolderId, folderName, logger):
         data = {"parentFolderId": parentFolderId, "folderName": folderName}
     else:
         data = {"parentFolderId": parentFolderId}
-    response = requests.post("https://api.gofile.io/contents/createFolder", headers=headers, json=data).json()
+    # response = requests.post("https://api.gofile.io/contents/createFolder", headers=headers, json=data).json()
+    response = reqst("https://api.gofile.io/contents/createFolder", headers=headers, json=data, logger=logger, method="post")
     if response["status"] == "ok":
         name = response["data"]["name"]
         code = response["data"]["code"]
-        folderId = response["data"]["folderId"]
+        folderId = response["data"]["id"]
         logger.debug(f"""Folder {name} created with code {code} and folderId {folderId}""")
         return folderId
     else:
@@ -165,15 +280,57 @@ def createfolder(parentFolderId, folderName, logger):
         return None
 
 
+def read_in_chunks(file_object, CHUNK_SIZE):
+    while True:
+        data = file_object.read(CHUNK_SIZE)
+        if not data:
+            break
+        yield data
+
+
 def uploadfile(serverName, folderId, filePath, logger):
-    headers = {"Authorization": f"Bearer {TOKEN}"}
+    # reference : https://api.video/blog/tutorials/upload-a-big-video-file-using-python/
-    files = {
-        'file': (filePath, open(filePath, 'rb')),
-        'folderId': (None, folderId),
-    }
     start_time = time.time()
-    response = requests.post(f"https://{serverName}.gofile.io/contents/uploadfile", headers=headers, files=files).json()
+    # CHUNK_SIZE = 6000000
+    # content_size = os.stat(filePath).st_size
+    # f = open(filePath, "rb")
+    # index = 0
+    # offset = 0
+    # headers = {"Authorization": f"Bearer {TOKEN}", 'content-type': 'multipart/form-data',}
+    # with tqdm(total=content_size, unit='B', unit_scale=True, desc='Uploading', leave=False) as progress_bar:
+    #     for chunk in read_in_chunks(f, CHUNK_SIZE):
+    #         offset = index + len(chunk)
+    #         headers['Content-Range'] = 'bytes %s-%s/%s' % (index, offset - 1, content_size)
+    #         index = offset
+    #         try:
+    #             # file = {"file": chunk, 'folderId': (None, folderId)}
+    #             # response = requests.post(f"https://{serverName}.gofile.io/contents/uploadfile", files=file, headers=headers)
+    #             # files = {"file": chunk}
+    #             # data = {"folderId": folderId}
+    #             files = {"file": chunk,}
+    #             response = requests.post(f"https://{serverName}.gofile.io/contents/uploadfile",files=files,headers=headers)
+    #             logger.debug("r: %s, Content-Range: %s" % (response, headers['Content-Range']))
+    #             progress_bar.update(len(chunk))
+    #         except Exception as e:
+    #             logger.error(f"Error: {e}")
+    # logger.debug(f"{response.text}")
+    # response = response.json()
+    command = [
+        "curl",
+        "-X", "POST",
+        f"https://{serverName}.gofile.io/contents/uploadfile",
+        "-H", f"Authorization: Bearer {TOKEN}",
+        "-F", f"file=@{filePath}",
+        "-F", f"folderId={folderId}",
+    ]
+    response = subprocess.run(command, capture_output=True, text=True)
+    try:
+        response_json = json.loads(response.stdout)
+    except json.JSONDecodeError:
+        logger.error("Failed to parse response as JSON.")
+        return None
     speed, elapsed_time = calculate_upload_speed(filePath, start_time)
+    response = response_json
     if response["status"] == "ok":
         logger.debug(response)
         name = response["data"]["name"]
@@ -186,10 +343,11 @@ def uploadfile(serverName, folderId, filePath, logger):
         return None
 
 
-def actionFolder(folderId, attributeValue):
+def actionFolder(folderId, attributeValue, logger):
     headers = {"Content-Type": "application/json", "Authorization": f"Bearer {TOKEN}"}
     data = {"attribute": "public", "attributeValue": attributeValue}
-    response = requests.put(f"https://api.gofile.io/contents/{folderId}/update", headers=headers, json=data).json()
+    # response = requests.put(f"https://api.gofile.io/contents/{folderId}/update", headers=headers, json=data).json()
+    response = reqst(f"https://api.gofile.io/contents/{folderId}/update", headers=headers, json=data, logger=logger, method="put")
     if response["status"] == "ok":
         return True
     else:
@@ -204,17 +362,25 @@ def upload(filePath, folderPath, folderName, parentFolderId, private, logger):
         files = get_file_paths(folderPath)
         if not files:
             logger.error("No files found in folder")
-            sys.exit("No files found in folder")
+            sys.exit()
     else:
         if os.path.exists(filePath):
             files = [filePath]
         else:
             logger.error("File not found")
-            sys.exit("File not found")
+            sys.exit()
-
+    
+    # Getting servers
     servers = getservers(logger)
     if servers:
-        serverName = random.choice(servers)
+        if len(servers) > 1: # If there are multiple servers, check the size of the files
+            if max([os.path.getsize(file) for file in files]) > 100 * 1024 * 1024:  # 100 MB in bytes
+                logger.debug("One of the file have a size > 100 MB. Fetching best server...")
+                serverName = test_servers(servers, logger)
+            else:
+                serverName = random.choice(servers)
+        else:
+            serverName = servers[0]
         logger.debug(f"Selected server: {serverName}")
 
         if folderName and parentFolderId:
@@ -231,48 +397,60 @@ def upload(filePath, folderPath, folderName, parentFolderId, private, logger):
             parentFolderId = parentFolderId
             logger.debug(f"FolderId: {parentFolderId}")
         else:
-            parentFolderId = None
+            # parentFolderId = PRIVATE_PARENT_ID
+            logger.info(f"Creating folder: {folderName} for PRIVATE_PARENT_ID: {PRIVATE_PARENT_ID}")
+            folderId = createfolder(PRIVATE_PARENT_ID, None, logger)
+            parentFolderId = folderId
             logger.debug(f"FolderId: {parentFolderId}")
 
         for file in files:
             if parentFolderId:
                 logger.info(f"Uploading file: '{file}' ({file_size(file)}) to: '{parentFolderId}' on: '{serverName}'")
             else:
-                logger.info(f"Uploading file: {file} ({file_size(file)}) on: {serverName}")
+                logger.info(f"Uploading file: '{file}' ({file_size(file)}) on: '{serverName}'")
             downloadPage, parentFolderId, speed, elapsed_time = uploadfile(serverName, parentFolderId, file, logger)
             logger.info(f"File uploaded to: {downloadPage} in {elapsed_time} at {speed}")
 
         if not private:
-            action = actionFolder(parentFolderId, "true")
+            action = actionFolder(parentFolderId, "true", logger)
             if action:
                 logger.info("Folder made public")
             else:
                 logger.error(f"{action}")
         else:
-            action = actionFolder(parentFolderId, "false")
+            action = actionFolder(parentFolderId, "false", logger)
             if action:
                 logger.info("Folder made private")
             else:
                 logger.error(f"{action}")
+        play_sound(logger)
     else:
-        time.spleed(10)
+        time.sleep(10)
-        sys.exit("No server available")
+        sys.exit()
 
 
 def downloadFile(downloadUrl, path, logger):
     start_time = time.time()
     headers = {"Authorization": f"Bearer {TOKEN}"}
-    response = requests.get(downloadUrl, headers=headers)
+    response = requests.get(downloadUrl, headers=headers, stream=True)
-    with open(path, "wb") as f:
+    # response = reqst(downloadUrl, headers=headers, logger=logger, method="get")
-        f.write(response.content)
+    total_size = int(response.headers.get('content-length', 0))
+    
+    with open(path, "wb") as f, tqdm(total=total_size, unit='B', unit_scale=True, desc='Downloading', leave=False) as progress_bar:
+        for chunk in response.iter_content(1024):
+            if chunk:
+                f.write(chunk)
+                progress_bar.update(len(chunk))
     logger.debug(f"File downloaded: {path}")
     speed, elapsed_time = calculate_upload_speed(path, start_time)
     return speed, elapsed_time
 
 
-def download(folderId, logger, folderPath=None):
+def download(folderId, folderPath, force, logger):
     if 'https' in folderId:
         folderId = folderId.split('/')[-1]
+    if len(folderId) == 36:
+        folderId = get_code(folderId)
     files = []
     logger.info("Fetching files")
     logger.debug("FolderId: %s", folderId)
@@ -286,17 +464,28 @@ def download(folderId, logger, folderPath=None):
                     files.append(children[child])
                 else:
                     folderIdList.append(children[child]['id'])
+    nbdone = 0
     logger.info(f"Starting download of {len(files)} files")
     for file in files:
-        logger.info(f"Downloading file: {file['name']} ({file_size(num_bytes=file['size'])})")
+        logger.info(f"Downloading file {nbdone}/{len(files)}: {file['name']} ({file_size(num_bytes=file['size'])})")
         downloadUrl = file['link']
         name = file['name']
         if folderPath:
-            path = os.path.join(folderPath, name)
+            folderPath = check_folderPath(folderPath)
         else:
-            path = name
+            folderPath = check_folderPath(os.path.join(os.getcwd(), folderId))
-        speed, elapsed_time = downloadFile(downloadUrl, path, logger)
+        path = os.path.join(folderPath, name)
-        logger.info(f"File download to: {path} in {elapsed_time} at {speed}")
+        if os.path.exists(path) and not force:
+            logger.warning(f"File {name} already exists skipping (set --force to overwrite)")
+        elif os.path.exists(path) and force:
+            logger.warning(f"File {name} already exists overwriting")
+            speed, elapsed_time = downloadFile(downloadUrl, path, logger)
+            logger.info(f"File download to: {path} in {elapsed_time} at {speed}")
+        else:
+            speed, elapsed_time = downloadFile(downloadUrl, path, logger)
+            logger.info(f"File download to: {path} in {elapsed_time} at {speed}")
+        nbdone += 1
+    play_sound(logger)
 
 
 def opt():
@@ -316,21 +505,20 @@ def opt():
     exclusive_group.add_argument('--stats', "-s",  action='store_true', help='Display account stats.')
     
     exclusive_group.add_argument("--download", "-d", type=str, help="Id or code to the folder to be downloaded")
+    parser.add_argument("--output", "-o", type=str, help="¨Path to the folder to be downloaded")
+    parser.add_argument("--force", "-fo", action="store_true", help="Overwrite existing files")
 
-    # parser.add_argument_group
-    # parser.add_mutually_exclusive_group
     return parser.parse_args()
 
 
 def init():
     args = opt()
-    # print('args: ', args)
-    # print(sys.argv)
     log_format = "%(asctime)s %(levelname)s: %(message)s"
     logging.basicConfig(level=getattr(logging, args.log_level.upper()),format=log_format,datefmt="%H:%M:%S",)
     logger = logging.getLogger(__name__)
 
     load_dotenv()
+
     global TOKEN
     global PRIVATE_PARENT_ID
     global ACCOUNT_ID
@@ -344,7 +532,7 @@ def init():
             set_env_var("GOPLOAD_TOKEN", args.token)
         else:
             logger.error("Error: GOPLOAD_TOKEN not found, add GOPLOAD_TOKEN to your environment variables")
-            sys.exit("Error: GOPLOAD_TOKEN not found, add GOPLOAD_TOKEN to your environment variables")
+            sys.exit()
 
     if not PRIVATE_PARENT_ID:
         if args.private_parent_id:
@@ -357,8 +545,13 @@ def init():
             set_env_var("GOPLOAD_PRIVATE_PARENT_ID", args.private_parent_id)
         else:
             logger.error("Error: GOPLOAD_PRIVATE_PARENT_ID not found, add GOPLOAD_PRIVATE_PARENT_ID to your environment variables")
-            sys.exit("Error: GOPLOAD_PRIVATE_PARENT_ID not found, add GOPLOAD_PRIVATE_PARENT_ID to your environment variables")
+            sys.exit()
 
+    # No params:
+    if len(sys.argv) == 1:
+        logger.error("No arguments specified. Use -h for help")
+        sys.exit("")
+        
     # Stats section
     if args.stats:
         if len(sys.argv) == 2:
@@ -371,13 +564,13 @@ def init():
     elif args.file:
         if args.folder:
             logger.error("Both file and folder specified")
-            sys.exit("Both file and folder specified")
+            sys.exit()
         else:
             upload(args.file, args.folder, args.name, args.parent, args.private, logger)
     elif args.folder:
         if args.file:
             logger.error("Both file and folder specified")
-            sys.exit("Both file and folder specified")
+            sys.exit()
         else:
             upload(args.file, args.folder, args.name, args.parent, args.private, logger)
 
@@ -387,8 +580,12 @@ def init():
 
     # Download section
     elif args.download:
-        download(args.download, logger)
+        download(args.download, args.output, args.force, logger)
 
 
 if __name__ == "__main__":
-    init()
+    try:
+        init()
+    except KeyboardInterrupt:
+        print("\nExiting...")
+        sys.exit()

requirements.txt

@@ -1,3 +1,5 @@
 cryptography==39.0.1
 python-dotenv==1.0.1
 Requests==2.32.3
+simpleaudio==1.0.4
+tqdm==4.65.0

scripts/build.sh

@@ -10,7 +10,8 @@ tag=$1
 
 source ~/miniconda3/etc/profile.d/conda.sh
 conda activate 310
-python -m nuitka --onefile --assume-yes-for-downloads --output-dir=dist --static-libpython=no gofilecli.py 
+pip install -r requirements.txt
+python -m nuitka --onefile --assume-yes-for-downloads --output-dir=dist --static-libpython=no gofilecli.py --include-data-files=./assets/sounds/Blow_edited.wav=assets/sounds/Blow_edited.wav
 mv dist/gofilecli.bin ./gofilecli
 
 mkdir -p GoFileCLI_linux-arm64