Peter
25 lines
633 B
Python
25 lines
633 B
Python
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")
|