diff --git a/upload.py b/upload.py index 0bde7af..fc9791a 100644 --- a/upload.py +++ b/upload.py @@ -1,6 +1,7 @@ import os import requests from pathlib import Path +from requests_toolbelt import MultipartEncoder # Endpoints ottenuti tramite http://xvideosharing.com/api/folder/list?key=key&fld_id=fld_id @@ -14,6 +15,8 @@ headers = {"Content-Type": "application/json"} # Global Variables fileinDirectory = [] +MAX_UPLOAD_GB = 750 +UPLOADED_GB = 0 def visit_list(l): @@ -67,7 +70,7 @@ def getFolderList(): def checkIfFileExists(title_file, fld_id): - url = f"{endpoint_fileList}&fld_id={fld_id}" + url = f"{endpoint_fileList}&fld_id={fld_id}&per_page=2000" response = requests.get(url, headers=headers) if response.ok: data = response.json() @@ -100,23 +103,26 @@ def upload(file_obj): print(file_path) content_name = str(fileslist) fileNoExtension = Path(file_path).stem + file_size = os.path.getsize(file_path) + file_size_gb = round(file_size / (1024 * 1024 * 1024), 3) + global UPLOADED_GB + UPLOADED_GB += file_size_gb count += 1 videoexists = checkIfFileExists(fileNoExtension, fld_id) - if videoexists: + if videoexists or UPLOADED_GB == MAX_UPLOAD_GB or file_size_gb >= 10: pass else: + pass url = "https://marlin2.vtube.to/upload/01" - files = { - "file": ( - content_name, - open( - file_path, - "rb", - ), - ), - "api_key": (None, "473u3rs97it2ge3hk77"), - } - response = requests.post(url, files=files) + m = MultipartEncoder( + fields={ + "file": (content_name, open(file_path, "rb")), + "api_key": "473u3rs97it2ge3hk77", + } + ) + response = requests.post( + url, data=m, headers={"Content-Type": m.content_type} + ) try: response.raise_for_status() except: @@ -137,4 +143,4 @@ def upload(file_obj): if __name__ == "__main__": folderUpload = input("Inserisci la cartella dove sono presenti i file : ") - upload(folderUpload) + upload(folderUpload) \ No newline at end of file