110 lines
5.2 KiB
Python
110 lines
5.2 KiB
Python
import os
|
|
import subprocess
|
|
import requests
|
|
import time
|
|
import socket
|
|
import datetime
|
|
import system_info
|
|
import net_volume
|
|
|
|
def request(function,variable): ##API-Server
|
|
url = (f"http://api.stines.de:8001/{function}")
|
|
response = requests.post(url,json = variable,headers={'access_token':'^YWUbG7yX*V!tV^KBSd*2c&vdN3wV9a2i7f3hfGFMBYFxi6#mMiJGiaA5KEHE%B*miK%qb7rQ67gmcYP@gqmux8'})
|
|
return response
|
|
|
|
|
|
def space_free():
|
|
space_free = str(system_info.get_single_hdd("free")).replace("['", "").replace("]", "").replace("'", "").replace(
|
|
",", ";") ## FREE SPACE OF VOLUME
|
|
return space_free
|
|
|
|
def space_used():
|
|
space_used = str(system_info.get_single_hdd("used")).replace("['", "").replace("]", "").replace("'", "").replace(
|
|
",", ";") ## USED SPACE OF VOLUME
|
|
return space_used
|
|
|
|
def space_total():
|
|
space_total = str(system_info.get_single_hdd("total")).replace("['", "").replace("]", "").replace("'", "").replace(
|
|
",", ";") ## TOTAL SPACE OF VOLUME
|
|
return space_total
|
|
|
|
|
|
# def pushover_send(errorcode,hostname): ## SEND PUSHOVER
|
|
# variable = {"token": "az8aqcsg45ikm73q3ry34vxs2z9kgj","user": "uo2sf2pmrtjvt8auu786fviabimimr","message":f"{errorcode}","title":f"VOLUME_MOUNT ERROR on HOST:{hostname}"}
|
|
# requests.post("https://api.pushover.net:443/1/messages.json",variable, {"Content-type": "application/x-www-form-urlencoded"})
|
|
|
|
|
|
hostname = socket.gethostname() ## GET HOSTNAME
|
|
user = os.getlogin() ## GET CURRENT USER
|
|
|
|
## Check Database
|
|
database = request("host/get/database",{"ip":system_info.get_local_ip()}).text.replace('"','') ##GET DATABASE FROM SQL-SERVER
|
|
check_hostname = request("host/get",{"hostname":hostname,"dbname":database}) ##GET VOLUMES FROM SQL-SERVER
|
|
|
|
## Check Host
|
|
if check_hostname.text == "false":
|
|
request("host/set",{"hostname":hostname,"ipadresse":system_info.get_local_ip(),"ram":system_info.get_ram_info(),"hddused":space_used(),"hddfree":space_free(),"hddtotal":space_total(),"dbname":database})
|
|
else:
|
|
request("host/update",
|
|
{"hostname": hostname, "ipadresse": system_info.get_local_ip(), "ram": system_info.get_ram_info(),
|
|
"hddused": space_used(), "hddfree": space_free(), "hddtotal": space_total(), "dbname": database})
|
|
|
|
### Check User
|
|
check_user_exist = request("user/get",{"user":user,"dbname":database})
|
|
if check_user_exist.text == "false":
|
|
request("user/set", {"user": user, "dbname": database,"hardwareID": check_hostname.text})
|
|
else:
|
|
request("user/set/lastlogon", {"userID": check_user_exist.text, "dbname": database, "lastlogon": system_info.get_lastlogon(user)})
|
|
|
|
### Check CPU
|
|
check_cpu = request("host/get/cpu",{"cputype":system_info.get_cpu_brand(),"ghz":system_info.get_cpu_ghz()})
|
|
if check_cpu.text == "false":
|
|
request("host/set/cpu",{"cputype":system_info.get_cpu_brand(),"cpucore":system_info.get_cpu_core()})
|
|
|
|
### Check Mainboard
|
|
request("host/mainboard",{"mainboardmodel":system_info.get_mainboard_model(),"mainboardvendor":system_info.get_mainboard_vendor()})
|
|
|
|
### Check NetVolume Exist
|
|
netvolume = request("host/get/netvolume",{"user":user,"dbname":database})
|
|
|
|
if netvolume.text == "false":
|
|
print(net_volume.get_netvolume())
|
|
if net_volume.get_netvolume() is None:
|
|
with open("log.txt", "a") as datei: ## Schreibt ergebnissse in Log File
|
|
datei.writelines(f"Keine Netvolume vorhanden!" + "\n") ## WRITE TO ERRORLOG
|
|
# pushover_send("Keine Netvolume vorhanden!", hostname) ## SEND ERROR LOG TO PUSHOVER
|
|
else:
|
|
for i in netvolume.json(): ##LIST FORM netvolume
|
|
print(i[2])
|
|
if i != "-":
|
|
if not os.path.isdir(f"{i[1]}"):
|
|
try:
|
|
result = subprocess.run(["net", "use", f"{i[1]}:", f"{i[0]}"], capture_output=True, check=True) ## MOUNT VOLUME
|
|
script = fr'''$shell = New-Object -ComObject Shell.Application
|
|
$shell.NameSpace("{i[1]}:").Self.Name = "{i[2]}"''' ### Schreibt den Namen der Netzlaufwerke um
|
|
subprocess.run(["powershell", "-Command", script], capture_output = True,text = True)
|
|
print(script)
|
|
print("net", "use", f"{i[1]}:", f"{i[0]}")
|
|
except subprocess.CalledProcessError as e:
|
|
time = datetime.datetime.now().strftime("%d.%m.%Y %H:%M")
|
|
if "85" in str(e.stderr):
|
|
print(f"{time} - Volume exists {i[0]} {i[1]}")
|
|
elif "1223" in str(e.stderr):
|
|
errorcode = f"{time} - {user} has no access to {i[0]} {i[1]}"
|
|
elif "53" in str(e.stderr):
|
|
errorcode = f"{time} - Volume was not Found {i[0]} {i[1]}"
|
|
elif "64" in str(e.stderr):
|
|
errorcode = f"{time} - Volume Path not found {i[0]} {i[1]}"
|
|
elif "1219" in str(e.stderr):
|
|
errorcode = f"{time} - Multiconnection to Folder not allow {i[0]} {i[1]}"
|
|
else:
|
|
errorcode = f"{time} - Unknow ERROR Code {str(e.stderr)}"
|
|
|
|
|
|
with open("log.txt", "a") as datei:
|
|
datei.writelines(f"{errorcode}" + "\n") ## WRITE TO ERRORLOG
|
|
# # pushover_send(errorcode, hostname) ## SEND ERROR LOG TO PUSHOVER
|
|
#
|
|
# while True:
|
|
# check_volume_exist()
|
|
# time.sleep(60) |