Files
infoscreen/server/dummy_events.py

236 lines
6.8 KiB
Python

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from models import Event, EventMedia, EventType
from dotenv import load_dotenv
import os
from datetime import datetime, timedelta
import random
# .env laden
load_dotenv()
DB_USER = os.getenv("DB_USER")
DB_PASSWORD = os.getenv("DB_PASSWORD")
DB_HOST = os.getenv("DB_HOST")
DB_NAME = os.getenv("DB_NAME")
db_conn_str = f"mysql+pymysql://{DB_USER}:{DB_PASSWORD}@{DB_HOST}/{DB_NAME}"
engine = create_engine(db_conn_str)
Session = sessionmaker(bind=engine)
session = Session()
# Beispiel-Events (client_uuid und created_by müssen existieren!)
client_uuid = "36671351-7a90-44af-bfef-a192c541373b"
created_by = 1 # ID eines existierenden Users
# --- Alte Testdaten löschen ---
# Zuerst alle EventMedia zu den Events dieses Clients löschen
event_ids = [e.id for e in session.query(
Event.id).filter_by(client_uuid=client_uuid).all()]
if event_ids:
session.query(EventMedia).filter(EventMedia.event_id.in_(
event_ids)).delete(synchronize_session=False)
# Dann alle Events dieses Clients löschen
session.query(Event).filter_by(client_uuid=client_uuid).delete()
session.commit()
# --- Ende Löschungen ---
now = datetime.now()
def random_time_on_day(day_offset: int, duration_hours: int = 1):
"""Erzeugt eine zufällige Start- und Endzeit zwischen 8 und 16 Uhr für einen Tag."""
start_hour = random.randint(8, 15 - duration_hours + 1)
start = (now + timedelta(days=day_offset)).replace(hour=start_hour,
minute=0, second=0, microsecond=0)
end = start + timedelta(hours=duration_hours)
return start, end
events = [
Event(
client_uuid=client_uuid,
title="Mathe Präsentation",
description="Präsentation zum Thema Algebra.",
start=random_time_on_day(1)[0],
end=random_time_on_day(1)[1],
event_type=EventType.presentation,
created_by=created_by,
updated_by=None,
is_active=True
),
Event(
client_uuid=client_uuid,
title="Schulwebsite Update",
description="Neue Inhalte auf der Schulwebsite.",
start=random_time_on_day(2)[0],
end=random_time_on_day(2)[1],
event_type=EventType.website,
created_by=created_by,
updated_by=None,
is_active=True
),
Event(
client_uuid=client_uuid,
title="Lehrvideo ansehen",
description="Video zum Thema Photosynthese.",
start=random_time_on_day(3)[0],
end=random_time_on_day(3)[1],
event_type=EventType.video,
created_by=created_by,
updated_by=None,
is_active=True
),
Event(
client_uuid=client_uuid,
title="Nachricht vom Lehrer",
description="Wichtige Mitteilung zum Unterricht.",
start=random_time_on_day(4)[0],
end=random_time_on_day(4)[1],
event_type=EventType.message,
created_by=created_by,
updated_by=None,
is_active=True
),
Event(
client_uuid=client_uuid,
title="Sonstiges Event",
description="Allgemeines Event ohne Kategorie.",
start=random_time_on_day(5)[0],
end=random_time_on_day(5)[1],
event_type=EventType.other,
created_by=created_by,
updated_by=None,
is_active=True
),
Event(
client_uuid=client_uuid,
title="WebUntis Termin",
description="Termin aus WebUntis importiert.",
start=random_time_on_day(6)[0],
end=random_time_on_day(6)[1],
event_type=EventType.webuntis,
created_by=created_by,
updated_by=None,
is_active=True
),
Event(
client_uuid=client_uuid,
title="Englisch Präsentation",
description="Präsentation zu Shakespeare.",
start=random_time_on_day(8)[0],
end=random_time_on_day(8)[1],
event_type=EventType.presentation,
created_by=created_by,
updated_by=None,
is_active=True
),
Event(
client_uuid=client_uuid,
title="Website Relaunch",
description="Vorstellung der neuen Schulwebsite.",
start=random_time_on_day(10)[0],
end=random_time_on_day(10)[1],
event_type=EventType.website,
created_by=created_by,
updated_by=None,
is_active=True
),
Event(
client_uuid=client_uuid,
title="Videoanalyse",
description="Analyse eines Lehrvideos.",
start=random_time_on_day(12)[0],
end=random_time_on_day(12)[1],
event_type=EventType.video,
created_by=created_by,
updated_by=None,
is_active=True
),
Event(
client_uuid=client_uuid,
title="WebUntis Info",
description="Weitere Termine aus WebUntis.",
start=random_time_on_day(14)[0],
end=random_time_on_day(14)[1],
event_type=EventType.webuntis,
created_by=created_by,
updated_by=None,
is_active=True
),
]
# Events anlegen und speichern
for event in events:
session.add(event)
session.commit() # Jetzt haben alle Events eine ID
# EventMedia mit den richtigen event_id-Werten anlegen
event_media = [
EventMedia(
event_id=events[0].id,
media_type=EventType.presentation,
url="https://example.com/praesentation-mathe.pdf",
order=1
),
EventMedia(
event_id=events[1].id,
media_type=EventType.website,
url="https://schule.example.com/update",
order=1
),
EventMedia(
event_id=events[2].id,
media_type=EventType.video,
url="https://example.com/photosynthese-video.mp4",
order=1
),
EventMedia(
event_id=events[3].id,
media_type=EventType.message,
url="https://example.com/lehrer-nachricht",
order=1
),
EventMedia(
event_id=events[4].id,
media_type=EventType.other,
url="https://example.com/sonstiges-event.pdf",
order=1
),
EventMedia(
event_id=events[5].id,
media_type=EventType.webuntis,
url="https://webuntis.example.com/termin",
order=1
),
EventMedia(
event_id=events[6].id,
media_type=EventType.presentation,
url="https://example.com/praesentation-englisch.pdf",
order=1
),
EventMedia(
event_id=events[7].id,
media_type=EventType.website,
url="https://schule.example.com/relaunch",
order=1
),
EventMedia(
event_id=events[8].id,
media_type=EventType.video,
url="https://example.com/videoanalyse.mp4",
order=1
),
EventMedia(
event_id=events[9].id,
media_type=EventType.webuntis,
url="https://webuntis.example.com/info",
order=1
),
]
for media in event_media:
session.add(media)
session.commit()
print("Test-Events und EventMedia wurden angelegt.")