distribusi-verse/verse/admintool.py

import sys

from sqlalchemy.exc import (
	DatabaseError,
	DataError,
	InterfaceError,
	InvalidRequestError,
)

from app import create_app, db
from models.distribusi_model import Distribusis  # noqa: F401
from models.user_model 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()