mirror of
https://github.com/Neocky/pluGET.git
synced 2024-04-29 16:12:30 +00:00
Added plugin downloader capabilities
Added plugin downloader capabilities and added full console argument support. Fixed #46 and added error handling for it
This commit is contained in:
parent
ac139ed048
commit
3943357569
19
pluGET.py
19
pluGET.py
@ -27,21 +27,14 @@ if __name__ == "__main__":
|
|||||||
#parser.add_argument("-a", "--archive", action="store_true", help="archive mode")
|
#parser.add_argument("-a", "--archive", action="store_true", help="archive mode")
|
||||||
#parser.add_argument("--exclude", help="files to exclude")
|
#parser.add_argument("--exclude", help="files to exclude")
|
||||||
parser.add_argument("mode", help="Mode (install/update)", nargs='?', default=None)
|
parser.add_argument("mode", help="Mode (install/update)", nargs='?', default=None)
|
||||||
parser.add_argument("plugin", help="Plugin Name", nargs='?', default=None)
|
parser.add_argument("object", help="Object/Plugin Name", nargs='?', default=None)
|
||||||
|
parser.add_argument("version", help="Version", nargs='?', default=None)
|
||||||
args = vars(parser.parse_args())
|
args = vars(parser.parse_args())
|
||||||
if args["mode"] is not None and args["plugin"] is not None:
|
if args["mode"] is not None and args["object"] is not None:
|
||||||
print("arguments")
|
# arguments were give and call the handle_input function to get the right function call
|
||||||
|
handle_input(args["mode"], args["object"], args["version"], arguments_from_console=True)
|
||||||
else:
|
else:
|
||||||
print("no arguments")
|
# no arguments were given so start pluGET console
|
||||||
clear_console()
|
clear_console()
|
||||||
print_logo()
|
print_logo()
|
||||||
rich_print_error("test")
|
|
||||||
rich_print_error("test2")
|
|
||||||
handle_input()
|
handle_input()
|
||||||
config = config_value()
|
|
||||||
print(config.connection)
|
|
||||||
print(config.path_to_plugin_folder)
|
|
||||||
print(config.sftp_port)
|
|
||||||
print(config.local_seperate_download_path)
|
|
||||||
|
|
||||||
|
@ -19,7 +19,7 @@ class config_value():
|
|||||||
yaml = ruamel.yaml.YAML()
|
yaml = ruamel.yaml.YAML()
|
||||||
with open("pluGET_config.yaml", "r") as config_file:
|
with open("pluGET_config.yaml", "r") as config_file:
|
||||||
data = yaml.load(config_file)
|
data = yaml.load(config_file)
|
||||||
self.connection = data["Connection"]
|
self.connection = str(data["Connection"]).lower()
|
||||||
self.path_to_plugin_folder = Path(data["Local"]["PathToPluginFolder"])
|
self.path_to_plugin_folder = Path(data["Local"]["PathToPluginFolder"])
|
||||||
self.local_seperate_download_path = True if data["Local"]["SeperateDownloadPath"] == True else False
|
self.local_seperate_download_path = True if data["Local"]["SeperateDownloadPath"] == True else False
|
||||||
self.local_path_to_seperate_download_path = Path(data["Local"]["PathToSeperateDownloadPath"])
|
self.local_path_to_seperate_download_path = Path(data["Local"]["PathToSeperateDownloadPath"])
|
||||||
@ -100,7 +100,7 @@ def validate_config() -> None:
|
|||||||
config = config_value()
|
config = config_value()
|
||||||
# rich console for nice colors
|
# rich console for nice colors
|
||||||
console = Console()
|
console = Console()
|
||||||
if (config.connection).lower() not in accepted_values[0]:
|
if config.connection not in accepted_values[0]:
|
||||||
console.print(f"Error in Config! Accepted values for key 'Connection' are {accepted_values[0]}",
|
console.print(f"Error in Config! Accepted values for key 'Connection' are {accepted_values[0]}",
|
||||||
style="bright_red")
|
style="bright_red")
|
||||||
exit_afterwards = True
|
exit_afterwards = True
|
||||||
|
@ -1,8 +1,9 @@
|
|||||||
""""
|
""""
|
||||||
Handles the input through the pluGET command line
|
Handles the input through the pluGET command line
|
||||||
"""
|
"""
|
||||||
|
|
||||||
from src.utils.utilities import rich_print_error
|
from src.utils.utilities import rich_print_error
|
||||||
|
from src.plugin.plugin_downloader import get_specific_plugin
|
||||||
|
|
||||||
|
|
||||||
# check
|
# check
|
||||||
@ -16,11 +17,13 @@ from src.utils.utilities import rich_print_error
|
|||||||
# search ???
|
# search ???
|
||||||
|
|
||||||
|
|
||||||
def handle_input() -> None:
|
def handle_input(input_command=None, input_selected_object=None, input_parameter=None, arguments_from_console=False) -> None:
|
||||||
"""
|
"""
|
||||||
Manages the correct function calling from the given input
|
Manages the correct function calling from the given input
|
||||||
"""
|
"""
|
||||||
while True:
|
while True:
|
||||||
|
# when arguemnts were not passed from console
|
||||||
|
if arguments_from_console is False:
|
||||||
try:
|
try:
|
||||||
input_command, input_selected_object, input_parameter = get_input()
|
input_command, input_selected_object, input_parameter = get_input()
|
||||||
except TypeError:
|
except TypeError:
|
||||||
@ -31,8 +34,7 @@ def handle_input() -> None:
|
|||||||
case "get":
|
case "get":
|
||||||
match input_selected_object.isdigit():
|
match input_selected_object.isdigit():
|
||||||
case True:
|
case True:
|
||||||
print("get specific package")
|
get_specific_plugin(input_selected_object, input_parameter)
|
||||||
#getSpecificPackage(inputSelectedObject, pluginPath, inputParams)
|
|
||||||
case _:
|
case _:
|
||||||
print("get search specific package")
|
print("get search specific package")
|
||||||
#searchPackage(inputSelectedObject)
|
#searchPackage(inputSelectedObject)
|
||||||
@ -67,16 +69,23 @@ def handle_input() -> None:
|
|||||||
# download papermc
|
# download papermc
|
||||||
print("download papermc")
|
print("download papermc")
|
||||||
#papermc_downloader(inputSelectedObject, inputParams)
|
#papermc_downloader(inputSelectedObject, inputParams)
|
||||||
|
case "exit":
|
||||||
|
return
|
||||||
case _:
|
case _:
|
||||||
rich_print_error("Error: Command not found. Please try again. :(")
|
rich_print_error("Error: Command not found. Please try again. :(")
|
||||||
rich_print_error("Use: 'help command' to get all available commands")
|
rich_print_error("Use: 'help command' to get all available commands")
|
||||||
|
|
||||||
|
# return to break out of while loop
|
||||||
|
if arguments_from_console:
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
def get_input() -> None:
|
def get_input() -> None:
|
||||||
"""
|
"""
|
||||||
Gets command line input and calls the handle input function
|
Gets command line input and calls the handle input function
|
||||||
"""
|
"""
|
||||||
input_command = None
|
input_command = None
|
||||||
|
print("'STRG + C' to exit")
|
||||||
while True:
|
while True:
|
||||||
try:
|
try:
|
||||||
input_command, input_selected_object, *input_parameter = input("pluGET >> ").split()
|
input_command, input_selected_object, *input_parameter = input("pluGET >> ").split()
|
||||||
|
161
src/plugin/plugin_downloader.py
Normal file
161
src/plugin/plugin_downloader.py
Normal file
@ -0,0 +1,161 @@
|
|||||||
|
"""
|
||||||
|
File and functions which handle the download of the specific plugins
|
||||||
|
"""
|
||||||
|
|
||||||
|
import re
|
||||||
|
import urllib.request
|
||||||
|
from urllib.error import HTTPError
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
import rich
|
||||||
|
|
||||||
|
from src.utils.utilities import rich_print_error, convert_file_size_down, remove_temp_plugin_folder, create_temp_plugin_folder
|
||||||
|
from src.utils.utilities import api_do_request
|
||||||
|
from src.handlers.handle_config import config_value
|
||||||
|
|
||||||
|
|
||||||
|
def handle_regex_package_name(full_plugin_name) -> str:
|
||||||
|
"""
|
||||||
|
Return the plugin name after trimming clutter from name with regex operations
|
||||||
|
"""
|
||||||
|
# trims the part of the package that has for example "[1.1 Off]" in it
|
||||||
|
unwanted_plugin_name = re.search(r'(^\[+[a-zA-Z0-9\s\W*\.*\-*\+*\%*\,]*\]+)', full_plugin_name)
|
||||||
|
if bool(unwanted_plugin_name):
|
||||||
|
unwanted_plugin_name_string = unwanted_plugin_name.group()
|
||||||
|
full_plugin_name = full_plugin_name.replace(unwanted_plugin_name_string, '')
|
||||||
|
|
||||||
|
# gets the real plugin_name "word1 & word2" is not supported only gets word1
|
||||||
|
plugin_name = re.search(r'([a-zA-Z]\d*)+(\s?\-*\_*[a-zA-Z]\d*\+*\-*\'*)+', full_plugin_name)
|
||||||
|
try:
|
||||||
|
plugin_name_full_string = plugin_name.group()
|
||||||
|
found_plugin_name = plugin_name_full_string.replace(' ', '')
|
||||||
|
except AttributeError:
|
||||||
|
found_plugin_name = unwanted_plugin_name_string
|
||||||
|
return found_plugin_name
|
||||||
|
|
||||||
|
|
||||||
|
def get_version_id(plugin_id, plugin_version) -> str:
|
||||||
|
"""
|
||||||
|
Returns the version id of the plugin
|
||||||
|
"""
|
||||||
|
if plugin_version == None or plugin_version == 'latest':
|
||||||
|
url = f"https://api.spiget.org/v2/resources/{plugin_id}/versions/latest"
|
||||||
|
response = api_do_request(url)
|
||||||
|
version_id = response["id"]
|
||||||
|
return version_id
|
||||||
|
|
||||||
|
|
||||||
|
url = f"https://api.spiget.org/v2/resources/{plugin_id}/versions?size=100&sort=-name"
|
||||||
|
version_list = api_do_request(url)
|
||||||
|
for plugins in version_list:
|
||||||
|
plugin_update = plugins["name"]
|
||||||
|
version_id = plugins["id"]
|
||||||
|
if plugin_update == plugin_version:
|
||||||
|
return version_id
|
||||||
|
return version_list[0]["id"]
|
||||||
|
|
||||||
|
|
||||||
|
def getVersionName(plugin_id, plugin_version_id) -> str:
|
||||||
|
"""
|
||||||
|
Returns the name of a specific version
|
||||||
|
"""
|
||||||
|
url = f"https://api.spiget.org/v2/resources/{plugin_id}/versions/{plugin_version_id}"
|
||||||
|
response = api_do_request(url)
|
||||||
|
version_name = response["name"]
|
||||||
|
return version_name
|
||||||
|
|
||||||
|
|
||||||
|
def get_download_path() -> str:
|
||||||
|
"""
|
||||||
|
Reads the config and gets the path of the plugin folder
|
||||||
|
"""
|
||||||
|
config_values = config_value()
|
||||||
|
|
||||||
|
if config_values.connection == "local":
|
||||||
|
if config_values.local_seperate_download_path is True:
|
||||||
|
return config_values.local_path_to_seperate_download_path
|
||||||
|
else:
|
||||||
|
return config_values.path_to_plugin_folder
|
||||||
|
|
||||||
|
else:
|
||||||
|
if config_values.remote_seperate_download_path is True:
|
||||||
|
return config_values.remote_path_to_seperate_download_path
|
||||||
|
else:
|
||||||
|
return config_values.remote_plugin_folder_on_server
|
||||||
|
|
||||||
|
|
||||||
|
def download_specific_plugin_version(plugin_id, download_path, version_id="latest") -> None:
|
||||||
|
"""
|
||||||
|
Download a specific plugin
|
||||||
|
"""
|
||||||
|
#config_values = config_value()
|
||||||
|
if version_id != "latest" and version_id != None:
|
||||||
|
#url = f"https://spigotmc.org/resources/{plugin_id}/download?version={versionID}"
|
||||||
|
rich_print_error("Sorry but specific version downloads aren't supported because of cloudflare protection. :(")
|
||||||
|
rich_print_error("Reverting to latest version.")
|
||||||
|
|
||||||
|
#url = f"https://api.spiget.org/v2/resources/{plugin_id}/versions/latest/download" #throws 403 forbidden error...cloudflare :(
|
||||||
|
url = f"https://api.spiget.org/v2/resources/{plugin_id}/download"
|
||||||
|
|
||||||
|
urrlib_opener = urllib.request.build_opener()
|
||||||
|
urrlib_opener.addheaders = [('User-agent', 'pluGET/1.0')]
|
||||||
|
urllib.request.install_opener(urrlib_opener)
|
||||||
|
|
||||||
|
remote_file = urllib.request.urlopen(url)
|
||||||
|
try:
|
||||||
|
file_size = int(remote_file.info()['Content-Length'])
|
||||||
|
except TypeError:
|
||||||
|
# if api won't return file size set it to 0 to avoid throwing an error
|
||||||
|
file_size = 0
|
||||||
|
urllib.request.urlretrieve(url, download_path)
|
||||||
|
print(" ", end='')
|
||||||
|
if file_size >= 1000000 and file_size != 0:
|
||||||
|
file_size_data = convert_file_size_down(convert_file_size_down(file_size))
|
||||||
|
print("Downloaded " + (str(file_size_data)).rjust(9) + f" MB here {download_path}")
|
||||||
|
elif file_size != 0:
|
||||||
|
file_size_data = convert_file_size_down(file_size)
|
||||||
|
print("Downloaded " + (str(file_size_data)).rjust(9) + f" KB here {download_path}")
|
||||||
|
else:
|
||||||
|
print(f"Downloaded file here {download_path}")
|
||||||
|
# TODO add sftp and ftp support
|
||||||
|
#if config_values.connection == "sftp":
|
||||||
|
# sftp_session = createSFTPConnection()
|
||||||
|
# sftp_upload_file(sftp_session, download_path)
|
||||||
|
#elif config_values.connection == "ftp":
|
||||||
|
# ftp_session = createFTPConnection()
|
||||||
|
# ftp_upload_file(ftp_session, download_path)
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
def get_specific_plugin(plugin_id, plugin_version="latest") -> None:
|
||||||
|
"""
|
||||||
|
Gets the specific plugin and calls the download function
|
||||||
|
"""
|
||||||
|
config_values = config_value()
|
||||||
|
# use a temporary folder to store plugins until they are uploaded
|
||||||
|
if config_values.connection != "local":
|
||||||
|
download_path = create_temp_plugin_folder()
|
||||||
|
else:
|
||||||
|
download_path = get_download_path()
|
||||||
|
|
||||||
|
url = f"https://api.spiget.org/v2/resources/{plugin_id}"
|
||||||
|
plugin_details = api_do_request(url)
|
||||||
|
try:
|
||||||
|
plugin_name = plugin_details["name"]
|
||||||
|
except KeyError:
|
||||||
|
# exit if plugin id coudn't be found
|
||||||
|
rich_print_error("Error: Plugin ID couldn't be found")
|
||||||
|
return None
|
||||||
|
plugin_name = handle_regex_package_name(plugin_name)
|
||||||
|
plugin_version_id = get_version_id(plugin_id, plugin_version)
|
||||||
|
plugin_version_name = getVersionName(plugin_id, plugin_version_id)
|
||||||
|
plugin_download_name = f"{plugin_name}-{plugin_version_name}.jar"
|
||||||
|
download_plugin_path = Path(f"{download_path}/{plugin_download_name}")
|
||||||
|
# set the plugin_version_id to None if a specific version wasn't given as parameter
|
||||||
|
if plugin_version == "latest" or plugin_version is None:
|
||||||
|
plugin_version_id = None
|
||||||
|
download_specific_plugin_version(plugin_id, download_plugin_path, plugin_version_id)
|
||||||
|
|
||||||
|
if config_values.connection != "local":
|
||||||
|
remove_temp_plugin_folder()
|
||||||
|
return None
|
@ -1,5 +1,5 @@
|
|||||||
"""
|
"""
|
||||||
Holds all the utilitie code for pluGET and the webrequests function
|
Holds all the utilitie code for pluGET and the webrequests function
|
||||||
"""
|
"""
|
||||||
|
|
||||||
import os
|
import os
|
||||||
@ -78,3 +78,12 @@ def remove_temp_plugin_folder() -> None:
|
|||||||
except OSError as e:
|
except OSError as e:
|
||||||
rich_print_error(f"Error: {e.filename} - {e.strerror}")
|
rich_print_error(f"Error: {e.filename} - {e.strerror}")
|
||||||
return
|
return
|
||||||
|
|
||||||
|
|
||||||
|
def convert_file_size_down(file_size) -> float:
|
||||||
|
"""
|
||||||
|
Convert the size of the number one down. E.g. MB -> KB through division with 1024
|
||||||
|
"""
|
||||||
|
converted_file_size = (int(file_size)) / 1024
|
||||||
|
converted_file_size = round(converted_file_size, 2)
|
||||||
|
return converted_file_size
|
||||||
|
Loading…
Reference in New Issue
Block a user