mirror of
https://codeberg.org/Mo8it/AdvLabDB.git
synced 2024-11-12 21:40:41 +00:00
54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
|
from flask_security import admin_change_password
|
||
|
|
|
||
|
|
from ... import app, db
|
||
|
|
from ...independent_funs import randomPassword
|
||
|
|
from ...models import Admin, User
|
||
|
|
from ..terminal_utils import box, spaced_hl, validating_input
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
print("This script will generate a new random password for a choosen admin.")
|
||
|
|
print()
|
||
|
|
|
||
|
|
with app.app_context():
|
||
|
|
with db.session.begin():
|
||
|
|
admin_query = Admin.query
|
||
|
|
admins = admin_query.filter(Admin.user.has(User.active == True))
|
||
|
|
activate_user = False
|
||
|
|
if admins.count() == 0:
|
||
|
|
print("There is no admin with an active user. The user of the choosen admin will be activated")
|
||
|
|
admins = admin_query.all()
|
||
|
|
activate_user = True
|
||
|
|
else:
|
||
|
|
admins = admins.all()
|
||
|
|
|
||
|
|
num_admins = len(admins)
|
||
|
|
|
||
|
|
prompt = "Admins:\n"
|
||
|
|
for ind, admin in enumerate(admins):
|
||
|
|
user = admin.user
|
||
|
|
prompt += f"[{ind}] {user.first_name} {user.last_name}: {user.email}\n"
|
||
|
|
prompt += "Enter number"
|
||
|
|
|
||
|
|
admin_index = validating_input(
|
||
|
|
prompt,
|
||
|
|
options=[str(ind) for ind in range(num_admins)],
|
||
|
|
format_function=lambda ans: int(ans),
|
||
|
|
)
|
||
|
|
|
||
|
|
choosen_admin_user = admins[admin_index].user
|
||
|
|
|
||
|
|
new_password = randomPassword()
|
||
|
|
|
||
|
|
admin_change_password(
|
||
|
|
choosen_admin_user, new_password, notify=False
|
||
|
|
) # Password is automatically hashed with this function
|
||
|
|
|
||
|
|
box(new_password, "New password")
|
||
|
|
|
||
|
|
print("Done!")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|