distribusi-verse/verse/admintool.py

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()