Как скрыть кнопку регистрации администратора, если пользователь администратора существует

Question

Я создаю небольшое flask приложение, которое требует, чтобы администратор был первым пользователем после установки, чтобы быть администратором, но после того, как этот пользователь создан, кнопка «регистрация» скрывается.

Im на самом деле не знаю, с чего начать, так как у меня в данный момент есть кнопка «зарегистрироваться» на экране выхода из системы, но я не уверен, как go скрыть это после создания первоначальной учетной записи администратора.

Любая помощь будет оценена.

Answer 1

Вы можете просто посчитать количество пользователей в пользовательской таблице. Если он равен нулю, вы можете показать кнопку регистрации, иначе она будет скрыта.

userCount = db.session.execute('select count(id) as c from user').scalar()
return render_template('register.html', userCount=userCount)

Проверьте, каково значение userCount, если оно равно нулю (0) или нет:

{% if userCount == 0 %}
    <a href="{{ url_for('register')}}">register</a>
{% endif %}