jaydee/python
1
0
Fork 0
mirror of https://gitlab.sectorq.eu/jaydee/python.git synced 2025-12-13 18:24:53 +01:00
Code Issues Packages Projects Releases Wiki Activity

bitwarden

Browse Source
This commit is contained in:
jaydee
2024-10-23 00:26:48 +02:00
parent c67be7706a
commit 2e0793fda1
1 changed files with 244 additions and 207 deletions
Download Patch File Download Diff File Expand all files Collapse all files

451
mqtt_srv.py
View File

@@ -19,58 +19,52 @@ import random
import requests
#import psutil
stats = {}
VERSION = "1.0.23"
curos = platform.system()
VERSION = "1.0.33"
curos = platform.system()
host = platform.node().lower()
print(host)
if curos == "Windows":
import winreg
def uptime():
try:
f = open( "/proc/uptime" )
contents = f.read().split()
f.close()
except:
try:
f = open( "/proc/uptime" )
contents = f.read().split()
f.close()
except:
return "Cannot open uptime file: /proc/uptime"
total_seconds = float(contents[0])
# Helper vars:
MINUTE = 60
HOUR = MINUTE * 60
DAY = HOUR * 24
# Get the days, hours, etc:
days = int( total_seconds / DAY )
hours = int( ( total_seconds % DAY ) / HOUR )
minutes = int( ( total_seconds % HOUR ) / MINUTE )
seconds = int( total_seconds % MINUTE )
total_seconds = float(contents[0])
# Helper vars:
MINUTE = 60
HOUR = MINUTE * 60
DAY = HOUR * 24
# Get the days, hours, etc:
days = int( total_seconds / DAY )
hours = int( ( total_seconds % DAY ) / HOUR )
minutes = int( ( total_seconds % HOUR ) / MINUTE )
seconds = int( total_seconds % MINUTE )
# Build up the pretty string (like this: "N days, N hours, N minutes, N seconds")
string = ""
string = ""
# if days > 0:
# string += str(days) + " " + (days == 1 and "d" or "d" ) + ", "
# if len(string) > 0 or hours > 0:
# string += str(hours) + " " + (hours == 1 and "h" or "h" ) + ", "
# if len(string) > 0 or minutes > 0:
# string += str(minutes) + " " + (minutes == 1 and "m" or "m" ) + ", "
if days > 0:
string += str(days) + " " + (days == 1 and "d" or "d" ) + ", "
if len(string) > 0 or hours > 0:
if hours < 10:
hours = "0{}".format(hours)
string += str(hours) + ":"
if len(string) > 0 or minutes > 0:
if minutes < 10:
minutes = "0{}".format(minutes)
string += str(minutes) + ":"
if seconds < 10:
seconds = "0{}".format(seconds)
string += str(seconds)
return string;
if days > 0:
string += str(days) + " " + (days == 1 and "d" or "d" ) + ", "
if len(string) > 0 or hours > 0:
string += str(hours) + ":"
if len(string) > 0 or minutes > 0:
string += str(minutes) + ":"
string += str(seconds)
return string;
def writeLog(msg, svr="INFO"):
ts = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
if curos == "Windows":
@@ -79,32 +73,37 @@ def writeLog(msg, svr="INFO"):
log_path = "/tmp/mqtt_srv.log"
f = open(log_path, "a")
f.write(ts + " | " + svr + " | " + str(msg) + "\n")
f.close()
f.close()
def check_router():
stats["uptime"] = uptime()
stats["version"] = VERSION
stats["status"] = "on"
cmnd = "crontab -l"
status, output = subprocess.getstatusoutput(cmnd)
if "#Skynet_banmalware#" in output:
stats["skynet"] = "on"
stats["skynet"] = "On"
else:
stats["skynet"] = "off"
stats["skynet"] = "Off"
if "#Diversion_CountAds#" in output:
stats["diversion"] = "on"
stats["diversion"] = "On"
else:
stats["diversion"] = "off"
stats["diversion"] = "Off"
cmnd = "nvram get vpn_client1_state"
status, output = subprocess.getstatusoutput(cmnd)
writeLog("Openvpn1 Status : " + str(output))
stats["vpnclient1"] = output
writeLog("VpnClient1 Status : " + str(output))
stats["vpnclient1"] = "On" if output == "2" else "Off"
cmnd = "nvram get vpn_client2_state"
status, output = subprocess.getstatusoutput(cmnd)
writeLog("Openvpn2 Status : " + str(output))
stats["vpnclient2"] = output
writeLog("VpnClient2 Status : " + str(output))
stats["vpnclient2"] = "On" if output == "2" else "Off"
cmnd = "nvram get vpn_server1_state"
status, output = subprocess.getstatusoutput(cmnd)
writeLog("VpnServer1 Status : " + str(output))
stats["vpnserver1"] = "On" if output == "2" else "Off"
cmnd = "ncat -z 192.168.77.106 25"
status, output = subprocess.getstatusoutput(cmnd)
if status == 0:
@@ -128,16 +127,15 @@ def check_router():
serv_d[rul[0]] = {"src_port":rul[1],"dest_ip":rul[2],"dest_port":rul[3],"prot":rul[4]}
stats["serv_dest"] = rul[2]
break
stats["status"] = "on"
#stats["serv_dest"] = json.dumps(serv_d)
#stats["serv_dest"] = json.dumps(serv_d)
#out = subprocess.Popen(cmnd.split())
#status, output = subprocess.getstatusoutput(cmnd)
'''
print("Getting fw info")
cmnd = "/root/router_cmd.sh -t firewall -n TOSHIBA-WIFI -m get"
#out = subprocess.Popen(cmnd.split())
@@ -156,15 +154,14 @@ def check_router():
# if output == "allow":\
writeLog("Payload : " + str(json.dumps(stats)))
'''
stats["status"] = "on"
client.publish("home-assistant/router/hwstats", json.dumps(stats), qos=0, retain=True)
#client.publish("sectorq/systems/router/stats", payload=json.dumps(stats), qos=0, retain=True)
#else:
# client.publish("sectorq/systems/router/macfilter", payload=0, qos=0, retain=True)
if host != "router" and host != "nas":
import psutil
processes = list(p.name() for p in psutil.process_iter())
# print(processes)
count = processes.count("mqtt_srv.exe")
@@ -178,12 +175,8 @@ else:
if int(count) >= 2:
writeLog("script exist")
sys.exit()
# Stops duplicate instance from running
writeLog(count)
writeLog("Starting : " + str(VERSION))
@@ -222,13 +215,13 @@ for o, a in opts:
_MESSAGE = a
elif o in ("-t", "--publish"):
_TOPIC = a
elif o in ("-S", "--setvar"):
elif o in ("-S", "--setvar"):
_SETVAR = True
_THR_NAME = sys.argv[2]
_THR_VAL = sys.argv[3]
else:
_WIZZARD = True
MQTT_HOST = "192.168.77.106"
MQTT_HOST = "mqtt.lan"
MQTT_PORT = 1883
if _PUBLISH:
@@ -318,15 +311,19 @@ def get_ip():
finally:
s.close()
return IP
IP = get_ip()
ex = 0
while ex == 0:
try:
IP = get_ip()
ex = 1
except:
time.sleep(5)
print("OS : " + curos)
writeLog("OS : " + curos)
if curos != "Windows":
import autorandr
import autorandr
else:
import winreg
@@ -337,6 +334,8 @@ elif host == "SERVER":
mac = str(get_mac_address("enp2s0"))
elif host == "ASUS" and curos != "Windows":
mac = str(get_mac_address("enp3s0f1"))
elif host == "morefine" and curos != "Windows":
mac = str(get_mac_address("eno1"))
else:
mac = str(get_mac_address("Ethernet"))
@@ -365,14 +364,14 @@ def on_connect(client, userdata, flags, rc):
client.publish("sectorq/systems/" + host.lower() + "/stat", payload="on", qos=0, retain=True)
if host.lower() == "router":
check_router()
if host.lower() == "nas":
print("Getting VM info")
cmnd = "/share/ZFS530_DATA/.qpkg/QKVM/usr/bin/virsh list --all"
cmnd = "/share/ZFS532_DATA/.qpkg/QKVM/usr/bin/virsh list --all"
print(cmnd)
status, output = subprocess.getstatusoutput(cmnd)
l = 1
l = 1
lines = output.splitlines()
VMS = {}
for i in lines:
@@ -382,12 +381,12 @@ def on_connect(client, userdata, flags, rc):
line = re.split(r" {2,}", i)
print(str(line[0].strip()) + " " + str(line[1].strip()) + " " + str(line[2].strip()))
VMS[str(line[1].strip())] = str(line[2].strip())
for m in VMS:
cmnd = "/share/ZFS530_DATA/.qpkg/QKVM/usr/bin/virsh domstate " + m
cmnd = "/share/ZFS532_DATA/.qpkg/QKVM/usr/bin/virsh domstate " + m
#out = subprocess.Popen(cmnd.split())
print(cmnd)
status, output = subprocess.getstatusoutput(cmnd)
@@ -396,7 +395,7 @@ def on_connect(client, userdata, flags, rc):
client.publish("sectorq/systems/nas/" + m, payload=1, qos=0, retain=True)
else:
client.publish("sectorq/systems/nas/" + m, payload=0, qos=0, retain=True)
client.subscribe(topic)
@@ -424,14 +423,14 @@ def on_message(client, userdata, msg):
except:
print("Action not set")
writeLog("Action not set", "WARNING")
try:
print(myObj["cmd"])
writeLog(myObj["cmd"])
except:
print("Cmd not set")
writeLog("Cmd not set", "WARNING")
try:
print(myObj["target"])
writeLog(myObj["target"])
@@ -451,7 +450,7 @@ def on_message(client, userdata, msg):
except:
print("Command Not Set")
writeLog("Command Not Set")
try:
print(myObj["cmd_line"])
writeLog(myObj["cmd_line"])
@@ -473,10 +472,11 @@ def on_message(client, userdata, msg):
ctypes.windll.user32.MessageBoxW(0, "Your text", "Your title", 16)
print("Executed " + myObj["cmd_line"])
writeLog("Executed " + myObj["cmd_line"])
if myObj["cmd"] == "nas_cmd":
print("Executing NAS")
cmnd = "/share/ZFS530_DATA/.qpkg/QKVM/usr/bin/virsh domstate " + str(myObj["target"])
#cmnd = "/share/ZFS530_DATA/.qpkg/QKVM/usr/bin/virsh domstate " + str(myObj["target"])
cmnd = "/share/ZFS532_DATA/.qpkg/QKVM/usr/bin/virsh domstate " + str(myObj["target"])
print(cmnd)
status, output = subprocess.getstatusoutput(cmnd)
print(output)
@@ -490,9 +490,9 @@ def on_message(client, userdata, msg):
cmd_type = "start"
if myObj["command"] == "suspend":
cmd_type = "shutdown"
cmnd = "/share/ZFS530_DATA/.qpkg/QKVM/usr/bin/virsh " + cmd_type + " " + str(myObj["target"])
cmnd = "/share/ZFS532_DATA/.qpkg/QKVM/usr/bin/virsh " + cmd_type + " " + str(myObj["target"])
subprocess.Popen(cmnd.split())
print(cmnd)
if myObj["command"] == "start":
@@ -509,14 +509,14 @@ def on_message(client, userdata, msg):
except:
print("failed to execute!")
client.publish("sectorq/systems/nas/" + str(myObj["target"]), payload="failed", qos=0, retain=False)
if myObj["cmd"] == "distrib":
print("distributin313g")
# os.startfile(sys.argv[0])
#sys.exit()
subprocess.Popen(['/bin/systemctl', 'restart', 'mqtt'])
if myObj["cmd"] == "uci":
print("uci1")
print("Executing " + myObj["cmd_line"])
@@ -533,7 +533,7 @@ def on_message(client, userdata, msg):
except:
print("failed to execute!")
client.publish("sectorq/systems/router/stats", payload="failed", qos=0, retain=False)
elif myObj["cmd_line"] == "wireless0":
cmnd = "/root/router_cmd.sh -t wireless -v 0 -m set"
try:
@@ -573,10 +573,10 @@ def on_message(client, userdata, msg):
cmnd = "service {}".format(myObj["target"])
print(cmnd)
status, output = subprocess.getstatusoutput(cmnd)
elif myObj["cmd_line"] == "serv_dest":
writeLog("Payload111111 : " + myObj["target"].lower())
cmnd = "nvram get vts_rulelist"
status, output = subprocess.getstatusoutput(cmnd)
rule_data = output.split("<")
@@ -604,7 +604,7 @@ def on_message(client, userdata, msg):
serv_d[r]["dest_ip"] = myObj["target"]
except:
serv_d[r] = rules[r]
lala = []
for sr in serv_d:
lala.append("{}>{}>{}>{}>{}>".format(sr,serv_d[sr]["src_port"],serv_d[sr]["dest_ip"],serv_d[sr]["dest_port"],serv_d[sr]["prot"]))
@@ -618,7 +618,7 @@ def on_message(client, userdata, msg):
status, output = subprocess.getstatusoutput(cmnd)
cmnd = "service restart_firewall"
status, output = subprocess.getstatusoutput(cmnd)
writeLog("Payload111114")
writeLog("Payload111114")
writeLog(output)
print(json.dumps(serv_d))
'''
@@ -626,29 +626,29 @@ def on_message(client, userdata, msg):
dest_ip = "192.168.77.246"
elif myObj["target"].lower() == "192.168.77.106":
dest_ip = "192.168.77.106"
RULES = ["EMAIL993", "EMAIL995", "EMAIL143", "EMAIL25", "EMAIL993", "EMAIL465", "HTTPS", "HTTP"]
for i in RULES:
cmnd = "uci set firewall.{}.dest_ip={}".format(i,dest_ip)
writeLog(cmnd)
subprocess.Popen(cmnd.split())
#cmnd = "uci set dhcp.@dnsmasq[0].address=/mqtt_broker/" + dest_ip
#writeLog(cmnd)
#subprocess.Popen(cmnd.split())
time.sleep(1)
cmnd = "uci commit"
subprocess.Popen(cmnd.split())
writeLog("Starting " + myObj["target"].lower())
if myObj["target"].lower() == "192.168.77.246":
dest_ip = "192.168.77.246"
cmnd = "ssh root@192.168.77.246 'docker start mailu_resolver_1 mailu_redis_1 mailu_front_1 mailu_antispam_1 mailu_imap_1 mailu_admin_1 mailu_smtp_1 mailu_webmail_1 watchtower_watchtower_1 nginx-proxy-manager_db_1 nginx-proxy-manager_app_1 HomeAssistant webhub_client_1 dockermon_docker_mon_1 mosquitto_mosquitto_1 node-red_node-red_1 nextcloud_db_1 nextcloud_app_1'"
status, output = subprocess.getstatusoutput(cmnd)
elif myObj["target"].lower() == "192.168.77.106":
@@ -664,7 +664,7 @@ def on_message(client, userdata, msg):
MQTT_BROKER = dest_ip
MQTT_USER = "jaydee"
MQTT_PASS = "jaydee1"
cmnd = "nmap -sP 192.168.77.*|grep \"Nmap scan report\"|egrep -o \"[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\""
#print(cmnd)
status, output = subprocess.getstatusoutput(cmnd)
@@ -681,11 +681,11 @@ def on_message(client, userdata, msg):
if status == 0:
try:
# url = "http://" + sys_ip + "/cm?user=admin&password=l4c1j4yd33Du5l0&cmnd=STATUS+5"
url = "http://" + sys_ip + "/cm?cmnd=Backlog%20MqttHost%20" + MQTT_BROKER + "%3BMqttUser%20" + MQTT_USER + "%3BMqttPassword%20" + MQTT_PASS
print(url)
requests.get(url)
except:
#print(sys_ip + " : Not a tasmota!")
pass
@@ -693,8 +693,8 @@ def on_message(client, userdata, msg):
#print(sys_ip + " : Not a listening")
pass
time.sleep(1)
cmnd = "sed -i 's/^MQTT_HOST = .*/MQTT_HOST = \"" + dest_ip + "\"/' /root/mqtt_srv.py"
writeLog(cmnd)
@@ -703,13 +703,13 @@ def on_message(client, userdata, msg):
else:
pass
#cmnd = "/root/router_cmd.sh -t wireless -v 0"
print("Executed " + cmnd)
writeLog("Executed " + cmnd)
'''
if myObj["cmd"] == "banip":
if myObj["status"] == "ban":
cmnd = "echo " + str(myObj["target"]) + " >> /etc/banip/banip.blacklist; /etc/init.d/banip refresh"
@@ -721,13 +721,13 @@ def on_message(client, userdata, msg):
stats["banip"] = "paused"
writeLog(cmnd)
status, output = subprocess.getstatusoutput(cmnd)
client.publish("sectorq/systems/router/stats", payload=json.dumps(stats), qos=0, retain=True)
if myObj["cmd"] == "vnc":
print(myObj["target"])
writeLog(myObj["target"])
# os.startfile(sys.argv[0])
#sys.exit()
myCmd = 'c:\\Program Files\\RealVNC\\VNC Viewer\\vncviewer.exe C:\\Users\\jaydee\\' + myObj["target"].upper() + '.vnc'
@@ -744,7 +744,7 @@ def on_message(client, userdata, msg):
client.publish("sectorq/monitor/" + host.lower(), payload="{\"mb_temp\":0,\"mem_usage\":0,\"cpu_temp\":0,\"cpu_usage\":0,\"gpu_temp\":0,\"gpu_usage\":0}", qos=0, retain=False)
print("lalaa2")
if myObj["cmd"] == "display":
if curos == "Windows1":
# os.startfile(sys.argv[0])
@@ -762,21 +762,21 @@ def on_message(client, userdata, msg):
myCmd = ""
if myObj["target"] == "mid":
myCmd = "xrandr --output HDMI-2 --primary --auto --mode 1920x1080 --output DVI-I-2-1 --off --output HDMI-1 --off --output eDP-1 --off"
elif myObj["target"] == "all":
elif myObj["target"] == "all":
myCmd = "xrandr --output HDMI-2 --primary --auto --mode 1920x1080 --output DVI-I-2-1 --auto --above HDMI-2 --mode 1920x1080 --output HDMI-1 --auto --left-of HDMI-2 --mode 1920x1080 --output eDP-1 --auto --right-of HDMI-2 --mode 1920x1080"
elif myObj["target"] == "midleft":
elif myObj["target"] == "midleft":
myCmd = "xrandr --output HDMI-2 --primary --auto --mode 1920x1080 --output DVI-I-2-1 --off --output HDMI-1 --auto --left-of HDMI-2 --mode 1920x1080 --output eDP-1 --off"
elif myObj["target"] == "midtop":
elif myObj["target"] == "midtop":
myCmd = "xrandr --output HDMI-2 --primary --auto --mode 1920x1080 --output DVI-I-2-1 --auto --above HDMI-2 --mode 1920x1080 --output HDMI-1 --off --output eDP-1 --off"
elif myObj["target"] == "midleftbuild":
myCmd = "xrandr --output HDMI-2 --primary --auto --mode 1920x1080 --output DVI-I-2-1 --off --output HDMI-1 --auto --left-of HDMI-2 --mode 1920x1080 --output eDP-1 --auto --right-of HDMI-2 --mode 1920x1080"
elif myObj["target"] == "midbuild":
elif myObj["target"] == "midleftbuild":
myCmd = "xrandr --output HDMI-2 --primary --auto --mode 1920x1080 --output DVI-I-2-1 --off --output HDMI-1 --auto --left-of HDMI-2 --mode 1920x1080 --output eDP-1 --auto --right-of HDMI-2 --mode 1920x1080"
elif myObj["target"] == "midbuild":
myCmd = "xrandr --output HDMI-2 --primary --auto --mode 1920x1080 --output DVI-I-2-1 --off --output HDMI-1 --off --output eDP-1 --auto --right-of HDMI-2 --mode 1920x1080"
elif myObj["target"] == "midlefttop":
elif myObj["target"] == "midlefttop":
myCmd = "xrandr --output HDMI-2 --primary --auto --mode 1920x1080 --output DVI-I-2-1 --auto --above HDMI-2 --mode 1920x1080 --output HDMI-1 --auto --left-of HDMI-2 --mode 1920x1080 --output eDP-1 --off"
else:
pass
pass
if myCmd != "":
print(myCmd)
subprocess.Popen(myCmd.split())
@@ -794,37 +794,38 @@ def on_message(client, userdata, msg):
mon3 = "DVI-I-2-1"
mon4 = "eDP-1-1"
'''
mon1 = "HDMI-1"
mon2 = "HDMI-2"
mon3 = "DVI-I-2-1"
mon4 = "eDP-1"
right = ["HDMI-A-0", "1920x1080"]
tleft = ["HDMI-A-0", "1920x1080"]
tright = ["DVI-I-1-1", "1920x1080"]
left = ["DisplayPort-0", "2560x1440"]
mid = ["DisplayPort-1", "3440x1440"]
#xrandr --output DisplayPort-1 --primary --auto --mode 3440x1440 --output HDMI-A-0 --auto --left-of DVI-I-1-1 --mode 1920x1080 --output DisplayPort-0 --auto --left-of DisplayPort-1 --mode 2560x1440 --output DVI-I-1-1 --above DisplayPort-1 --mode 1920x1080
if myObj["target"] == "mid":
myCmd = "xrandr --output " + mon2 + " --primary --auto --mode 1920x1080 --output " + mon3 + " --off --output " +mon1 + " --off --output " + mon4 + " --off"
elif myObj["target"] == "left":
myCmd = "xrandr --output " + mon1 + " --primary --auto --mode 1920x1080 --output " + mon3 + " --off --output " +mon2 + " --off --output " + mon4 + " --off"
myCmd = f"xrandr --output {mid[0]} --primary --auto --mode {mid[1]} --output {left[0]} --off --output {tleft[0]} --off --output {tright[0]} --off"
elif myObj["target"] == "left":
myCmd = f"xrandr --output {left[0]} --primary --auto --mode {left[1]} --output {mid[0]} --off --output {tleft[0]} --off --output {tright[0]} --off"
elif myObj["target"] == "all":
myCmd = "xrandr --output " + mon2 + " --primary --auto --mode 1920x1080 --output " + mon3 + " --auto --right-of " + mon2 + " --mode 1920x1080 --output " + mon1 + " --auto --left-of " + mon2 + " --mode 1920x1080 --output " + mon4 + " --auto --right-of " + mon2 + " --mode 1920x1080"
elif myObj["target"] == "midleft":
myCmd = "xrandr --output " + mon2 + " --primary --auto --mode 1920x1080 --output " + mon3 + " --off --output " + mon1 + " --auto --left-of " + mon2 + " --mode 1920x1080 --output " + mon4 + " --off"
elif myObj["target"] == "midtop":
myCmd = "xrandr --output " + mon2 + " --primary --auto --mode 1920x1080 --output " + mon3 + " --auto --right-of " + mon2 + " --mode 1920x1080 --output " + mon1 + " --off --output " + mon4 + " --off"
elif myObj["target"] == "midleftbuild":
myCmd = "xrandr --output " + mon2 + " --primary --auto --mode 1920x1080 --output " + mon3 + " --off --output " + mon1 + " --auto --left-of " + mon2 + " --mode 1920x1080 --output " + mon4 + " --auto --right-of " + mon2 + " --mode 1920x1080"
elif myObj["target"] == "midbuild":
myCmd = "xrandr --output " + mon2 + " --primary --auto --mode 1920x1080 --output " + mon3 + " --off --output " + mon1 + " --off --output " + mon4 + " --auto --right-of " + mon2 + " --mode 1920x1080"
elif myObj["target"] == "midlefttop":
myCmd = "xrandr --output " + mon2 + " --primary --auto --mode 1920x1080 --output " + mon3 + " --auto --right-of " + mon2 + " --mode 1920x1080 --output " + mon1 + " --auto --left-of " + mon2 + " --mode 1920x1080 --output " + mon4 + " --off"
elif myObj["target"] == "build":
myCmd = "xrandr --output " + mon2 + " --off --output " + mon3 + " --off --output " + mon1 + " --off --output " + mon4 + " --auto --mode 1920x1080"
myCmd = f"xrandr --output {mid[0]} --primary --auto --mode {mid[1]} --pos 0x0 --output {tleft[0]} --auto --pos -1920x-1080 --mode {tleft[1]} --output {left[0]} --auto --left-of {mid[0]} --mode {left[1]} --output {tright[0]} --above {mid[0]} --mode {tright[1]}"
#myCmd = f"xrandr --output {mid[0]} --primary --auto --mode {mid[1]} --output {right[0]} --auto --right-of {mid[0]} --mode {right[1]} --output {left[0]} --auto --left-of {mid[0]} --mode {left[1]}"
elif myObj["target"] == "midleft":
myCmd = f"xrandr --output {mid[0]} --primary --auto --mode {mid[1]} --output {tleft[0]} --off --output {left[0]} --auto --left-of {mid[0]} --mode {left[1]} --output {tright[0]} --off"
elif myObj["target"] == "midtop":
myCmd = f"xrandr --output {mid[0]} --primary --auto --mode {mid[1]} --output {tright[0]} --auto --above {mid[0]} --mode {tright[1]} --output {left[0]} --off --output {tleft[0]} --off"
elif myObj["target"] == "midlefttleft":
myCmd = f"xrandr --output {mid[0]} --primary --auto --mode {mid[1]} --pos 0x0 --output {tright[0]} --off --output {left[0]} --auto --left-of {mid[0]} --mode {left[1]} --output {tleft[0]} --auto --pos -1920x-1080 --mode {tleft[1]}"
elif myObj["target"] == "lefttlefttright":
myCmd = f"xrandr --output {mid[0]} --off --output {tleft[0]} --auto --pos 640x-1080 --mode {tright[1]} --output {left[0]} --primary --auto --mode {left[1]} --pos 0x0 --output {tright[0]} --right-of {tleft[0]} --mode {tright[1]}"
elif myObj["target"] == "lefttleft":
myCmd = f"xrandr --output {mid[0]} --off --output {tleft[0]} --auto --pos 640x-1080 --mode {tright[1]} --output {left[0]} --primary --auto --mode {left[1]} --pos 0x0 --output {tright[0]} --off"
else:
pass
pass
writeLog(myCmd)
if myCmd != "":
print(myCmd)
writeLog("Executing")
writeLog("Executing : {myCmd}")
#subprocess.Popen(myCmd.split())
status, output = subprocess.getstatusoutput(myCmd)
writeLog(output)
@@ -833,12 +834,12 @@ def on_message(client, userdata, msg):
# os.startfile(sys.argv[0])
#sys.exit()
#print("reconfigure")
subprocess.Popen(['MultiMonitorTool.exe', '/SaveConfig', 'c:\\Program Files\\jaydee\\switcher.cfg'])
subprocess.Popen(['MultiMonitorTool.exe', '/SaveConfig', 'c:\\Program Files\\jaydee\\switcher.cfg'])
myCmd = "powershell switcher.ps1 " + str(myObj["target"]) + " " + str(myObj["status"])
writeLog(myCmd)
subprocess.Popen(myCmd.split())
time.sleep(1)
subprocess.Popen(['MultiMonitorTool.exe', '/LoadConfig', 'c:\\Program Files\\jaydee\\switcher.cfg'])
subprocess.Popen(['MultiMonitorTool.exe', '/LoadConfig', 'c:\\Program Files\\jaydee\\switcher.cfg'])
client.publish("sectorq/feedback", payload='{"action":"resp", "source":"ASUS", "dev":{"display' + str(myObj["target"]) + '":'+ str(myObj["status"]) +'} }' , qos=0, retain=False)
if myObj["cmd"] == "displ":
print("Display")
@@ -848,7 +849,7 @@ def on_message(client, userdata, msg):
if myObj["cmd"] == "xrandr":
if curos == "Windows":
subprocess.Popen(['MultiMonitorTool.exe', '/LoadConfig', 'c:\\Program Files\\jaydee\\all_on.cfg'])
subprocess.Popen(['MultiMonitorTool.exe', '/LoadConfig', 'c:\\Program Files\\jaydee\\all_on.cfg'])
else:
print("Display")
# os.startfile(sys.argv[0])
@@ -871,8 +872,11 @@ def on_message(client, userdata, msg):
else:
if host == "nas" or host == "octopi":
subprocess.Popen(["/sbin/poweroff"])
elif host == "openmediavault" or host == "omv":
elif host == "openmediavault" or host == "omv" or host == "amd":
subprocess.Popen(["poweroff"])
elif host == "amd" or host == "ryzen":
subprocess.Popen(["systemctl suspend"])
else:
#subprocess.Popen(["/bin/systemctl", "suspend", "-i"])
#subprocess.Popen(["sudo", "pm-suspend"])
@@ -928,19 +932,29 @@ def on_message(client, userdata, msg):
print("not a json!")
client = mqtt.Client()
<<<<<<< HEAD
client.username_pw_set("jaydee", password="jaydee1")
client.will_set("home-assistant/" + host + "/hwstats", payload='{"status":"off"}', qos=0, retain=True)
=======
payload = '{"status":"off"}'
payload = '{"status": "off"}'
client.username_pw_set("jaydee", password="jaydee1")
client.will_set("home-assistant/" + host.lower() + "/hwstats", payload=payload, qos=0, retain=True)
>>>>>>> f8ca3c609387acf650821dc35de80b826724ee5f
client.on_connect = on_connect
client.on_message = on_message
writeLog(MQTT_HOST)
client.connect(MQTT_HOST, MQTT_PORT, 25)
connected = False
while not connected:
try:
client.connect(MQTT_HOST, MQTT_PORT, 25)
connected = True
except:
writeLog("Failed to connect repeating...")
time.sleep(10)
# client.publish("sectorq/status", payload="{'action':'connect', 'system':'" + host + "', 'ip': '" + IP + "', 'mac':'" + mac + "', 'ver':'?'}", qos=0, retain=False)
# Blocking call that processes network traffic, dispatches callbacks and
# handles reconnecting.
# Other loop*() functions are available that give a threaded interface and a
# manual interface.
def checkIfProcessRunning(processName):
'''
@@ -988,14 +1002,15 @@ if curos == "Windows":
'TMOBO':'MB_TEMP'
}
#ks = ["TCPU", "TPCHDIO", "SMEMUTI", "SVIRTMEMUTI", "SCPUUTI", "FCPU", "TGPU1DIO", "SGPU1BIUTI"]
path = winreg.HKEY_CURRENT_USER
try:
key = winreg.OpenKeyEx(path, r"SOFTWARE\\FinalWire\\AIDA64\\SensorValues")
results = {}
results["status"] = "on"
for k in aide_refs:
k2 = "Value." + str(k)
try:
@@ -1004,12 +1019,11 @@ if curos == "Windows":
results[aide_refs[k]] = value[0]
except:
pass
results["status"] = "on"
except:
print("aida initializing...")
results = {"status":"init"}
else:
print("CPU Cnt : {}".format(psutil.cpu_count()))
print(psutil.virtual_memory())
@@ -1021,7 +1035,8 @@ if curos == "Windows":
print(swap_data)
print("MEM : {}".format(mem_data.percent))
print(psutil.getloadavg())
try:
#lodctr /R
try:
print(psutil.sensors_temperatures())
except:
pass
@@ -1029,15 +1044,12 @@ if curos == "Windows":
print(psutil.sensors_fans())
except:
pass
<<<<<<< HEAD
results = {"status":"on", "version":VERSION,"MB_TEMP":"NA", "MEM_UTIL":str(mem_data.percent), "CPU_TEMP":"NA", "FAN_CPU":"NA", "VMEM_UTIL":swap_data.percent, "CPU_UTIL":cpu_usage, "GPU_TEMP":"NA", "GPU_UTIL":"NA"}
=======
results = {"status":"on","version":VERSION,"MB_TEMP":"NA", "MEM_UTIL":str(mem_data.percent), "CPU_TEMP":"NA", "FAN_CPU":"NA", "VMEM_UTIL":swap_data.percent, "CPU_UTIL":cpu_usage, "GPU_TEMP":"NA", "GPU_UTIL":"NA"}
>>>>>>> fba8088e24f4b4228930b1a72ca37b712d90c824
print(json.dumps(results))
r = json.dumps(results)
#client.publish("home-assistant/" + host + "/hwstats", payload='{"TPCHDIO":"' + str(results["TPCHDIO"]) + '", "SMEMUTI":"' + str(results["SMEMUTI"]) + '", "TCPU":"' + str(results["TCPU"]) + '", "TGPU1DIO":"' + str(results["TGPU1DIO"]) + '", "SGPU1BIUTI":"' + str(results["SGPU1BIUTI"]) + '", "FCPU":"' + str(results["FCPU"]) + '", "SVIRTMEMUTI":"' + str(results["SVIRTMEMUTI"]) + '", "SCPUUTI":"' + str(results["SCPUUTI"]) + '"}', qos=0, retain=False)
client.publish("home-assistant/" + host + "/hwstats", payload=r, qos=0, retain=False)
client.publish("sectorq/systems/" + host.lower() + "/stat", payload="on", qos=0, retain=True)
#print(json.dumps(stats))
#writeLog(json.dumps(stats))
else:
@@ -1046,30 +1058,60 @@ else:
NEXT_CHECK = 0
while True:
print(VERSION)
#print(VERSION)
time.sleep(5)
try:
mtime = os.path.getmtime(sys.argv[0])
except OSError:
mtime = 0
print(mtime)
#print(mtime)
if NEXT_CHECK < int(time.time()):
NEXT_CHECK = int(time.time()) + 20
writeLog("Blalala : Next : " + str(NEXT_CHECK))
if host.lower() == "asus":
if host.lower() == "asus" or host.lower() == "morefine" or host.lower() == "ryzen" or host.lower() == "amd" or host.lower() == "rack":
writeLog("Blalala : AAA")
print(psutil.sensors_temperatures())
#print(psutil.sensors_fans())
sences = psutil.sensors_temperatures()
cpu_data = sences["coretemp"][0]
mb_data = sences["acpitz"][0]
fan_data = psutil.sensors_fans()["asus"][0]
print("SENCES")
print(sences)
print("#"*40)
try:
cpu_data = sences["coretemp"][0]
except:
try:
cpu_data = sences["k10temp"][0]
except:
cpu_data = "None"
writeLog(cpu_data)
print("CPUDATA")
print(cpu_data)
print("#"*40)
print("MBDATA")
try:
mb_data = sences["acpitz"][0]
print(mb_data)
print(mb_data.current)
mb_data = mb_data.current
except:
mb_data = 0
print("#"*40)
try:
print(psutil.sensors_fans())
fan_data = psutil.sensors_fans()["asus"][0].current
except:
fan_data = "None"
mem_data = psutil.virtual_memory()
swap_data = psutil.swap_memory()
#cpu_usage = psutil.cpu_percent(interval=1)
load1, load5, load15 = psutil.getloadavg()
cpu_usage = round((load1/os.cpu_count()) * 100, 1)
print(mem_data)
print(mem_data.percent)
print(psutil.disk_usage('/'))
@@ -1080,16 +1122,13 @@ else:
#print(output)
#print(type(output))
#jn1 = json.loads(output)
#print(str(jn1["coretemp-isa-0000"]["Package id 0"]["temp1_input"]))
#print(str(jn1["asus-isa-0000"]["cpu_fan"]["fan1_input"]))
#client.publish("sectorq/monitor/" + host.lower(), payload="{\"mb_temp\":0,\"mem_usage\":0,\"cpu_temp\":0,\"cpu_usage\":0,\"gpu_temp\":0,\"gpu_usage\":0}", qos=0, retain=False)
<<<<<<< HEAD
client.publish("home-assistant/" + host + "/hwstats", payload='{"status":"on","version":"' + VERSION + '","MB_TEMP":' + str(mb_data.current) + ', "MEM_UTIL":' + str(mem_data.percent) + ', "CPU_TEMP":' + str(cpu_data.current) + ', "GPU_TEMP":0, "GPU_UTIL":0, "FAN_CPU":' + str(fan_data.current) + ', "VMEM_UTIL":' + str(swap_data.percent) + ', "CPU_UTIL":' + str(cpu_usage) + '}', qos=0, retain=False)
=======
client.publish("home-assistant/" + host + "/hwstats", payload='{"status":"on", "version":"' + VERSION + '","MB_TEMP":' + str(mb_data.current) + ', "MEM_UTIL":' + str(mem_data.percent) + ', "CPU_TEMP":' + str(cpu_data.current) + ', "GPU_TEMP":0, "GPU_UTIL":0, "FAN_CPU":' + str(fan_data.current) + ', "VMEM_UTIL":' + str(swap_data.percent) + ', "CPU_UTIL":' + str(cpu_usage) + '}', qos=0, retain=False)
writeLog('{"status":"on", "version":"' + VERSION + '","TCPU":"NA", "MB_TEMP":' + str(mb_data) + ', "MEM_UTIL":' + str(mem_data.percent) + ', "CPU_TEMP":' + str(cpu_data.current) + ', "GPU_TEMP":0, "GPU_UTIL":0, "FAN_CPU":"' + str(fan_data) + '", "VMEM_UTIL":' + str(swap_data.percent) + ', "CPU_UTIL":' + str(cpu_usage) + '}')
client.publish("home-assistant/" + host + "/hwstats", payload='{"status":"on", "version":"' + VERSION + '","TCPU":"NA", "MB_TEMP":' + str(mb_data) + ', "MEM_UTIL":' + str(mem_data.percent) + ', "CPU_TEMP":' + str(cpu_data.current) + ', "GPU_TEMP":0, "GPU_UTIL":0, "FAN_CPU":"' + str(fan_data) + '", "VMEM_UTIL":' + str(swap_data.percent) + ', "CPU_UTIL":' + str(cpu_usage) + '}', qos=0, retain=False)
client.publish("sectorq/systems/" + host.lower() + "/stat", payload="on", qos=0, retain=True)
>>>>>>> fba8088e24f4b4228930b1a72ca37b712d90c824
elif host.lower() == "openmediavault" or host.lower() == "omv":
#print(psutil.sensors_temperatures())
#print(psutil.sensors_fans())
@@ -1130,20 +1169,17 @@ else:
#print(str(jn1["coretemp-isa-0000"]["Package id 0"]["temp1_input"]))
#print(str(jn1["asus-isa-0000"]["cpu_fan"]["fan1_input"]))
#client.publish("sectorq/monitor/" + host.lower(), payload="{\"mb_temp\":0,\"mem_usage\":0,\"cpu_temp\":0,\"cpu_usage\":0,\"gpu_temp\":0,\"gpu_usage\":0}", qos=0, retain=False)
<<<<<<< HEAD
client.publish("home-assistant/" + host + "/hwstats", payload='{"status":"on", "version":"' + VERSION + '","reboot":"' + str(reboot_pending) + '", "uptime":"' + str(sys_uptime) + '", "raid_usage": ' + str(disk_usage) + ',"rsync":"' + str(stat_rsync) +'", "MB_TEMP":' + str(mb_data.current) + ', "MEM_UTIL":' + str(mem_data.percent) + ', "CPU_TEMP":' + str(cpu_data.current) + ', "GPU_TEMP":0, "GPU_UTIL":0, "FAN_CPU":0, "VMEM_UTIL":' + str(swap_data.percent) + ', "CPU_UTIL":' + str(cpu_usage) + '}', qos=0, retain=False)
=======
client.publish("home-assistant/" + host + "/hwstats", payload='{"status":"on","version":"' + VERSION + '","reboot":"' + str(reboot_pending) + '", "uptime":"' + str(sys_uptime) + '", "raid_usage": ' + str(disk_usage) + ',"rsync":"' + str(stat_rsync) +'", "MB_TEMP":' + str(mb_data.current) + ', "MEM_UTIL":' + str(mem_data.percent) + ', "CPU_TEMP":' + str(cpu_data.current) + ', "GPU_TEMP":0, "GPU_UTIL":0, "FAN_CPU":0, "VMEM_UTIL":' + str(swap_data.percent) + ', "CPU_UTIL":' + str(cpu_usage) + '}', qos=0, retain=False)
>>>>>>> 74b22b8d43db9a5fce3a9a54c22c6f876043206c
writeLog("lalal")
elif host.lower() == "router":
check_router()
elif host.lower() == "nas":
print("Getting VM info")
cmnd = "/share/ZFS530_DATA/.qpkg/QKVM/usr/bin/virsh list --all"
payload = {"status":"on"}
cmnd = "/share/ZFS532_DATA/.qpkg/QKVM/usr/bin/virsh list --all"
writeLog(cmnd)
status, output = subprocess.getstatusoutput(cmnd)
l = 1
l = 1
lines = output.splitlines()
VMS = {}
for i in lines:
@@ -1153,30 +1189,31 @@ else:
line = re.split(r" {2,}", i)
print(str(line[0].strip()) + " " + str(line[1].strip()) + " " + str(line[2].strip()))
VMS[str(line[1].strip())] = str(line[2].strip())
for m in VMS:
cmnd = "/share/ZFS530_DATA/.qpkg/QKVM/usr/bin/virsh domstate " + m
cmnd = "/share/ZFS532_DATA/.qpkg/QKVM/usr/bin/virsh domstate " + m
#out = subprocess.Popen(cmnd.split())
writeLog(cmnd)
status, output = subprocess.getstatusoutput(cmnd)
writeLog(output)
if output.strip() == "running":
client.publish("sectorq/systems/nas/" + m, payload=1, qos=0, retain=True)
payload[m] = 1
else:
client.publish("sectorq/systems/nas/" + m, payload=0, qos=0, retain=True)
payload[m] = 0
client.publish( "home-assistant/" + host + "/hwstats", payload=json.dumps(payload), qos=0, retain=True)
if FILE_TIMESTAMP != mtime:
writeLog("Script updated ! restarting...")
#os.execv(sys.executable, ['python'] + sys.argv)
#os.system("/share/ZFS530_DATA/.qpkg/QPython39/bin/python3 mqtt_srv.py")
if host.lower() == "nas":
sys.stdout.flush()
os.execv("/share/ZFS530_DATA/.qpkg/QPython39/bin/python3", sys.argv)
os.execv("/share/ZFS530_DATA/.qpkg/QPython39/bin/python3", sys.argv)
if host.lower() == "router" or host.lower() == "omv":
#sys.stdout.flush()
#os.execv(sys.argv, "")