import firebase_admin
from firebase_admin import credentials, firestore

def initialize_firebase():
    # Ścieżka do pliku JSON z kluczem usługi
    cred = credentials.Certificate('./dexterlab.json')

    # Inicjalizacja aplikacji Firebase
    firebase_admin.initialize_app(cred)

    # Inicjalizacja klienta Firestore
    return firestore.client()

def save_data(db):
    try:
        # Przykład zapisu danych do kolekcji 'users'
        doc_ref = db.collection('users').document('user_id_123')
        doc_ref.set({
            'first_name': 'Jan',
            'last_name': 'Kowalski',
            'age': 30
        })
        print("Dane zostały zapisane")

    except Exception as e:
        print(f"Wystąpił błąd podczas zapisywania danych: {e}")

def read_data(db):
    try:
        # Pobranie dokumentu z kolekcji 'users'
        doc = db.collection('users').document('user_id_123').get()

        if doc.exists:
            print(f"Dane użytkownika: {doc.to_dict()}")
        else:
            print("Dokument nie istnieje")

    except Exception as e:
        print(f"Wystąpił błąd podczas odczytywania danych: {e}")

if __name__ == "__main__":
    # Inicjalizacja Firebase
    db = initialize_firebase()

    # Zapis danych do Firestore
    save_data(db)

    # Odczyt danych z Firestore
    read_data(db)