mumble_bot.py

#https://github.com/azlux/pymumble-> PYMUMBLE python library

import os, sys
import subprocess
import time
import cberry
import telegram_v2 as telegram
#import telegram_v1 as telegram
import _thread as thread
import tools
import objects
import mumble_chat
import time
from signal import signal, SIGINT, SIGTERM, SIGABRT
from datetime import datetime, timedelta
import sys


event_counter=0
last_event_counter=0


# THIS IS THE MAIN ROUTINE --. START OF THE PROGRAM
def main():

	global last_event_counter
	global event_counter
	mode="debug"

	one_minute=60000/1000#seconds
	one_second=one_minute/60

	berry =None	
	
	#if os.getuid() != 0:
	#	print("This program is not run as sudo or elevated this it will not work")
	#	exit()	

	if len(sys.argv) == 2:			
		if sys.argv[1] == "--release" or sys.argv[1] == "--debug":
			mode=sys.argv[1][2:]#cut --
		else:

			print(sys.argv[1]+" is not a valid mode (--release or --debug)")
			exit()			

	telegram.init(mode)	
	#mumble_chat.init()




	print("start mumble_bot mode="+mode+" time= "+str(datetime.now()))

	delay=one_second#init delay
	cert_exp=""
	ip=""



	while True:

		#time.sleep(delay);#wait in the release version	

		time.sleep(1)


		online_users= read_Online_Users()# set  event_counter
		registeredUsers= read_Registered_Users()	

		if online_users==  "sql_no_time_error":
			# online user can not be identified
			telegram.send_message("Server Start Date could not be read  -> mumble server restart required".encode("utf-8"),telegram.telegram_private.chat_id_goto)
			telegram.shutdown()#die here


		one_user_online=False	

		#check if a user is loged in		
		for user in online_users:

			#print(user+" "+str(online_users[user]))		
			if online_users[user].event_counter%2!=0:
				one_user_online=True

		#write to std out or whatsapp
		if  last_event_counter != event_counter and len(online_users)!=0:	#there is a user stat change		
			message=""
			delay=one_minute*0.5;#wait with the next check 30 sec		

			#*****************************************************
			#********** telegram part****************************
			#*****************************************************
			
			#iterate over online users			
			for user in online_users:	
				if user in registeredUsers.keys():#user is registered
					#compute message
					if online_users[user].event_counter % 2 != 0:						
						message+=user+"\t"+ u'\U00002705'+"\n"#online
					else:						
						message+=user+"\t"+ u'\U0000274C'+"\n"#offline

			if message!="":#a registered user is online
				message="*****************\nPbtH-Mumble v2.1\n*****************\n"+message#add message to whats app command				
				#print(message.encode("utf-8").expandtabs())				
				telegram.send_message(message.encode("utf-8").expandtabs(16))

			last_event_counter = event_counter;#reset event counter


		
		else:
			delay=one_second#reset delay
				


def get_cert_validity():
#	cert_exp=tools.runCmd("openssl x509 -in /etc/letsencrypt/live/dom.pbth.de/cert.pem -noout -dates | grep 'notAfter'")				
	cert_exp=tools.runCmd("openssl s_client -connect localhost:64738 -showcerts </dev/null 2>/dev/null|openssl x509 -dates -noout | grep 'notAfter' | cut -c10- ")				
	start = cert_exp.find("=")+1
	return cert_exp[start:]#cut first part
			
 #get ip
def getIP():	
	return tools.runCmd('ifconfig wlan0 | grep "inet\ addr" | cut -d: -f2 | cut -d" " -f1')




 #get all registered users
def read_Registered_Users():   
	registered_users={}
	query= "sqlite3 /var/lib/mumble-server/mumble-server.sqlite 'SELECT name FROM users'"

	p = subprocess.Popen(query, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
	for user in p.stdout.readlines():		
		p.wait()		
		user=user[:-1].decode("utf-8")#cut \n at the end			
		registered_users[user]=""		

	return registered_users
	
	

#who is online
def read_Online_Users():  
	global event_counter
	event_counter=0#reset event counter
	online_users={}
	

	server_start_date = tools.runCmd("sqlite3 /var/lib/mumble-server/mumble-server.sqlite 'SELECT msgtime FROM slog WHERE msg LIKE \"%Server listening on%\" ORDER BY msgtime desc LIMIT 1'")
	
	if server_start_date == "": 
		return "sql_no_time_error"
		
	query= "sqlite3 /var/lib/mumble-server/mumble-server.sqlite 'SELECT * FROM slog  WHERE msgtime >=Datetime(\""+server_start_date+"\")' | grep 'Authenticated\\|Connection\\|Rejected'"
	
	
	
	p = subprocess.Popen(query, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
	
	
	
	for line in p.stdout.readlines():	
			
		p.wait()
		sline=line.decode("utf-8")# byte to str
		
					

		start = sline.find(":")+1
		stop = sline.find(">")
		if start <0 or stop <0:
			continue	

		substring=sline[start:stop]	
		stop= substring.rfind('(')		
		user=substring[:stop]

		last_event=sline[sline.rfind('|')+1:].replace("\n","")		

		if time.localtime().tm_isdst: # summertime :-)
			last_event=plus_x_hour(last_event,2)	
		else:
			last_event=plus_x_hour(last_event,1)	

		
		if user!="" and user !="PbtH_bot" :#user had a disconnect bevore authentication or is a bot
			online_users=update_user(online_users,user,last_event)			
			event_counter=event_counter+1#update global event counter
		
	
	return online_users


def plus_x_hour(value,hour):

	date_time= datetime.strptime(value, '%Y-%m-%d %H:%M:%S')+timedelta(hours=hour)	
	return str(date_time)

def update_user(online_users,user_name,time_stamp):

	
	if user_name in online_users.keys():# key is in dict = user is known to us		
		 online_users[user_name].event_counter=online_users[user_name].event_counter+1;# get the value and increase
		 online_users[user_name].last_active=time_stamp

	else:		 		
		 online_users[user_name]=objects.User(user_name,0,time_stamp,1)#create a new user

	return online_users


if __name__ == '__main__':
    main()