#!/usr/bin/python3

'''
Um die Werte für kommune und strasse zu erfahren, rufe https://www.egst.de/de/abfallabholung/ per
Browser auf und ermittle die gesendeten Daten in der Entwicklerkonsole deines Browsers.

Die Angaben hier im Skript stehen für Hörstel als Kommune und Im Wiesengrund als Strasse

Um die Daten an Telegram zu senden, musst du dir ein Skript/Programm schreiben, dass die API
von Telegram verwendet und Daten an dich oder deine Gruppe senden kann!

Beispiel für eine Konfigurationsdatei:
tgReceiver: <hier die ID der Gruppe/ des Teilnehmers eintragen>
kommune: 2601
strasse: 2146
pathBot: '~/bin/YMBot'
tgBotOwner: <hier DEINE Telegram-ID eintragen>
'''

import csv
import os
import re
import requests
import httplib2
import urllib
import yaml

from datetime import datetime, timedelta
from pathlib import Path

# Funktionen
def get_index_positions(list_of_elems, element):
    ''' Returns the indexes of all occurrences of given element in
    the list- listOfElements '''
    index_pos_list = []
    for i in range(len(list_of_elems)):
        if list_of_elems[i] == element:
            index_pos_list.append(i)

    return index_pos_list

def readConfig():
	''' Liest ~/.muell.yaml und speichert Daten in Dictionary '''

	home = Path.home() # home ist ohne / am Ende!
	config = {}
	with open(f'{home}/.muell.yaml', 'r') as config_file:
		config = yaml.safe_load(config_file)

	return config
    
def init(key, modus, host, headers):
    ''' Initialisiert das System und sucht entsprechende Daten heraus '''
    
    url = f'https://{host}/?key={key}&modus={modus}&waction=init'
    http = httplib2.Http()
    (resp, content) = http.request(url, "POST", headers = headers, body = urllib.parse.urlencode(postdata))

    result = re.findall(r"<input .*?name=.*? value=.*?/>", str(content))

    datum = result[1].split(' ')

    name = None
    value = None
    for i in datum:
        i = i.split('=')
        if i[0] == 'name':
            name = i[1][1:-1]
        if i[0] == 'value':
            value = i[1][1:-1]

    return name, value

def read_file(jahr):
	''' Liest Daten aus muell<jahr>.csv und liefert ggf. Ergebnis zurück'''
	
	antwort_liste = list()
	headline = list()
	try:
		with open(f'muell{jahr}.csv', 'r', encoding="latin1") as f:
			csv_reader = csv.reader(f, delimiter=';')
			lines = 0
			antwort_liste = []
			for row in csv_reader:
				if lines == 0:
					# print(f'Column names are {", ".join(row)}')
					headline = row
					lines += 1
				else:
					if isinstance(row, list):
						antwort_liste.append(row)
	except FileNotFoundError:
		pass

	return (antwort_liste, headline)
	
# Variablen
key = 'e21758b9c711463552fb9c70ac7d4273'
modus = 'd6c5855a62cf32a4dadbc2831f0f295f'

host = 'api.abfall.io'

url = f'https://{host}/?key={key}&modus={modus}&waction=export_csv'
export_als = f"{{'action':'{url}','target':''}}"

current_year = datetime.today().year
zeitraum = f'{current_year}0101-{current_year}1231'

postdata = {
    'f_id_abfalltyp_0': '50',
    'f_id_abfalltyp_1': '161',
    'f_id_abfalltyp_2': '53',
    'f_id_abfalltyp_3': '187',
    'f_id_abfalltyp_4': '169',
    'f_abfallarten_index_max': '5',
    'f_abfallarten': '50,161,53,187',
    'f_zeitraum': zeitraum,
    'f_export_als': export_als,
}

headers = {
    'User-Agent': 'Mozilla/5.0 (Linux; x68_64; x64; rv:88.0) Gecko/20100101 Firefox/88.0',
    'Content-Type': 'application/x-www-form-urlencoded',
}

if __name__ == '__main__':
	(antwort_liste, headline) = read_file(current_year)
	config = readConfig()

	if not antwort_liste:
		# Keine vernünftigen Daten, ergo mal gucken, was die Webseite ergibt
		postdata['f_id_kommune'] = config['kommune']
		postdata['f_id_strasse'] = config['strasse']

		(name, value) = init(key, modus, host, headers)
		if name != None and value != None:
			postdata[name] = value

		http = httplib2.Http()
		(resp, content) = http.request(url, "POST", headers = headers, body = urllib.parse.urlencode(postdata))

		antwort = str(content)
		f = open(f'muell{current_year}.csv', 'wb')
		f.write(content)
		f.close()
		(antwort_liste, headline) = read_file(current_year)

	tomorrow = (datetime.now() + timedelta(1)).strftime('%d.%m.%Y')
	index = set();
	for row in antwort_liste:
		try:
			if (row.count(tomorrow) > 1):
				for i in range(len(row)):
					if (row[i] == tomorrow and i < 4):
						index.add(i)
			else:
				pos = row.index(tomorrow)
				if pos < 4:
					index.add(pos)
		except ValueError:
			pass

	wird = 'wird' if len(index) < 2 else 'werden'
	tonnen = []
	for pos in index:
		tonnen.append(headline[pos])

	if len(tonnen) > 0:
		tonnen = ' und '.join(tonnen)
		# Jetzt den Bot ansprechen
		os.system(f'echo "Morgen {wird} {tonnen} abgeholt" | {config["pathBot"]} -u {config["tgReceiver"]}')