#!/usr/bin/python
# coding: utf-8
import psycopg2
import os

def connect():
	try:
		conn = psycopg2.connect(host="54.149.62.158", database="delfar", user="delfar", password="bidwbd@delfar")
		conn.set_client_encoding('UTF8')
		return conn
	except (Exception, psycopg2.DatabaseError) as error:
		print('ERRO: ')
		print(error)
		exit()

def readSqlVendedor():
	with open('vendas_vendedor.sql', mode='r', encoding='utf8') as f:
		sql = f.read()
	return sql

def readSqlAdmin():
	with open('vendas_admin.sql', mode='r', encoding='utf8') as f:
		sql = f.read()
	return sql


def salvar(dados, vendedor):
	dir = './'

	if os.name != 'nt':
		dir = '/var/www/html/autoreports'

	a = vendedor[0]+'_'+vendedor[1]+'.csv';

	try:
		file = open(dir+'/'+a, mode='w', encoding='utf8')

		linha = ''

		for row in dados:
			for col in row:
				linha += str(col) + ';'

			linha = linha[:-1]
			linha += "\n"

		file.write('\ufeff') #BOM
		file.write(linha)
		file.close()

	except (Exception) as e:
		print('Erro ao salvar arquivo:')
		print(e)

def getVendedores():
	return [
		('ADRIANO', 'Zm4r5anSMhakSkfa'),
		('ERNANI', 'tD4RM2qzwdECn9Sv'),
		('ROBERTO', 'WB3QWmDM2Pqqww79'),
		('VITOR', 'jWxm26aQFgUDFjeP'),
		('GRACIELA', 'PAtvAj567hC9DG3S'),
		('HELENO', '4zeDttEPKtH2nQ8D'),
		('CORPORATIVO', 'egzgMY8AjpRvGrnG')
	]

def init():
	admin = ('ADMIN', 'Ypv52Y5c8tG5Pgys')

	conn = connect()
	cur = conn.cursor()
	sql = readSqlVendedor()

	for v in getVendedores():
		cur.execute( sql.format(vendedor="'"+v[0]+"'") )
		salvar(cur.fetchall(), v)

	sql = readSqlAdmin()
	cur.execute(sql)
	salvar(cur.fetchall(), admin)

	cur.close()
	conn.close()
	print ('OK')


init()