[Из песочницы] Индивидуальный дневной лимит на исходящие звонки (ограничение платных направлений)19.08.2016 12:03

exten => _877[15-8]XXXXXXX,1,Macro(user-callerid,LIMIT,EXTERNAL,)
exten => _877[15-8]XXXXXXX/2055,1,Macro(user-callerid,LIMIT,EXTERNAL,)
exten => _877[15-8]XXXXXXX/2055,n,ExecIf($[ "${CALLEE_ACCOUNCODE}" != "" ] ?Set(CDR(accountcode)=${CALLEE_ACCOUNCODE}))
exten => _877[15-8]XXXXXXX/2055,n,Set(MOHCLASS=${IF($["${MOHCLASS}"=""]?default:${MOHCLASS})})
exten => _877[15-8]XXXXXXX/2055,n,ExecIf($["${KEEPCID}"!="TRUE" & ${LEN(${TRUNKCIDOVERRIDE})}=0]?Set(TRUNKCIDOVERRIDE=<7123456789>))
exten => _877[15-8]XXXXXXX/2055,n,Set(_NODEST=)
exten => _877[15-8]XXXXXXX/2055,n,Gosub(sub-record-check,s,1(out,${EXTEN},))
exten => _877[15-8]XXXXXXX/2055,n,Macro(dialout-trunk,10,${EXTEN},,off)
exten => _877[15-8]XXXXXXX/2055,n,Macro(outisbusy,)

CREATE TABLE `users` (
  `user_id` int(11) NOT NULL AUTO_INCREMENT,
  `user_name` varchar(100) COLLATE utf8_unicode_ci DEFAULT NULL COMMENT 'ФИО пользователя',
  `extension` varchar(5) CHARACTER SET utf8 DEFAULT '000' COMMENT 'Внутренний номер абонента',
  `mobile_limit_flag` int(11) DEFAULT '0' COMMENT 'Флаг для учета текущего лимита',
  `mobile_limit` int(11) DEFAULT '0' COMMENT 'Текущий лимит',
  `base_mobile_limit` int(11) DEFAULT NULL COMMENT 'Базовый лимит',
  PRIMARY KEY (`user_id`)
) ENGINE=InnoDB AUTO_INCREMENT=33 DEFAULT CHARSET=utf8 COLLATE=utf8_unicode_ci

def commentBlocks(numb):
	import xml.etree.cElementTree as ET
	tree = ET.ElementTree(file='conf.xml')
	root = tree.getroot()
	f = open(r'extensions_custom.conf')
	lines = f.readlines()
	f.close()
	for elem in tree.iterfind('block[@number="'+numb+'"]/element'):
	    lines[int(elem.get('first'))-2] = ";--\n"
	    lines[int(elem.get('last'))] = "--;\n"
	f = open(r'extensions_custom.conf','w')
	f.writelines(lines)
	f.close()

#мои функции
import send_email
import flags

print ('###########START_MOBILE_LIMIT############')

import pymysql
mainconn = pymysql.connect(host='10.10.2.1', user='user', passwd='password', db='asteriskcdrdb', charset='utf8')
maincur = mainconn.cursor()
maincur.execute("""SELECT SUM(billsec) AS sec, src 
	FROM cdr WHERE disposition = 'ANSWERED' 
	AND (dst LIKE '8700%' OR dst LIKE '8701%' 
	OR dst LIKE '8702%' OR dst LIKE '8705%' OR dst LIKE '8707%' 
	OR dst LIKE '8708%' OR dst LIKE '8747%' OR dst LIKE '8771%' 
	OR dst LIKE '8775%' OR dst LIKE '8776%' OR dst LIKE '8777%' 
	OR dst LIKE '8778%') AND DATE(calldate) = DATE(CURDATE()) 
	AND src in (2055,2066,2077)
	GROUP BY src;""")

row = maincur.fetchone()

print ('ROW COUNT: ' + str(self.maincur.rowcount))

while row is not None:

	#row[1] - внутренний номер
	#row[0] - исчерпанный лимит в секундах

	#изменяем текущий лимит
	flags.UpdateUserCurrentLimit(str(row[1]), str(row[0]))

	per = row[0] * 100 / flags.checkUserLimit(row[1])
	flag = flags.checkFlag(row[1])
	#вытаскиваем почтовый ящик абонента, ищем его по внутреннему номеру
	manager_mail = send_email.getEmail(row[1])

	#показываем % израсходонного трафика
	print (row[1] + ' (' + str(round(per,0)) + '%): ' + str(row[0]))

	#проверяем порог
	if per >= 50 and per < 90:
		message = 'Nomer ' + row[1] + ', limit ischerpan na ' + str(round(per, 0)) + '%'
		if flag == 0:
			print ('go email to ' + send_email.getEmail(row[1]))
			send_email.send_message(manager_mail, message)
			flags.changeFlag(row[1], 1)
			flags.insertLog(row[1], per)
		print (message)
	elif per > 90 and per < 100:
		message = 'Nomer ' + row[1] + ', limit ischerpan na ' + str(round(per, 0)) + '%'
		if flag == 1:
			print ('go email to ' + send_email.getEmail(row[1]))
			send_email.send_message(manager_mail, message)
			flags.changeFlag(row[1], 2)
			flags.insertLog(row[1], per)
		print (message)
	elif per >= 100:
		message = 'Nomer ' + row[1] + ', limit polnostiu ischerpan'
		if flag != 3:
			print ('go email to ' + send_email.getEmail(row[1]))
			send_email.send_message(manager_mail, message)
			flags.changeFlag(row[1], 3)
			flags.insertLog(row[1], per)
			#комментируем соответствующие блоки в конфиге
			flags.commentBlocks(str(row[1]))
			import subprocess
			#дергаем диалплан чтобы применить настройки
			subprocess.call(['./dialplan_reload.sh'])
		print (message)

	row = maincur.fetchone()

maincur.close()
mainconn.close()

print ('############END_MOBILE_LIMIT#############')

def AddUnusedLimit(ext):
	conn = pymysql.connect(host='10.10.2.2', user='user', passwd='password', db='crm', charset='utf8')
	cur = conn.cursor()

	cur.execute ("""
	   UPDATE users
	   SET mobile_limit=base_mobile_limit+mobile_limit WHERE extension=%s
	""", (ext))

	conn.commit()
	print('changed', cur.rowcount)
	cur.close()
	conn.close()

def uncommentBlocks():
	import xml.etree.cElementTree as ET
	tree = ET.ElementTree(file='conf.xml')
	root = tree.getroot()
	for elem in tree.iterfind('block/element'):
		first = int(elem.get('first'))
		last = int(elem.get('last'))
		lines[first-2] = "\n"
		lines[last] = "\n"
f = open(r'/etc/asterisk/extensions_custom.conf')
lines = f.readlines()
f.close()

############################################

cur.execute ("""
	   UPDATE users
	   SET mobile_limit_flag=%s
	""", (0))

#функция логирования израсходованного лимита
def insertLog(ext, per):
	import pymysql
	conn = pymysql.connect(host='10.10.2.1', user='user', passwd='password', db='crm', charset='utf8')
	cur = conn.cursor()

	cur.execute ("""
	   INSERT INTO mobile_limit
	   (extension, percent)
	   VALUES (%s, %s)
	""", (ext, per))

	conn.commit()
	print('insert', cur.rowcount)
	cur.close()
	conn.close()