crunk
3 years ago
4 changed files with 45 additions and 3 deletions
@ -0,0 +1,41 @@ |
|||
import sys |
|||
from app import create_app, db |
|||
from sqlalchemy.exc import ( |
|||
InvalidRequestError, |
|||
InterfaceError, |
|||
DataError, |
|||
DatabaseError, |
|||
) |
|||
from usermodel import User # noqa: F401 |
|||
from distribusimodel import Distribusis # 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() |
Loading…
Reference in new issue