aboutsummaryrefslogtreecommitdiff
path: root/ihatemoney/manage.py
diff options
context:
space:
mode:
Diffstat (limited to 'ihatemoney/manage.py')
-rwxr-xr-xihatemoney/manage.py10
1 files changed, 9 insertions, 1 deletions
diff --git a/ihatemoney/manage.py b/ihatemoney/manage.py
index 3207b55..a9eca0f 100755
--- a/ihatemoney/manage.py
+++ b/ihatemoney/manage.py
@@ -10,7 +10,7 @@ from flask_migrate import Migrate, MigrateCommand
from werkzeug.security import generate_password_hash
from ihatemoney.run import create_app
-from ihatemoney.models import db
+from ihatemoney.models import db, Project
from ihatemoney.utils import create_jinja_env
@@ -57,6 +57,13 @@ class GenerateConfig(Command):
))
+class DeleteProject(Command):
+ def run(self, project_name):
+ demo_project = Project.query.get(project_name)
+ db.session.delete(demo_project)
+ db.session.commit()
+
+
def main():
QUIET_COMMANDS = ('generate_password_hash', 'generate-config')
@@ -76,6 +83,7 @@ def main():
manager.add_command('db', MigrateCommand)
manager.add_command('generate_password_hash', GeneratePasswordHash)
manager.add_command('generate-config', GenerateConfig)
+ manager.add_command('delete-project', DeleteProject)
manager.run()