main/tools/pixeldrain.py

import os
import requests


def pixeldrain(url, output_dir):
    try:
        file_id = url.split("pixeldrain.com/u/")[1]
        response = requests.get(f"https://pixeldrain.com/api/file/{file_id}")

        if response.status_code == 200:
            file_name = (response.headers.get("Content-Disposition").split("filename=")[-1].strip('";'))
            file_path = os.path.join(output_dir, file_name)

            with open(file_path, "wb") as newfile:
                newfile.write(response.content)
                return file_path
        else: return None
    except Exception as e:
        raise RuntimeError(e)