Add
This commit is contained in:
Saturneric
parent
7f7a8d2e00
commit
28203a8369
193
main.py
Normal file
193
main.py
Normal file
@ -0,0 +1,193 @@
|
|||||||
|
#!/usr/bin/python
|
||||||
|
import vultr
|
||||||
|
import threading
|
||||||
|
from tkinter import *
|
||||||
|
|
||||||
|
#Set
|
||||||
|
Location = "25";
|
||||||
|
API_Key = ""
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
#Func
|
||||||
|
def use_proxy(port):
|
||||||
|
CMD = "export ALL_PROXY=socks5://127.0.0.1:"+port;
|
||||||
|
os.system(CMD)
|
||||||
|
|
||||||
|
class Application():
|
||||||
|
app = None
|
||||||
|
widgets = {}
|
||||||
|
|
||||||
|
def __init__(self,title):
|
||||||
|
self.app = Tk(className=title)
|
||||||
|
def set_size(self,widget,width,height):
|
||||||
|
widget['height'] = height;
|
||||||
|
widget['width'] = width;
|
||||||
|
return widget
|
||||||
|
|
||||||
|
def set_widget(self,widget,row,column,sticky):
|
||||||
|
widget.grid(column = column, row = row, sticky = sticky)
|
||||||
|
return widget
|
||||||
|
|
||||||
|
def get_button(self,title,command,w_id):
|
||||||
|
app_button = Button(self.app)
|
||||||
|
app_button['text'] = title;
|
||||||
|
app_button['command'] = command;
|
||||||
|
self.widgets[w_id] = app_button;
|
||||||
|
return app_button;
|
||||||
|
def get_entry(self,w_id):
|
||||||
|
app_entry = Entry(self.app)
|
||||||
|
self.widgets[w_id] = app_entry;
|
||||||
|
return app_entry
|
||||||
|
def get_entry_long(self,w_id):
|
||||||
|
app_entry = Entry(self.app)
|
||||||
|
app_entry['width'] = 50;
|
||||||
|
self.widgets[w_id] = app_entry;
|
||||||
|
return app_entry
|
||||||
|
def get_label(self,text,w_id):
|
||||||
|
app_label = Label(self.app)
|
||||||
|
app_label['text'] = text
|
||||||
|
self.widgets[w_id] = app_label;
|
||||||
|
return app_label
|
||||||
|
def get_text(self,w_id):
|
||||||
|
app_text = Text(self.app)
|
||||||
|
self.widgets[w_id] = app_text;
|
||||||
|
return app_text
|
||||||
|
def get_menubtn(self,w_id):
|
||||||
|
app_menubtn = Menubutton(self.app)
|
||||||
|
self.widgets[w_id] = app_menubtn;
|
||||||
|
return app_menubtn
|
||||||
|
def set_munubtn(self,widget,text,items):
|
||||||
|
widget['text'] = text;
|
||||||
|
widget.menu = Menu(widget)
|
||||||
|
for item in items:
|
||||||
|
widget.menu.add_command(label=item)
|
||||||
|
widget['menu'] = widget.menu
|
||||||
|
return widget
|
||||||
|
|
||||||
|
def refresh_text(self,widget,w_id):
|
||||||
|
info_text.delete('1.0', END)
|
||||||
|
|
||||||
|
def get_widget(self,w_id):
|
||||||
|
return self.widgets[w_id]
|
||||||
|
|
||||||
|
|
||||||
|
class Display(Application):
|
||||||
|
display = None
|
||||||
|
info = {}
|
||||||
|
def __init__(self,title):
|
||||||
|
self.display = Toplevel()
|
||||||
|
info['decision'] = False
|
||||||
|
def click_yes(self,func):
|
||||||
|
info['decision'] = True
|
||||||
|
func();
|
||||||
|
def click_no(self,func):
|
||||||
|
info['decision'] = False
|
||||||
|
func();
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
account = vultr.account()
|
||||||
|
app = Application("vultr manager")
|
||||||
|
app.get_label("API_Key: ",'key_label')
|
||||||
|
app.set_widget(app.widgets['key_label'],0,0,W)
|
||||||
|
|
||||||
|
key_entry = app.get_entry_long('key_entry')
|
||||||
|
app.set_widget(app.get_widget('key_entry'),1,0,W)
|
||||||
|
|
||||||
|
app.get_text('info_text')
|
||||||
|
app.set_size(app.get_widget('info_text'),65,20)
|
||||||
|
app.set_widget(app.get_widget('info_text'),4,0,W)
|
||||||
|
|
||||||
|
|
||||||
|
app.set_widget(app.get_label("SUBID: ",'sid_label'),6,0,E)
|
||||||
|
app.get_entry('sid_entry')
|
||||||
|
app.set_widget(app.get_widget('sid_entry'),6,1,E)
|
||||||
|
|
||||||
|
app.set_widget(app.get_label('Status:','tstatus_label'),5,0,W)
|
||||||
|
app.set_widget(app.get_label('','status_label'),5,1,W)
|
||||||
|
|
||||||
|
religions = ["Los Angeles","Tokyo","London","Sydney","Singapore"]
|
||||||
|
|
||||||
|
app.set_widget(app.set_munubtn(app.get_menubtn('religion_mbtn'),"religion",religions),4,2,W)
|
||||||
|
app.set_widget(app.get_menubtn('plan_mbtn'),4,3,W)
|
||||||
|
|
||||||
|
def Refresh_Info():
|
||||||
|
app.get_widget('info_text').delete('1.0', END)
|
||||||
|
app.get_widget('info_text').insert(INSERT,"Account Info\n")
|
||||||
|
Balance = "Balance: "+str(account.balance)+"\n"
|
||||||
|
app.get_widget('info_text').insert(INSERT,Balance)
|
||||||
|
data = account.get_server_info()
|
||||||
|
for item in data:
|
||||||
|
server_data = data[item]
|
||||||
|
app.get_widget('info_text').insert(INSERT,"==========================\n")
|
||||||
|
app.get_widget('info_text').insert(INSERT,"Server "+server_data['SUBID']+"\n")
|
||||||
|
app.get_widget('info_text').insert(INSERT,"IP: "+server_data["main_ip"]+"\n")
|
||||||
|
app.get_widget('info_text').insert(INSERT,"Location: "+server_data["location"]+"\n")
|
||||||
|
app.get_widget('info_text').insert(INSERT,"OS: "+server_data["os"]+"\n")
|
||||||
|
app.get_widget('info_text').insert(INSERT,"Location: "+server_data["location"]+"\n")
|
||||||
|
app.get_widget('info_text').insert(INSERT,"Status: "+server_data["status"]+"\n")
|
||||||
|
try:
|
||||||
|
server_data["ssr_installation"]
|
||||||
|
except KeyError:
|
||||||
|
app.get_widget('info_text').insert(INSERT,"SSR: "+"None"+"\n")
|
||||||
|
else:
|
||||||
|
app.get_widget('info_text').insert(INSERT,"SSR: "+server_data["ssr_installation"]+"\n")
|
||||||
|
|
||||||
|
|
||||||
|
app.get_widget('info_text').insert(END," \n")
|
||||||
|
|
||||||
|
def Set_Api_Key():
|
||||||
|
API_Key = key_entry.get()
|
||||||
|
account.set_account(API_Key)
|
||||||
|
(back, stdout) = account.get_account()
|
||||||
|
if (back == "DONE"):
|
||||||
|
Refresh_Info()
|
||||||
|
else:
|
||||||
|
app.get_widget('info_text').delete('1.0', END)
|
||||||
|
app.get_widget('info_text').insert(INSERT,stdout);
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
def Add_Server():
|
||||||
|
if (account.CMD_Head != ""):
|
||||||
|
account.create_server('201','193','5');
|
||||||
|
Refresh_Info();
|
||||||
|
|
||||||
|
def Thread_ISSR(server_data):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
def Install_SSR():
|
||||||
|
|
||||||
|
SUBID = app.get_widget('sid_entry').get()
|
||||||
|
server_data = account.servers[SUBID]
|
||||||
|
vultr.connect_ssh(server_data)
|
||||||
|
app.get_widget('status_label').config(text = SUBID+" Installing UFW and SSR");
|
||||||
|
server_data['ssr_installation'] = "Installing"
|
||||||
|
#threading.Thread(target = Thread_ISSR,args = (server_data))
|
||||||
|
|
||||||
|
vultr.install_ssr(server_data)
|
||||||
|
app.get_widget('info_text').delete(END)
|
||||||
|
app.get_widget('status_label').insert(END,SUBID+" Installation Done.");
|
||||||
|
app.get_widget('status_label').config(text = SUBID+" UFW and SSR Installed");
|
||||||
|
Refresh_Info();
|
||||||
|
|
||||||
|
|
||||||
|
def Destroy_Server():
|
||||||
|
#display = Display("Display")
|
||||||
|
#account.destroy_server(sid_text.get())
|
||||||
|
SUBID = app.get_widget('sid_entry').get()
|
||||||
|
account.destroy_server(SUBID)
|
||||||
|
Refresh_Info()
|
||||||
|
|
||||||
|
app.set_size(app.set_widget(app.get_button("Connect",Set_Api_Key,'connect_button'),1,5,W),10,1)
|
||||||
|
app.set_size(app.set_widget(app.get_button("Add a Server",Add_Server,'add_button'),4,5,W),10,1)
|
||||||
|
app.set_size(app.set_widget(app.get_button("Refresh",Refresh_Info,'refresh_button'),2,5,W),10,1)
|
||||||
|
app.set_size(app.set_widget(app.get_button("Install SSR",Install_SSR,'install_button'),5,5,W),10,1)
|
||||||
|
app.set_size(app.set_widget(app.get_button("Destroy Server",Destroy_Server,'destroy_button'),6,5,W),10,1)
|
||||||
|
|
||||||
|
|
||||||
|
app.app.mainloop()
|
||||||
|
|
||||||
|
|
||||||
163
vultr.py
Normal file
163
vultr.py
Normal file
@ -0,0 +1,163 @@
|
|||||||
|
import os
|
||||||
|
import json
|
||||||
|
import subprocess
|
||||||
|
import paramiko
|
||||||
|
import time
|
||||||
|
|
||||||
|
class account:
|
||||||
|
API_Key = None;
|
||||||
|
balance = 0;
|
||||||
|
last_pay_date = None;
|
||||||
|
last_pay_amount = 0;
|
||||||
|
servers_subid = [];
|
||||||
|
servers = [];
|
||||||
|
|
||||||
|
CMD_CMD = None
|
||||||
|
CMD_INFO = None
|
||||||
|
CMD_OPTION = None
|
||||||
|
CMD_Head = None
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def set_account(self,API_Key):
|
||||||
|
self.API_Key = API_Key;
|
||||||
|
self.CMD_CMD = "curl"
|
||||||
|
self.CMD_OPTION = " -s"
|
||||||
|
self.CMD_INFO = " -H 'API-Key:"+API_Key+"' "
|
||||||
|
self.CMD_Head = self.CMD_CMD+self.CMD_OPTION+self.CMD_INFO
|
||||||
|
|
||||||
|
def do_command(self,CMD):
|
||||||
|
child = subprocess.Popen([CMD],shell=True,stdout=subprocess.PIPE,stderr=subprocess.PIPE);
|
||||||
|
child.wait();
|
||||||
|
(stdout,error) = child.communicate()
|
||||||
|
str_out = stdout.decode()
|
||||||
|
str_err = error.decode()
|
||||||
|
if (str_err != ''): print("Error: "+error)
|
||||||
|
return str_out;
|
||||||
|
|
||||||
|
def get_account(self):
|
||||||
|
CMD_HTML = "https://api.vultr.com/v1/account/info"
|
||||||
|
CMD = self.CMD_Head+CMD_HTML;
|
||||||
|
stdout = self.do_command(CMD);
|
||||||
|
try:
|
||||||
|
data = json.loads(stdout);
|
||||||
|
except json.decoder.JSONDecodeError:
|
||||||
|
print("Ask account information failed.")
|
||||||
|
print(stdout)
|
||||||
|
return "ERROR",stdout;
|
||||||
|
else:
|
||||||
|
self.balance = data["balance"];
|
||||||
|
self.last_pay_date = data["last_payment_date"];
|
||||||
|
self.last_pay_amount = data["last_payment_amount"];
|
||||||
|
return "DONE","NONE";
|
||||||
|
|
||||||
|
def create_server(self, plan, os, religion):
|
||||||
|
CMD_HTML = "https://api.vultr.com/v1/server/create"
|
||||||
|
CMD_DATA = " --data 'DCID="+religion+"' --data 'VPSPLANID="+plan+"' --data 'OSID="+os+"'"
|
||||||
|
CMD = self.CMD_Head+CMD_HTML+CMD_DATA;
|
||||||
|
stdout = self.do_command(CMD);
|
||||||
|
data = json.loads(stdout);
|
||||||
|
self.servers_subid.append(data["SUBID"]);
|
||||||
|
self.servers_subid =[]
|
||||||
|
self.get_server_info()
|
||||||
|
|
||||||
|
return data["SUBID"];
|
||||||
|
|
||||||
|
def create_ipv4(self,SUBID):
|
||||||
|
CMD_HTML = "https://api.vultr.com/v1/server/create_ipv4";
|
||||||
|
CMD_DATA = " --data 'SUBID="+SUBID+"'"
|
||||||
|
CMD = self.CMD_Head + CMD_HTML + CMD_DATA;
|
||||||
|
self.do_command(CMD)
|
||||||
|
|
||||||
|
def destroy_server(self,SUBID):
|
||||||
|
CMD_HTML = "https://api.vultr.com/v1/server/destroy";
|
||||||
|
CMD_DATA = " --data 'SUBID="+SUBID+"'"
|
||||||
|
CMD = self.CMD_Head + CMD_HTML + CMD_DATA;
|
||||||
|
stdout = self.do_command(CMD)
|
||||||
|
|
||||||
|
def destroy_ipv4(self,SUBID):
|
||||||
|
CMD_HTML = "https://api.vultr.com/v1/server/destroy_ipv4";
|
||||||
|
CMD_DATA = " --data 'SUBID="+SUBID+"'"
|
||||||
|
CMD = self.CMD_Head + CMD_HTML + CMD_DATA;
|
||||||
|
self.do_command(CMD)
|
||||||
|
|
||||||
|
|
||||||
|
def server_reboot(self,SUBSID):
|
||||||
|
CMD_HTML = "https://api.vultr.com/v1/server/reboot";
|
||||||
|
CMD_DATA = " --data 'SUBID="+SUBID+"'"
|
||||||
|
CMD = self.CMD_Head + CMD_HTML + CMD_DATA;
|
||||||
|
stdout = self.do_command(CMD)
|
||||||
|
|
||||||
|
def get_server_info(self):
|
||||||
|
CMD_HTML = "https://api.vultr.com/v1/server/list";
|
||||||
|
CMD = self.CMD_Head + CMD_HTML;
|
||||||
|
stdout = self.do_command(CMD);
|
||||||
|
data = json.loads(stdout);
|
||||||
|
self.servers = data
|
||||||
|
#print(data)
|
||||||
|
self.servers_subid = []
|
||||||
|
for item in data:
|
||||||
|
self.servers_subid.append(item)
|
||||||
|
return data;
|
||||||
|
def get_sever_ipv4(self):
|
||||||
|
CMD_HTML = "https://api.vultr.com/v1/server/list_ipv4";
|
||||||
|
CMD_DATA = "?SUBID="+SUBID
|
||||||
|
CMD = self.CMD_Head + CMD_HTML + CMD_DATA;
|
||||||
|
stdout = self.do_command(CMD)
|
||||||
|
|
||||||
|
def connect_ssh(server_data):
|
||||||
|
if (server_data["status"] == "active"):
|
||||||
|
ssh = paramiko.SSHClient()
|
||||||
|
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
||||||
|
ssh.connect(server_data["main_ip"],22,"root",server_data["default_password"],timeout=8)
|
||||||
|
server_data['ssh_connection'] = ssh;
|
||||||
|
|
||||||
|
return server_data
|
||||||
|
|
||||||
|
def install_ssr(server_data):
|
||||||
|
ssh = server_data['ssh_connection']
|
||||||
|
if(server_data["status"] == "active"):
|
||||||
|
print("Waiting for installation...")
|
||||||
|
|
||||||
|
print("Installing Ufw...")
|
||||||
|
stdin,stdout,stderr = ssh.exec_command("apt-get install ufw")
|
||||||
|
stdin.write("Y\n");
|
||||||
|
print(stdout.read())
|
||||||
|
print("Ufw installed.")
|
||||||
|
server_data['ufw_installation'] = "Installed";
|
||||||
|
|
||||||
|
time.sleep(3)
|
||||||
|
|
||||||
|
print("Setting Ufw...")
|
||||||
|
stdin,stdout,stderr = ssh.exec_command("ufw enable")
|
||||||
|
stdin.write("Y\n")
|
||||||
|
stdin,stdout,stderr = ssh.exec_command("ufw allow 9048")
|
||||||
|
stdin,stdout,stderr = ssh.exec_command("ufw allow 22")
|
||||||
|
stdin,stdout,stderr = ssh.exec_command("ufw default deny")
|
||||||
|
print(stdout.read())
|
||||||
|
print("Ufw have been set");
|
||||||
|
server_data['ufw_setting_allow'] = [22,9048];
|
||||||
|
|
||||||
|
time.sleep(3)
|
||||||
|
|
||||||
|
print("Setting SSR...")
|
||||||
|
stdin,stdout,stderr = ssh.exec_command("wget --no-check-certificate https://raw.githubusercontent.com/teddysun/shadowsocks_install/master/shadowsocksR.sh")
|
||||||
|
stdin,stdout,stderr = ssh.exec_command("chmod +x shadowsocksR.sh");
|
||||||
|
stdin,stdout,stderr = ssh.exec_command("./shadowsocksR.sh 2>&1 | tee shadowsocksR.log")
|
||||||
|
stdin.write("#a9b9fa3456\n");
|
||||||
|
stdin.write("9048\n2\n7\n7\n\n");
|
||||||
|
print(stdout.read())
|
||||||
|
stdin,stdout,stderr = ssh.exec_command("systemctl start shadowsocks")
|
||||||
|
print("SSR have been set and now is running.")
|
||||||
|
print("Server IP :"+server_data["main_ip"])
|
||||||
|
server_data['ssr_installation'] = "Installed";
|
||||||
|
return "DONE"
|
||||||
|
|
||||||
|
def print_server(server_data):
|
||||||
|
print(server_data["SUBID"])
|
||||||
|
print(server_data["main_ip"])
|
||||||
|
print(server_data["location"])
|
||||||
|
print(server_data["os"])
|
||||||
|
print(server_data["status"])
|
||||||
|
print(server_data["default_password"])
|
||||||
Loading…
Reference in New Issue
Block a user