VOLUME_MOUNT/main.py

115 lines
5.4 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})
if system_info.get_client_info() == "Client":
user = system_info.get_smb_credential()
### Check User
check_user_exist = request("user/get",{"user":user,"dbname":database})
print(check_user_exist.text)
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)