diff --git a/bin/restore-database.sh b/bin/restore-database.sh new file mode 100755 index 00000000..1772e242 --- /dev/null +++ b/bin/restore-database.sh @@ -0,0 +1,11 @@ +#!/bin/bash + +while getopts d:f: flag +do + case "${flag}" in + d) database=${OPTARG};; + f) filename=${OPTARG};; + esac +done + +pg_restore --verbose --clean --no-acl --no-owner -d $database $filename diff --git a/server/core/management/commands/extend_license.py b/server/core/management/commands/extend_license.py new file mode 100644 index 00000000..0021e67a --- /dev/null +++ b/server/core/management/commands/extend_license.py @@ -0,0 +1,24 @@ +from datetime import date, timedelta +from django.core.management import BaseCommand + +from users.models import User + + +class Command(BaseCommand): + def add_arguments(self, parser): + parser.add_argument("username") + + def handle(self, *args, **options): + username = options["username"] + try: + user = User.objects.get(username=username) + today = date.today() + delta = timedelta(180) + user.license_expiry_date = today + delta + user.save() + self.stdout.write( + f"new date set for user{user.username}: {user.license_expiry_date}" + ) + + except Exception as e: + self.stdout.write(e)