Argumentparser hinzugefügt

python/Muell/__init__.py

@@ -23,8 +23,6 @@ import csv
 import os
 import re
 import requests
-import httplib2
-import urllib
 import yaml
 
 from datetime import datetime, timedelta
@@ -32,34 +30,40 @@ from pathlib import Path
 
 class Muell:
 
-    # Variablen
-    __key = 'e21758b9c711463552fb9c70ac7d4273'
-    __modus = 'd6c5855a62cf32a4dadbc2831f0f295f'
+    def __init__(self, config_file='.muell.yaml'):
 
-    __host = 'api.abfall.io'
+        # Variablen
+        self.__config_file = config_file
+        self.__config = self.__readConfig()
+        self.__key = 'e21758b9c711463552fb9c70ac7d4273'
+        self.__modus = 'd6c5855a62cf32a4dadbc2831f0f295f'
 
-    __url = f'https://{__host}/?key={__key}&modus={__modus}&waction=export_csv'
-    __export_als = f"{{'action':'{__url}','target':''}}"
+        self.__host = 'api.abfall.io'
 
-    __current_year = datetime.today().year
-    __zeitraum = f'{__current_year}0101-{__current_year}1231'
+        self.__url = f'https://{self.__host}/?key={self.__key}&modus={self.__modus}'
+        self.__export_als = f"{{'action':'{self.__url}&waction=export_csv','target':''}}"
 
-    __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,
-    }
+        self.__current_year = datetime.today().year
+        self.__zeitraum = f'{self.__current_year}0101-{self.__current_year}1231'
 
-    __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',
-    }
+        self.__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': self.__zeitraum,
+            'f_export_als': self.__export_als,
+        }
+
+        self.__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',
+        }
+        kommune = self.__config['kommune']
+        self.__csv = f'muell-{kommune}-{self.__current_year}.csv'
 
 	# Methoden
     def __get_index_positions(self, list_of_elems, element):
@@ -73,21 +77,21 @@ class Muell:
         return index_pos_list
 
     def __readConfig(self):
-        ''' Liest ~/.muell.yaml und speichert Daten in Dictionary '''
+        ''' Liest Konfigdatei und speichert Daten in Dictionary '''
 
         home = Path.home() # home ist ohne / am Ende!
         config = {}
-        with open(f'{home}/.muell.yaml', 'r') as config_file:
+        with open(f'{home}/{self.__config_file}', 'r') as config_file:
             config = yaml.safe_load(config_file)
 
         return config
     
-    def __init(self, key, modus, host, headers):
+    def __start(self):
         ''' 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(self.__postdata))
+        url = f'{self.__url}&waction=init'
+        resp = requests.post(url, data = self.__postdata, headers = self.__headers)
+        content = resp.content
 
         result = re.findall(r"<input .*?name=.*? value=.*?/>", str(content))
 
@@ -104,13 +108,14 @@ class Muell:
 
         return name, value
 
-    def __read_file(self, jahr):
-        ''' Liest Daten aus muell<jahr>.csv und liefert ggf. Ergebnis zurück'''
+    def __read_file(self):
+        ''' Liest Daten aus self.__csv und liefert ggf. Ergebnis zurück'''
 	
         antwort_liste = list()
         headline = list()
+
         try:
-            with open(f'muell{jahr}.csv', 'r', encoding="latin1") as f:
+            with open(self.__csv, 'r', encoding="latin1") as f:
                 csv_reader = csv.reader(f, delimiter=';')
                 lines = 0
                 antwort_liste = []
@@ -128,26 +133,26 @@ class Muell:
         return (antwort_liste, headline)
 
     def get_data(self):
-        (antwort_liste, headline) = self.__read_file(self.__current_year)
-        config = self.__readConfig()
+        (antwort_liste, headline) = self.__read_file()
 
         if not antwort_liste:
             # Keine vernünftigen Daten, ergo mal gucken, was die Webseite ergibt
-            self.__postdata['f_id_kommune'] = config['kommune']
-            self.__postdata['f_id_strasse'] = config['strasse']
+            self.__postdata['f_id_kommune'] = self.__config['kommune']
+            self.__postdata['f_id_strasse'] = self.__config['strasse']
 
-            (name, value) = self.__init(key, modus, host, headers)
+            (name, value) = self.__start()
             if name != None and value != None:
                 self.__postdata[name] = value
 
-            http = httplib2.Http()
-            (resp, content) = http.request(url, "POST", headers = headers, body = urllib.parse.urlencode(self.__postdata))
+            url = f'{self.__url}&waction=export_csv'
+            resp = requests.post(url, data = self.__postdata, headers = self.__headers)
+            content = resp.content
 
             antwort = str(content)
-            f = open(f'muell{self.__current_year}.csv', 'wb')
+            f = open(self.__csv, 'wb')
             f.write(content)
             f.close()
-            (antwort_liste, headline) = self.__read_file(self.__current_year)
+            (antwort_liste, headline) = self.__read_file()
 
         tomorrow = (datetime.now() + timedelta(1)).strftime('%d.%m.%Y')
         index = set();
@@ -169,36 +174,36 @@ class Muell:
         for pos in index:
             tonnen.append(headline[pos])
 
-        return (tonnen, wird)
-
-    def ausgabe_telegram(self, tonnen, wird):
-        ''' Ausgabe Richtung Telegram'''
-
         if len(tonnen) > 0:
             tonnen = ' und '.join(tonnen)
+            ausgabe = f'Morgen {wird} {tonnen} abgeholt'
+        else:
+            ausgabe = None
+
+        return ausgabe
+
+    def ausgabe_telegram(self, ausgabe):
+        ''' Ausgabe Richtung Telegram'''
+
+        if ausgabe is not None:
             # Jetzt den Telegram Bot ansprechen
             try:
-                os.system(f'echo "Morgen {wird} {tonnen} abgeholt" | {config["telegram"]["pathBot"]} -u {config["telegram"]["tgReceiver"]}')
+                os.system(f'echo "{ausgabe}" | {self.__config["telegram"]["pathBot"]} -u {self.__config["telegram"]["tgReceiver"]}')
             except Exception as e:
                 os.system(f'logger "Fehler Muellbot Telegram: {e}"')
 
-    def ausgabe_signal(self, tonnen, wird):
+    def ausgabe_signal(self, ausgabe):
         ''' Ausgabe Richtung Signal'''
 
-        if len(tonnen) > 0:
-            
-            tonnen = ' und '.join(tonnen)
-
+        if ausgabe is not None:
             # Signal Bot
             try:
-                os.system(f'{config["signal"]["path"]} -a {config["signal"]["account"]} send -g {config["signal"]["group_id"]} -m "Morgen {wird} {tonnen} abgeholt"')
+                os.system(f'{self.__config["signal"]["path"]} -a {self.__config["signal"]["account"]} send -g {self.__config["signal"]["group_id"]} -m "{ausgabe}"')
             except Exception as e:
                 os.system(f'logger "Fehler Muellbot Signal: {e}"')
         
-    def ausgabe_text(self, tonnen, wird):
+    def ausgabe_text(self, ausgabe):
         ''' Ausgabe als Text'''
 
-        if len(tonnen) > 0:
-            tonnen = ' und '.join(tonnen)
-
-        print(f'Morgen {wird} {tonnen} abgeholt')
+        if ausgabe is not None:
+            print(f'{ausgabe}')

python/muell.py

@@ -1,9 +1,29 @@
 #!/usr/bin/python3
 
+import argparse
+
 from Muell import Muell
 	
 if __name__ == '__main__':
-    muell = Muell()
-    tonnen, wird = muell.get_data()
-    muell.ausgabe_telegram(tonnen, wird)
-    muell.ausgabe_signal(tonnen, wird)
+
+    parser = argparse.ArgumentParser(
+        prog='muell',
+        description='Benachrichtigt Empfänger über anstehende Müllabfuhr'
+    )
+    parser.add_argument('-c', '--config', default='.muell.yaml', help='Name der Konfigdatei. Diese muss sich im Benutzerverzeichnis befinden!')
+    parser.add_argument('-t', '--telegram', action='store_true', help='Ausgabe auf Telegram')
+    parser.add_argument('-s', '--signal', action='store_true', help='Ausgabe auf Signal')
+    parser.add_argument('-k', '--konsole', action='store_true', help='Ausgabe auf Konsole')
+    args = parser.parse_args()
+
+    muell = Muell(args.config)
+    ausgabe = muell.get_data()
+
+    if args.telegram == True:
+        muell.ausgabe_telegram(ausgabe)
+
+    if args.signal == True:
+        muell.ausgabe_signal(ausgabe)
+
+    if args.konsole == True:
+        muell.ausgabe_text(ausgabe)