CardsAgainstX/Glucose-Checker
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added All relevant code shit

Browse source
This commit is contained in:
Bart Koolen 2025-02-24 21:46:51 +01:00 • committed by GitHub
parent 872ba17e85
commit 309a1cfd58
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 387 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

154
src/ServerCheck.py Normal file
View file

@ -0,0 +1,154 @@
import requests
import paho.mqtt.client as mqtt
#from smbprotocol.connection import Connection
#from smbprotocol.session import Session
#from smbprotocol.tree import TreeConnect
from time import sleep
##### Mqtt vars #####
broker_address = "192.168.1.250"
broker_port = 1883
recieve_topic = "telegrambot/in"
send_topic = "telegrambot/out"
brokercreds = ["hass","Bruk#5"]
#######################
# Setup For Variables #
#######################
class Site: #Custom classes for my dictionaries
def __init__(self, url, typeVar):
self.url = url
self.typeVar = typeVar
webServers = {"Google" : Site(url="https://google.com", typeVar="URL"), #List with all the servers urls and names and types
"GameServer" : Site(url="http://192.168.0.251", typeVar="URL"),
"SpiritBody" : Site(url="https://spiritbodyhealing.org", typeVar="URL"),
"HeresYourSign": Site(url="https://heresyoursign.nl", typeVar="URL"),
#"SMB_server": Site(url="192.168.1.250", typeVar="SMB"),
}
webServersStatus = {}
mute_time = 86400 #This is in seconds, it defines how long the program /mute command lasts
interval_offline_message = 300 #This is in seconds, it defines how often the the program sents a message of wich sites are offline
green_circle = "🟢"#emoji
red_circle = "🔴"#emji
######################
# Here Begins Script #
######################
def checkWebServer(serverList): #This function checks all servers in list if they changed state
tempVarC = ""
for i in serverList: #Iterates over every server
if serverList[i].typeVar == "SMB": #This is type SMB
try:
guid = uuid.uuid4() # Unique identifier for the connection
connection = Connection(server_name=serverList[i].url, port=445, guid=guid) #Trys to connect to server
connection.connect()
connection.disconnect()
if i in webServersStatus:
if webServersStatus[i] == "Offline": #Checks if status of server has changed
webServersStatus[i] = "Online"
tempVarC = f"{tempVarC}{green_circle} {i} is {webServersStatus[i]}\n"
else:
webServersStatus[i] = "Online" #This runs on boot to get initial server state
tempVarC = f"{tempVarC}{green_circle} {i} is {webServersStatus[i]}\n"
#return True
except Exception as e: # Trows an exeption when the SMb server is unreachble
#print(f"Error accessing SMB share: {e}")
if i in webServersStatus:
if webServersStatus[i] == "Online":
webServersStatus[i] = "Offline"
tempVarC = f"{tempVarC}{red_circle} {i} is {webServersStatus[i]}\n"
else:
webServersStatus[i] = "Offline" #This runs on boot to get initial server state
tempVarC = f"{tempVarC}{red_circle} {i} is {webServersStatus[i]}\n"
#return False
else: #This is type url
try:
r = requests.get(serverList[i].url, timeout=20) #sends request to server
r.raise_for_status()
except (requests.exceptions.ConnectionError, requests.exceptions.Timeout): #this runs if server is down or unreacheble
if i in webServersStatus:
if webServersStatus[i] == "Online": #Checks if status of server has changed
webServersStatus[i] = "Offline"
tempVarC = f"{tempVarC}{red_circle} {i} is {webServersStatus[i]}\n"
#else:
#do nothing
else:
webServersStatus[i] = "Offline" #This runs on boot to get initial server state
tempVarC = f"{tempVarC}{red_circle} {i} is {webServersStatus[i]}\n"
except requests.exceptions.HTTPError:
if i in webServersStatus:
if webServersStatus[i] == "Online": #Checks if status of server has changed
webServersStatus[i] = "Unavailable"
#else:
#do nothing
else:
webServersStatus[i] = "Unavailable" #This runs on boot to get initial server state
tempVarC = f"{tempVarC}{red_circle} {i} is {webServersStatus[i]}\n"
else:
if i in webServersStatus:
if webServersStatus[i] == "Offline": #Checks if status of server has changed
webServersStatus[i] = "Online"
tempVarC = f"{tempVarC}{green_circle} {i} is {webServersStatus[i]}\n"
#else:
#do nothing
else:
webServersStatus[i] = "Online" #This runs on boot to get initial server state
tempVarC = f"{tempVarC}{green_circle} {i} is {webServersStatus[i]}\n"
return tempVarC
def statusCheck(statusReal): #This command can be run by the user to check the state off all servers
tempVarS = ""
for i in webServersStatus: #Adds fancy emoji depending on if server is online or offline
if webServersStatus[i] == "Online":
if statusReal: #This is so depending on what is needed can return every server that is offline and not online
tempVarS = f"{tempVarS}{green_circle} {i} is {webServersStatus[i]}\n"
else:
tempVarS = f"{tempVarS}{red_circle} {i} is {webServersStatus[i]}\n"
return tempVarS
def on_connect(client, userdata, flags, reason_code, properties): #This runs once mqqt starts and try's to connect to mqtt broker
print(f"Connected with result code {reason_code}")
if reason_code == 0:
# success connect
print("Connected to MQTT broker")
Mqtt.subscribe(recieve_topic)
if reason_code > 0:
# error processing
print(f"Failed to connect to MQTT broker. Error code: {rc}")
def on_message(client, userdata, msg): #This runs everytime a mqtt message is recieved
checkWebServer(webServers) #Update list of websites status
messageStatus = statusCheck(True) #True returns every server and state
print(messageStatus)
Mqtt.publish(send_topic, messageStatus)
Mqtt = mqtt.Client(mqtt.CallbackAPIVersion.VERSION2) #Creates a mqqt object
Mqtt.on_connect = on_connect #Create callback on connect
Mqtt.on_message = on_message #Create callback on message
Mqtt.username_pw_set(brokercreds[0], password=brokercreds[1]) #Set User and Password
Mqtt.connect(broker_address, broker_port, 60) #Set Connection settings
Mqtt.loop_start() #Start connecting
def main(): #This is the main loop
while True:
checkWebServer(webServers) #Update list of websites status
messageStatus = statusCheck(False) #False is only on state change
Mqtt.publish(send_topic, messageStatus)
sleep(30)
if __name__ == '__main__':
main()