elements.py/cli.py
2023-11-09 12:10:46 -05:00

115 lines
3.0 KiB
Python

import os
import json
import subprocess
import sys
from src.help import help
def clear_console():
# Clear console based on the operating system
if os.name == "nt": # For Windows
os.system("cls")
else: # For Unix/Linux/Mac
os.system("clear")
def check_python_installation():
try:
python_version = subprocess.check_output(
["python", "--version"], stderr=subprocess.STDOUT, universal_newlines=True
)
print(f"Python is already installed ({python_version})")
return True
except FileNotFoundError:
return False
def install_python():
print("Python is not installed. Please download and install it from:")
print("https://www.python.org/downloads/")
input("Press Enter to continue after installing Python...")
sys.exit(1)
def check_rpc_config():
if os.path.exists("rpc_config.json"):
return True
return False
def get_rpc_config():
rpc_host = input("Enter the RPC host: ")
rpc_port = input("Enter the RPC port: ")
rpc_user = input("Enter the RPC user: ")
rpc_password = input("Enter the RPC password: ")
rpc_config = {
"rpc_host": rpc_host,
"rpc_port": rpc_port,
"rpc_user": rpc_user,
"rpc_password": rpc_password,
}
with open("rpc_config.json", "w") as config_file:
json.dump(rpc_config, config_file)
def run_script(script_name):
script_path = os.path.join("src", f"{script_name}.py")
if os.path.exists(script_path):
# print(f"Running {script_name}...")
os.system(f"python {script_path}")
else:
print(f"Error: {script_name} not found!")
def main():
setup_completed = False
while not check_python_installation():
install_python()
subprocess.run(["pip", "install", "-r", "requirements.txt"])
while not check_rpc_config():
clear_console()
print("RPC config file not found.")
user_input = input(
"Do you have credentials for your Elements node RPC (yes/no)? "
).lower()
if user_input in ["no", "n"]:
print(
"Please follow the instructions to set up an Elements node: [https://docs.liquid.net/docs/building-on-liquid]"
)
input("Press Enter to continue after setting up the Elements node...")
elif user_input in ["yes", "y"]:
get_rpc_config()
else:
print("Invalid input. Please enter 'y' or 'n'.")
setup_completed = True
print("You are ready to proceed.")
clear_console()
print("\nElements Python Command Line Interface")
help()
while True:
user_input = input("Enter a command: ")
if user_input.lower() == "exit":
break
if user_input.lower() == "help":
help()
if setup_completed:
run_script(user_input)
else:
print("Error: Setup not completed. Use 'help' to see available commands.")
if __name__ == "__main__":
main()