noham/GoFileCLI
1
0
Fork 0
mirror of https://github.com/NohamR/GoFileCLI.git synced 2025-05-24 00:49:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

changes

Browse Source
This commit is contained in:
√(noham)² 2024-08-16 19:01:36 +02:00
parent 2d20ce0ee7
commit e908521f83
2 changed files with 192 additions and 144 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

126
gopload.py
View File

@ -4,12 +4,14 @@ import time
import argparse import argparse
import os import os
from dotenv import load_dotenv from dotenv import load_dotenv
from tqdm import tqdm
def get_file_paths(folderPath): def get_file_paths(folderPath):
filePaths = [] filePaths = []
for root, directories, files in os.walk(folderPath): for root, directories, files in os.walk(folderPath):
for file in files: for file in files:
if '.DS_Store' not in file: if ".DS_Store" not in file:
file_path = os.path.join(root, file) file_path = os.path.join(root, file)
filePaths.append(file_path) filePaths.append(file_path)
return filePaths return filePaths
@ -17,10 +19,10 @@ def get_file_paths(folderPath):
def getservers(logger): def getservers(logger):
servers = [] servers = []
response = requests.get('https://api.gofile.io/servers').json() response = requests.get("https://api.gofile.io/servers").json()
if response['status'] == "ok": if response["status"] == "ok":
for server in response['data']['servers']: for server in response["data"]["servers"]:
servers.append(server['name']) servers.append(server["name"])
logger.debug(f"{response}") logger.debug(f"{response}")
return servers return servers
else: else:
@ -29,17 +31,21 @@ def getservers(logger):
def createfolder(parentFolderId, folderName, logger): def createfolder(parentFolderId, folderName, logger):
headers = {'Content-Type': 'application/json','Authorization': f'Bearer {TOKEN}',} headers = {"Content-Type": "application/json", "Authorization": f"Bearer {TOKEN}"}
if folderName: if folderName:
data = {"parentFolderId": parentFolderId, "folderName": folderName} data = {"parentFolderId": parentFolderId, "folderName": folderName}
else: else:
data = {"parentFolderId": parentFolderId} data = {"parentFolderId": parentFolderId}
response = requests.post('https://api.gofile.io/contents/createFolder', headers=headers, json=data).json() response = requests.post(
if response['status'] == "ok": "https://api.gofile.io/contents/createFolder", headers=headers, json=data
name = response['data']['name'] ).json()
code = response['data']['code'] if response["status"] == "ok":
folderId = response['data']['folderId'] name = response["data"]["name"]
logger.debug(f"""Folder {name} created with code {code} and folderId {folderId}""") code = response["data"]["code"]
folderId = response["data"]["folderId"]
logger.debug(
f"""Folder {name} created with code {code} and folderId {folderId}"""
)
return folderId return folderId
else: else:
logger.error(f"{response}") logger.error(f"{response}")
@ -47,26 +53,34 @@ def createfolder(parentFolderId, folderName, logger):
def uploadfile(serverName, folderId, filePath, logger): def uploadfile(serverName, folderId, filePath, logger):
headers = {'Authorization': f'Bearer {TOKEN}'} headers = {"Authorization": f"Bearer {TOKEN}"}
files = {'file': open(filePath, 'rb')} files = {"file": open(filePath, "rb")}
data = {'folderId': folderId} data = {"folderId": folderId}
response = requests.post(f'https://{serverName}.gofile.io/contents/uploadfile', headers=headers, files=files, data=data).json() response = requests.post(
if response['status'] == "ok": f"https://{serverName}.gofile.io/contents/uploadfile",
name = response['data']['fileName'] headers=headers,
code = response['data']['code'] files=files,
downloadPage = response['data']['downloadPage'] data=data,
parentFolderId = response['data']['parentFolder'] ).json()
if response["status"] == "ok":
name = response["data"]["fileName"]
code = response["data"]["code"]
downloadPage = response["data"]["downloadPage"]
parentFolderId = response["data"]["parentFolder"]
logger.debug(f"""File {name} uploaded to {downloadPage} with code {code}""") logger.debug(f"""File {name} uploaded to {downloadPage} with code {code}""")
return downloadPage, parentFolderId return downloadPage, parentFolderId
else: else:
logger.error(f"{response}") logger.error(f"{response}")
return None return None
def actionFolder(folderId, attributeValue): def actionFolder(folderId, attributeValue):
headers = {'Content-Type': 'application/json','Authorization': f'Bearer {TOKEN}'} headers = {"Content-Type": "application/json", "Authorization": f"Bearer {TOKEN}"}
data = {"attribute": "public", "attributeValue": attributeValue} data = {"attribute": "public", "attributeValue": attributeValue}
response = requests.put(f'https://api.gofile.io/contents/{folderId}/update', headers=headers, json=data).json() response = requests.put(
if response['status'] == "ok": f"https://api.gofile.io/contents/{folderId}/update", headers=headers, json=data
).json()
if response["status"] == "ok":
return True return True
else: else:
return response return response
@ -99,7 +113,9 @@ def main(filePath, folderPath, folderName, parentFolderId, private, logger):
logger.debug(f"FolderId: {folderId}") logger.debug(f"FolderId: {folderId}")
parentFolderId = folderId parentFolderId = folderId
elif folderName: elif folderName:
logger.info(f"Creating folder: {folderName} for PRIVATE_PARENT_ID: {PRIVATE_PARENT_ID}") logger.info(
f"Creating folder: {folderName} for PRIVATE_PARENT_ID: {PRIVATE_PARENT_ID}"
)
folderId = createfolder(PRIVATE_PARENT_ID, folderName, logger) folderId = createfolder(PRIVATE_PARENT_ID, folderName, logger)
logger.debug(f"FolderId: {folderId}") logger.debug(f"FolderId: {folderId}")
parentFolderId = folderId parentFolderId = folderId
@ -112,20 +128,24 @@ def main(filePath, folderPath, folderName, parentFolderId, private, logger):
for file in files: for file in files:
if parentFolderId: if parentFolderId:
logger.info(f"Uploading file: {file} to: {parentFolderId} on: {serverName}") logger.info(
f"Uploading file: {file} to: {parentFolderId} on: {serverName}"
)
else: else:
logger.info(f"Uploading file: {file} on: {serverName}") logger.info(f"Uploading file: {file} on: {serverName}")
downloadPage, parentFolderId = uploadfile(serverName, parentFolderId, file, logger) downloadPage, parentFolderId = uploadfile(
serverName, parentFolderId, file, logger
)
logger.info(f"Download page: {downloadPage}") logger.info(f"Download page: {downloadPage}")
if not private: if not private:
action = actionFolder(parentFolderId, 'true') action = actionFolder(parentFolderId, "true")
if action: if action:
logger.info("Folder made public") logger.info("Folder made public")
else: else:
logger.error(f"{action}") logger.error(f"{action}")
else: else:
action = actionFolder(parentFolderId, 'false') action = actionFolder(parentFolderId, "false")
if action: if action:
logger.info("Folder made private") logger.info("Folder made private")
else: else:
@ -135,19 +155,43 @@ def main(filePath, folderPath, folderName, parentFolderId, private, logger):
exit(1) exit(1)
if __name__ == '__main__': if __name__ == "__main__":
parser = argparse.ArgumentParser(description='Upload a file to GoFile.') parser = argparse.ArgumentParser(description="Upload a file to GoFile.")
parser.add_argument('--file', '-i', type=str, help='Path to the file to be uploaded') parser.add_argument(
parser.add_argument('--folder', '-f', type=str, help='Path to the folder to be uploaded') "--file", "-i", type=str, help="Path to the file to be uploaded"
parser.add_argument('--name', '-n', type=str, help='Name of the folder on the server') )
parser.add_argument('--parent', '-p', type=str, help='Folder id to upload the file to') parser.add_argument(
parser.add_argument('--private', '-pr', action='store_true', help='Upload to private folder default=False') "--folder", "-f", type=str, help="Path to the folder to be uploaded"
parser.add_argument('--log-level', type=str, choices=['DEBUG', 'ERROR', 'INFO', 'OFF', 'WARN'], default='INFO', help='Set log level [default: INFO]') )
parser.add_argument(
"--name", "-n", type=str, help="Name of the folder on the server"
)
parser.add_argument(
"--parent", "-p", type=str, help="Folder id to upload the file to"
)
parser.add_argument(
"--private",
"-pr",
action="store_true",
help="Upload to private folder default=False",
)
parser.add_argument(
"--log-level",
type=str,
choices=["DEBUG", "ERROR", "INFO", "OFF", "WARN"],
default="INFO",
help="Set log level [default: INFO]",
)
args = parser.parse_args() args = parser.parse_args()
import logging import logging
log_format = '%(asctime)s %(levelname)s: %(message)s'
logging.basicConfig(level=getattr(logging, args.log_level.upper()), format=log_format, datefmt='%H:%M:%S') 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__) logger = logging.getLogger(__name__)
load_dotenv() load_dotenv()
@ -157,7 +201,9 @@ if __name__ == '__main__':
logger.error("Error: TOKEN not found, create a .env file with TOKEN") logger.error("Error: TOKEN not found, create a .env file with TOKEN")
exit(1) exit(1)
if not PRIVATE_PARENT_ID: if not PRIVATE_PARENT_ID:
logger.error("Error: PRIVATE_PARENT_ID not found, create a .env file with PRIVATE_PARENT_ID") logger.error(
"Error: PRIVATE_PARENT_ID not found, create a .env file with PRIVATE_PARENT_ID"
)
exit(1) exit(1)
if args.name and not args.parent: if args.name and not args.parent:

2
requirements.txt Normal file
View File

@ -0,0 +1,2 @@
python-dotenv==1.0.1
Requests==2.32.3