#!/usr/bin/env python
# -*- coding: utf-8 -*-

from models	import *

from django.db import IntegrityError
from django.core.exceptions import ObjectDoesNotExist

from datetime import datetime, timedelta, tzinfo, date
import time

from calendar import monthrange

from string import Formatter

from ponto.models import Comentario


###################################### 
#	    FUNÇÕES GERAIS
###################################### 


def puxar_informacoes_henry(lista_funcionarios):
	
	# Arquivo com as saídas do Henry;
	arquivo = open("/mnt/henry/MeusArquivosSaida/Registros.txt.txt", 'r')

	# Ler todas as linhas do arquivo e fecha-o em seguida;
	conteudo_arquivo = arquivo.readlines()
	arquivo.close()
	
	formato = "%d/%m/%Y %H:%M:%S"
		
	for linha in conteudo_arquivo:
		
		# Matrícula com oito dígitos.
		matricula = linha.split(" ")[-1].rstrip()[12:20]
		
		# Compara matrícula do funcionário com a do arquivo.
		for funcionario in lista_funcionarios:
			
			if funcionario.matricula in matricula:
								
				# Data no formato 'DD/MM/AAAA'.
				data = linha.split(" ")[-3]
				
				# Hora no formato '00:00:00'.
				hora = linha.split(" ")[-2]
				
				data_hora = datetime.strptime("%s %s" %(data, hora), formato)
				
				batida_ponto = BatidaPonto()
		
				batida_ponto.funcionario = funcionario
				batida_ponto.momento_batidaponto = data_hora
				
				try:
					batida_ponto.save()
				except IntegrityError:
					pass
										

# Sincroniza informações do Henry com o banco de dados

def sincronizar():
	
	lista_funcionarios = Funcionario.objects.all()
	
	if lista_funcionarios:
		puxar_informacoes_henry(lista_funcionarios)
		
		nova_sincronizacao = Sincronizacao()
		agora = datetime.now()
		nova_sincronizacao.data_hora = agora
		nova_sincronizacao.save()
		
		return True
	else:
		return False
		
		
######################################
# 	  OUTRAS FUNÇÕES ÚTEIS
######################################
	
def strfdelta(tdelta, fmt):
    f = Formatter()
    d = {}
    l = {'D': 86400, 'H': 3600, 'M': 60, 'S': 1}
    k = map( lambda x: x[1], list(f.parse(fmt)))
    rem = int(tdelta.total_seconds())

    for i in ('D', 'H', 'M', 'S'):
        if i in k and i in l.keys():
            d[i], rem = divmod(rem, l[i])

    return f.format(fmt, **d)

###################################### 
#	  FUNÇÕES QUE GERAM RELATÓRIOS	
###################################### 	

class ResumoDiario():
	
	def __init__(self, data, horas):
		self.data = data
		self.horas = horas
		
		
	def horas_formatadas(self):
				
		return strfdelta(self.horas, "{H:02}:{M:02}:{S:02}")
		
	
	def data_formatada(self):
		
		return self.data.strftime("%d/%m/%Y")
		
	
		
		
def organizar_batidasponto(batidas_do_dia, data):
	
#	intervalo_almoco = datetime(data.year, data.month, data.day, 12, 30)
	
	quantidade_batidas_do_dia = len(batidas_do_dia)
	
	batida_entrada_manha = None
	batida_saida_manha = None
	batida_entrada_tarde = None
	batida_saida_tarde = None
	
	if quantidade_batidas_do_dia >= 1:
		batida_entrada_manha = batidas_do_dia[0].momento_batidaponto	
	if quantidade_batidas_do_dia >= 2:
		batida_saida_manha = batidas_do_dia[1].momento_batidaponto
	if quantidade_batidas_do_dia >= 3:
		batida_entrada_tarde = batidas_do_dia[2].momento_batidaponto
	if quantidade_batidas_do_dia >= 4:
		batida_saida_tarde = batidas_do_dia[3].momento_batidaponto

	
	"""
	
	if quantidade_batidas_do_dia > 1:
		
		batidas_antes_intervalo_almoco = []
		batidas_depois_intervalo_almoco = []
			
		for bd in batidas_do_dia:
			
			if bd.momento_batidaponto < intervalo_almoco:
				batidas_antes_intervalo_almoco.append(bd.momento_batidaponto)
			else:
				batidas_depois_intervalo_almoco.append(bd.momento_batidaponto)
				
		
		batidas_antes_intervalo_almoco.sort()
		batidas_depois_intervalo_almoco.sort()
		
		if len(batidas_antes_intervalo_almoco) == 1:
			batida_entrada_manha = batidas_antes_intervalo_almoco[0]
		elif len(batidas_antes_intervalo_almoco) > 1:
			batida_entrada_manha = batidas_antes_intervalo_almoco[0]
			batida_saida_manha = batidas_antes_intervalo_almoco[-1]
			
			
		if len(batidas_depois_intervalo_almoco) == 1:
			batida_entrada_tarde = batidas_depois_intervalo_almoco[0]
		elif len(batidas_depois_intervalo_almoco) > 1:
			batida_entrada_tarde = batidas_depois_intervalo_almoco[0]
			batida_saida_tarde = batidas_depois_intervalo_almoco[-1]
	"""
		
	"""
		if quantidade_batidas_do_dia == 1:
			pass
			
		elif quantidade_batidas_do_dia == 2:
			

			if batidas_do_dia[0].momento_batidaponto < intervalo_almoco and batidas_do_dia[-1].momento_batidaponto < intervalo_almoco:	
				
				batida_entrada_manha = batidas_do_dia[0]
				batida_saida_manha = batidas_do_dia[-1]
				
			elif batidas_do_dia[0].momento_batidaponto > intervalo_almoco and batidas_do_dia[-1].momento_batidaponto > intervalo_almoco:			
				
				
				batida_entrada_tarde = batidas_do_dia[0]
				batida_saida_tarde = batidas_do_dia[-1]

			else:
				pass
				
		
		elif quantidade_batidas_do_dia == 3:
			
			if batidas_do_dia[0].momento_batidaponto < intervalo_almoco and batidas_do_dia[1].momento_batidaponto < intervalo_almoco:
				batida_entrada_manha = batidas_do_dia[0]
				batida_saida_manha = batidas_do_dia[1]
				
			elif batidas_do_dia[0].momento_batidaponto > intervalo_almoco and batidas_do_dia[1].momento_batidaponto > intervalo_almoco:
				batida_entrada_tarde = batidas_do_dia[0]
				batida_saida_tarde = batidas_do_dia[1]
				
			elif batidas_do_dia[1].momento_batidaponto < intervalo_almoco and batidas_do_dia[2].momento_batidaponto < intervalo_almoco:
				batida_entrada_manha = batidas_do_dia[1]
				batida_saida_manha = batidas_do_dia[2]
				
			elif batidas_do_dia[1].momento_batidaponto > intervalo_almoco and batidas_do_dia[2].momento_batidaponto > intervalo_almoco:
				batida_entrada_tarde = batidas_do_dia[1]
				batida_saida_tarde = batidas_do_dia[2]	
				
			else:
				pass
			
			
		else:
			
			batidas_antes_intervalo_almoco = []
			batidas_depois_intervalo_almoco = []
			
			for bd in batidas_do_dias:
				
				if bd.momento_batidaponto < intervalo_almoco:
					batidas_antes_intervalo_almoco.append(bd)
				else:
					batidas_depois_intervalo_almoco.append(bd)
				
			
			if batidas_do_dia[0].momento_batidaponto < intervalo_almoco 
			batida_entrada_manha = batidas_do_dia[0]
			batida_saida_tarde = batidas_do_dia[-1]
			for i in range(quantidade_batidas_do_dia):
				if batidas_do_dia[i].momento_batidaponto >= intervalo_almoco:
					batida_saida_manha = batidas_do_dia[i-1]
					batida_entrada_tarde = batidas_do_dia[i]
					break
	"""
		
	return batida_entrada_manha, batida_saida_manha, batida_entrada_tarde, batida_saida_tarde

		
def calcular_horas_trabalhadas_dia(funcionario, data):
		
	batidas_do_dia = list(BatidaPonto.objects.filter(funcionario=funcionario, momento_batidaponto__day=data.day, momento_batidaponto__month=data.month, momento_batidaponto__year=data.year).order_by('momento_batidaponto'))
	
	tempo_total_dia = timedelta(0,0)
		
	batida_entrada_manha, batida_saida_manha, batida_entrada_tarde, batida_saida_tarde = organizar_batidasponto(batidas_do_dia, data)
	
	if batida_entrada_manha and batida_saida_manha:
		
		# Para limitar o horário de entrada
				
		horario_entrada = batida_entrada_manha.time()
		
		if horario_entrada < funcionario.horario_inicial_entrada:
			batida_entrada_manha = datetime.combine(data, funcionario.horario_inicial_entrada)
		
		horas_trabalhadas_manha = batida_saida_manha - batida_entrada_manha
		tempo_total_dia += horas_trabalhadas_manha
			
	if batida_entrada_tarde and batida_saida_tarde:
		
		# Para limitar o horário de saída
				
		horario_saida = batida_saida_tarde.time()
		
		if horario_saida > funcionario.horario_final_saida:
			batida_saida_tarde = datetime.combine(data, funcionario.horario_final_saida)
			
		horas_trabalhadas_tarde = batida_saida_tarde - batida_entrada_tarde
		tempo_total_dia += horas_trabalhadas_tarde
		
	
	try:
		
		comentario = Comentario.objects.get(funcionario=funcionario, data=data)
		
		if comentario.aprovado:
			
			tempo_gasto = comentario.tempo_gasto
			
			horas = tempo_gasto.hour
			minutos = tempo_gasto.minute
			segundos = tempo_gasto.second
			
			tempo_gasto_timedelta = timedelta(0,segundos,0,0,minutos,horas,0)
			
			tempo_total_dia += tempo_gasto_timedelta
	
	except ObjectDoesNotExist:
		
		pass
			

	"""
	if(quantidade_batidas_do_dia%2==0):
		for i in range(0, quantidade_batidas_do_dia, 2):
			hora_menor = batidas_do_dia[i].momento_batidaponto
			hora_maior = batidas_do_dia[i+1].momento_batidaponto
			tempo_parcial = hora_maior - hora_menor
			tempo_total_dia += tempo_parcial
			
	else:
		for i in range(0, quantidade_batidas_do_dia-1, 2):
			hora_menor = batidas_do_dia[i].momento_batidaponto
			hora_maior = batidas_do_dia[i+1].momento_batidaponto
			tempo_parcial = hora_maior - hora_menor
			tempo_total_dia += tempo_parcial
	"""
	

	return tempo_total_dia		
	
	
	
def horas_trabalhadas_dia(funcionario):
	
	resumos = []
	
	primeira_batidaponto = None
	
	try:
		primeira_batidaponto = BatidaPonto.objects.filter(funcionario=funcionario).earliest('momento_batidaponto')
	except ObjectDoesNotExist:
		pass
		
	if primeira_batidaponto:
		
		primeira_data = primeira_batidaponto.momento_batidaponto
		data_atual = datetime.now()
		
		diferenca_datas = data_atual - primeira_data
		quantidade_dias = diferenca_datas.days
	
		data_temp = primeira_data
		
		while data_temp.date() <= data_atual.date():
						
			horas_dia = calcular_horas_trabalhadas_dia(funcionario, data_temp)
			
			resumo = ResumoDiario(data_temp.date(), horas_dia)
			
			resumos.append(resumo)
			
			data_temp += timedelta(days=1)
			
	
	return resumos
	
	
class ResumoSemanal():
	
	def __init__(self, domingo, horas):
		
		self.domingo = domingo
		self.horas = horas
		
	def horas_formatadas(self):
		
		return strfdelta(self.horas, "{H:02}:{M:02}:{S:02}")
		
		
	def semana_formatada(self):
		
		meses  = ("Janeiro", "Fevereiro", "Março", "Abril", "Maio", "Junho", "Julho", "Agosto", "Setembro", "Outubro", "Novembro", "Dezembro")
		
		domingo = self.domingo
		sexta_feira = self.domingo + timedelta(days=6)
		
		dia_domingo = "%d" %(domingo.day)
		dia_sexta = "%d" %(sexta_feira.day)
		
		if domingo.day < 10:
			dia_domingo = "0%d" %(domingo.day)
			
		if sexta_feira.day < 10:
			dia_sexta = "0%d" %(sexta_feira.day)
		
		
		semana = "De %s de %s de %d a %s de %s de %d" %(dia_domingo, meses[domingo.month -1], domingo.year, dia_sexta, meses[sexta_feira.month -1], sexta_feira.year)
		
		return semana

	
def calcular_horas_trabalhadas_semana(funcionario, domingo):

	tempo_total_semana = timedelta(0,0)
	
	data_temp = domingo
	
	for i in range(7):
		tempo_total_semana += calcular_horas_trabalhadas_dia(funcionario, data_temp)
		data_temp += timedelta(days=1)

	# Retorna o tempo total da semana.
	return tempo_total_semana
	
	
def horas_trabalhadas_semana(funcionario):
	
	resumos = []
	primeira_batidaponto = None
	
	try:
		primeira_batidaponto = BatidaPonto.objects.filter(funcionario=funcionario).earliest('momento_batidaponto')
	except ObjectDoesNotExist:
		pass		
		
	if primeira_batidaponto:
		
		primeira_data = primeira_batidaponto.momento_batidaponto
		data_atual = datetime.now()
		
		dia_semana = primeira_data.weekday()
		
		# Caso não seja uma domingo, ele voltará alguns dias até encontrá-la.
		while dia_semana != 6:
			primeira_data -= timedelta(days=1)
			dia_semana = primeira_data.weekday()
		
		data_atual = datetime.now()
		data_temp = primeira_data
				
		while data_temp.date() < data_atual.date():
			
			horas_trabalhadas = calcular_horas_trabalhadas_semana(funcionario, data_temp)
			
			resumo_semanal = ResumoSemanal(data_temp.date(), horas_trabalhadas)
			
			resumos.append(resumo_semanal)
	
			data_temp += timedelta(days=7)	
		
	return resumos
	

def horas_trabalhadas_semana_durante_mes(funcionario, mes, ano):
	
	limites_mes = monthrange(ano, mes)
	
	primeiro_dia_mes = date(ano, mes, limites_mes[0])
	ultimo_dia_mes = date(ano, mes, limites_mes[1])
	
	primeira_data = primeiro_dia_mes
	
	resumos = []
	
	dia_semana = primeira_data.weekday()
	
	# Caso não seja uma domingo, ele voltará alguns dias até encontrá-la.
	
	while dia_semana != 6:
		primeira_data -= timedelta(days=1)
		dia_semana = primeira_data.weekday()

	data_temp = primeira_data
			
	while data_temp < ultimo_dia_mes:
		
		horas_trabalhadas = calcular_horas_trabalhadas_semana(funcionario, data_temp)
		
		resumo_semanal = ResumoSemanal(data_temp, horas_trabalhadas)
		
		resumos.append(resumo_semanal)

		data_temp += timedelta(days=7)	
		
	return resumos
	
def horas_trabalhadas_datas(funcionario, datas):
	
	resumos = []

	tempo_total = timedelta(0,0)

	for data in datas:	
		tempo_total += calcular_horas_trabalhadas_dia(funcionario,data)
		
	return strfdelta(tempo_total, "{H:02}:{M:02}:{S:02}")
		

class RelatorioDiario():
	
	def __init__(self, data, tempo_total_dia, batida_entrada_manha, batida_saida_manha, batida_entrada_tarde, batida_saida_tarde):
		self.data = data
		self.tempo_total_dia = tempo_total_dia
		
		if batida_entrada_manha:
			self.batida_entrada_manha = batida_entrada_manha.strftime("%H:%M:%S")
		if batida_saida_manha:
			self.batida_saida_manha =  batida_saida_manha.strftime("%H:%M:%S")
		if batida_entrada_tarde:
			self.batida_entrada_tarde = batida_entrada_tarde.strftime("%H:%M:%S")
		if batida_saida_tarde:
			self.batida_saida_tarde = batida_saida_tarde.strftime("%H:%M:%S")
		
		

def gerar_relatorio_mensal(funcionario, mes, ano):
		
	fim_do_mes = monthrange(ano, mes)[1]
	relatorios_diarios = []
	
	# Percorre todos os dias do mês.	
	for dia in range(1,fim_do_mes+1,1):
			
		batidas_formatadas = []
		data = date(ano, mes, dia)
		data_formatada = data.strftime("%d/%m/%Y")
		
		tempo_total_dia = calcular_horas_trabalhadas_dia(funcionario, data)
		
		tempo_total_dia_formatado = strfdelta(tempo_total_dia, "{H:02}:{M:02}:{S:02}")
		
		batidas_ponto_dia = list(BatidaPonto.objects.filter(funcionario=funcionario, momento_batidaponto__month=mes, momento_batidaponto__day=dia, momento_batidaponto__year=ano))

		batida_entrada_manha, batida_saida_manha, batida_entrada_tarde, batida_saida_tarde = organizar_batidasponto(batidas_ponto_dia , data)
		
		#for bp in batidas_ponto_dia:
		#	batidas_formatadas.append(bp.momento_batidaponto.strftime("%H:%M:%S"))
			
		relatorio_diario = RelatorioDiario(data_formatada, tempo_total_dia_formatado, batida_entrada_manha, batida_saida_manha, batida_entrada_tarde, batida_saida_tarde)
		relatorios_diarios.append(relatorio_diario)
		
	return relatorios_diarios