postgresql_import.py

"""
Dateiname: postgresql_import.py
Autor: Marcel Weller
"""

import psycopg2 as psy

# ein Paket um auf PostgreSQL Datenbanken zugreifen zu können (siehe: https://www.psycopg.org/docs/)
# vgl: https://pynative.com/python-postgresql-tutorial/

'''vor dem Ausführen des Codes muss die Datenbank erstellt werden.
Dazu müssen die Schritte der Dokumentaion Punkt 2.2 "Run the database locally" ausgeführt werden.
vgl. https://open-discourse.github.io/open-discourse-documentation/1.1.0/run-the-database-locally.html
Folgendes im Terminal/CMD ausführen:
einmalig:     docker pull ghcr.io/open-discourse/open-discourse/database:latest
zum Starten:  docker run --env POSTGRES_USER=postgres --env POSTGRES_DB=postgres --env POSTGRES_PASSWORD=postgres  -p 5432:5432 -d ghcr.io/open-discourse/open-discourse/database'''

# die Verbindung zur Datenbank aufbauen
conn = psy.connect(
    host="localhost",
    database="next",
    user="postgres",
    password="postgres",
    port="5432")
cur = None


# Abrufen von Informationen aus der Datenbank
def alle_reden():
    """Mit dieser Funktion werden alle Reden, aller Politiker aus der Datenbank ausgegeben."""
    try:
        conn

        cur = conn.cursor()  # Erstellung einer Variable zur Übergabe von SQL Code

        # Alle Werte der Spalte "speech_content" aus der Tabelle "speeches" ausgeben
        cur.execute("SELECT speech_content FROM open_discourse.speeches;")
        return cur.fetchall()
        # for alles in cur.fetchall():
        #    print(alles)

    except Exception as error:  # Wenn ein Fehler auftritt, wird der passende "error" ausgeworfen
        print("Fehler beim Aufbau der Datenbankverbindung (PostgreSQL): ", error)
    """finally: # Datenbankverbindung beenden
        if conn:
            cur.close()
            conn.close()
            print("Datenbankverbindung wurde beendet.")"""


class sql_abfragen:
    """Mit dieser Klasse können verschiedene Abfragen in der PostgreSQL Datenbank getätigt werden."""

    def __init__(self):
        self

    def alle_politiker(spalten):
        """Mit dieser Funktion können vom User ausgewählte Spalten aus der Tabelle 'politicians' der Datenbank
        abgerufen werden."""
        try:
            conn

            cur = conn.cursor()  # Erstellung einer Variable zur Übergabe von SQL Code

            # Alle Werte der ausgewählten Spalten aus der Tabelle "politicians" ausgeben
            cur.execute("select " + spalten + " from open_discourse.politicians")
            return cur.fetchall()
            # for politiker in cur.fetchall():
            #    print(politiker)

        except Exception as error:  # Wenn ein Fehler auftritt, wird der passende "error" ausgeworfen
            print("Fehler beim Aufbau der Datenbankverbindung (PostgreSQL): ", error)
        """finally: # Datenbankverbindung beenden
            if conn:
                cur.close()
                conn.close()
                print("Datenbankverbindung wurde beendet.")"""

    def reden_eines_pol(pol_id):
        """Mit dieser Funktion werden alle Reden einer vom User ausgewählten Politiker-ID (eines Politikers)
        ausgegeben."""
        try:
            conn

            cur = conn.cursor()  # Erstellung einer Variable zur Übergabe von SQL Code

            # Alle Werte der Spalte "speech_content" aus der Tabelle "speeches" ausgeben, die zur eingegebenen ID gehören
            cur.execute("SELECT speech_content FROM open_discourse.speeches where politician_id = " + pol_id + ";")
            return cur.fetchall()
            # for reden in cur.fetchall():
            #    print(reden)

        except Exception as error:  # Wenn ein Fehler auftritt, wird der passende "error" ausgeworfen
            print("Fehler beim Aufbau der Datenbankverbindung (PostgreSQL): ", error)
        """finally: # Datenbankverbindung beenden
            if conn:
                cur.close()
                conn.close()
                print("Datenbankverbindung wurde beendet.")"""