41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
import sys
|
|
|
|
from sqlalchemy.exc import (DatabaseError, DataError, InterfaceError,
|
|
InvalidRequestError)
|
|
|
|
from app import create_app, db
|
|
from models.distribusimodel import Distribusis # noqa: F401
|
|
from models.usermodel import User # noqa: F401
|
|
|
|
|
|
def admintool():
|
|
"""Admin CLI tool. To elevate a user to admin"""
|
|
app = create_app()
|
|
app.app_context().push()
|
|
elevateusertoadmin()
|
|
|
|
|
|
def elevateusertoadmin():
|
|
"""To elevates user of first command line argument to admin"""
|
|
user = User.query.filter_by(email=sys.argv[1]).first()
|
|
print(f"user {user.username} found with email {user.email}")
|
|
try:
|
|
user.admin = True
|
|
db.session.commit()
|
|
print(f"Account {user.email} succesfully made into an admin")
|
|
except InvalidRequestError:
|
|
db.session.rollback()
|
|
print("Something went wrong!")
|
|
except InterfaceError:
|
|
db.session.rollback()
|
|
print("Error connecting to the database")
|
|
except DataError:
|
|
db.session.rollback()
|
|
print("Invalid Entry")
|
|
except DatabaseError:
|
|
db.session.rollback()
|
|
print("Error connecting to the database")
|
|
|
|
|
|
admintool()
|