import re

with open('/root/reservation-system/app/main.py', 'r') as f:
    content = f.read()

# Einfachere Modifikation - füge Admin-Check hinzu nachdem rooms geladen wurden
# Suche nach der rooms Funktion und ergänze sie

old_code = """@app.route('/api/rooms')
def rooms():
    is_admin = session.get('user_role') == 'admin'"""

new_code = """@app.route('/api/rooms')
def rooms():
    is_admin = session.get('user_role') == 'admin'
    
    with get_db() as db:"""

content = content.replace(old_code, new_code)

with open('/root/reservation-system/app/main.py', 'w') as f:
    f.write(content)

print("Rooms endpoint updated")