Das deutsche Python-Forum

Code: Alles auswählen

from pathlib import Path

pfad = Path('C:/Users/chris/Desktop/Objekte')

print("Das ist meine Datenbank verschiedener Deepsky Objekte")
print("Möchtest du eine Information hinzufügen oder abrufen?")
auswahl = input()
if auswahl == "abrufen":
    print("Welches Objekt möchtest du abrufen? Bitte Namen oder Katalog-Nummer nennen")
    objekt = input()
    filename = pfad / objekt
    print(filename.read_text())
elif auswahl == "hinzufügen":
    print("Wie heißt das Objekt welches sie hinzufügen wollen?")
    objekt = input()
    print("Nennen sie bitte die Katalog-Nummer im NGC")
    katalog_nummer = input()
    print("Nennen sie bitte das Sternbild")
    sternbild = input()
    print("Nennen sie bitte die magnitude")
    magnitude = input()
    print ("Nennen sie bitte zu welcher Jahreszeit es gehört")
    jahreszeit = input()

    filename = pfad / objekt
    filename.write_text(
        f"{objekt}\n"
        f"{katalog_nummer}\n"
        f"Sternbild:{sternbild}\n"
        f"Jahreszeit:{jahreszeit}\n"
    )

Code: Alles auswählen

import inquirer
from os.path import exists as file_exists
import csv


def main():
    SAVE_FILE = "sternenhimmel.csv"
    input_var = input("Daten (h)inzufügen oder (a)brufen? >> ")
    if input_var == "a":
        if not file_exists(SAVE_FILE):
            print("Fehler: Keine Datei gefunden.")
            return
        else:
            input_search = input("Wonach soll gesucht werden?"
                                 " >> ").lower()
            with open(SAVE_FILE) as csvfile:
                reader = csv.DictReader(csvfile)
                for row in reader:
                    if input_search in row["objektname"].lower() or \
                       input_search in row["katalog_nr"].lower():
                        for name in row:
                            print(f"{name.title()}: {row[name]}")
                        break
                else:
                    print("Keinen passenden Eintrag gefunden.")
    elif input_var == "h":
        with open(SAVE_FILE) as csvfile:
            reader = csv.DictReader(csvfile)
            objekte = {row["objektname"] for row in reader}
        answers = inquirer.prompt([inquirer.Text('objektname',
                                                 message="Objektname:")])
        if answers["objektname"] in objekte:
            print("Fehler: Objekt ist bereits in der Datei.")
            return
        questions = [inquirer.Text('katalog_nr', message="Katalog Nummer:"),
                     inquirer.Text('sternenbild', message="Sternenbild:"),
                     inquirer.Text('magnidute', message="Magnidute:")]
        answers.update(inquirer.prompt(questions))
        questions = [inquirer.List('jahreszeit', message="Jahreszeit?",
                                   choices=['Winter', 'Frühling',
                                            'Sommer', 'Herbst'], ), ]
        answers.update(inquirer.prompt(questions))
        if file_exists(SAVE_FILE):
            file_mode = "a"
        elif not file_exists(SAVE_FILE):
            file_mode = "w"
        with open(SAVE_FILE, mode=file_mode) as csvfile:
            fieldnames = ['objektname', 'katalog_nr', 'sternenbild',
                          'magnidute', 'jahreszeit']
            writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
            if file_mode == "w":
                writer.writeheader()
            writer.writerow(answers)
    else:
        print("Falsche Eingabe.")


if __name__ == '__main__':
    main()

Code: Alles auswählen

import inquirer
from os.path import exists as file_exists
import csv


SAVE_FILE = "sternenhimmel.csv"


def main():
    antwort = input("Daten (h)inzufügen oder (a)brufen? >> ")
    if antwort == "a":
        if not file_exists(SAVE_FILE):
            print("Fehler: Keine Datei gefunden.")
            return
        else:
            name_search = input("Wonach soll gesucht werden?"
                                " >> ").lower()
            with open(SAVE_FILE) as csvfile:
                reader = csv.DictReader(csvfile)
                for row in reader:
                    if name_search in row["objektname"].lower() or \
                       name_search in row["katalog_nr"].lower():
                        for name in row:
                            print(f"{name.title()}: {row[name]}")
                        break
                else:
                    print("Keinen passenden Eintrag gefunden.")
    elif antwort == "h":
        if file_exists(SAVE_FILE):
            file_mode = "a"
            with open(SAVE_FILE) as csvfile:
                reader = csv.DictReader(csvfile)
                objekte = {row["objektname"] for row in reader}
        else:
            file_mode = "w"
        answers = inquirer.prompt([inquirer.Text('objektname',
                                                 message="Objektname:")])
        if file_mode == "a":
            if answers["objektname"] in objekte:
                print("Fehler: Objekt ist bereits in der Datei.")
                return
        questions = [inquirer.Text('katalog_nr', message="Katalog Nummer:"),
                     inquirer.Text('sternenbild', message="Sternenbild:"),
                     inquirer.Text('magnidute', message="Magnidute:")]
        answers.update(inquirer.prompt(questions))
        questions = [inquirer.List('jahreszeit', message="Jahreszeit?",
                                   choices=['Winter', 'Frühling',
                                            'Sommer', 'Herbst'], ), ]
        answers.update(inquirer.prompt(questions))
        with open(SAVE_FILE, mode=file_mode) as csvfile:
            fieldnames = ['objektname', 'katalog_nr', 'sternenbild',
                          'magnidute', 'jahreszeit']
            writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
            if file_mode == "w":
                writer.writeheader()
            writer.writerow(answers)
    else:
        print("Falsche Eingabe.")


if __name__ == '__main__':
    main()